TEAMModelOS/TEAMModelOS.FunctionV4/Program.cs

using DinkToPdf;
using DinkToPdf.Contracts;
using Microsoft.Extensions.Configuration;
using Microsoft.Extensions.DependencyInjection;
using Microsoft.Extensions.DependencyInjection.Extensions;
using Microsoft.Extensions.Hosting;

using System;
using System.Collections.Concurrent;
using System.Collections.Generic;
using System.Configuration;
using System.Diagnostics;
using System.IO;
using System.Linq;
using System.Net.Http;
using System.Runtime.InteropServices;
using System.Text;
using System.Text.Json;
using System.Text.RegularExpressions;
using System.Threading;
using System.Threading.Tasks;
using TEAMModelOS.FunctionV4;
using TEAMModelOS.FunctionV4.ServiceBus;
using TEAMModelOS.Models;
using TEAMModelOS.SDK;
using TEAMModelOS.SDK.DI;
using TEAMModelOS.SDK.DI.Mail;
using TEAMModelOS.SDK.DI.Multiple;

namespace TEAMModelOS.FunctionV4
{
    public static class Program
    {
        public static async Task Main(string[] args)
        {
#if DEBUG

            
            Debugger.Launch();
#endif

             
            //var host = new HostBuilder()
            //.ConfigureFunctionsWorkerDefaults(funccontext => {
            //    funccontext.UseMiddleware<TimeMonitorMiddleware>();
            //} )

             var host = new HostBuilder().ConfigureFunctionsWorkerDefaults()
           .ConfigureAppConfiguration((context, configure) =>
           {
               //configure.AddCommandLine(args);
              // configure.AddJsonFile("local.settings.json", optional: true, reloadOnChange: true).AddEnvironmentVariables();
           })
           .ConfigureServices((context, services) =>
           {
           
               //services.Configure<Option>(options => context. Configuration.GetSection("Option").Bind(options));
               services.AddHttpClient();
               services.AddHttpClient<DingDing>();
               services.AddHttpClient<HttpTrigger>();
               //services.AddCoreAPIHttpService(context.Configuration);
               services.AddHttpClient<CoreAPIHttpService>().ConfigureHttpMessageHandlerBuilder(builder =>
               {
                   builder.PrimaryHandler = new HttpClientHandler
                   {
                       ServerCertificateCustomValidationCallback = (message, cert, chain, errors) => true
                   };
               });
               List<(string name, string connectionString)> cosmosDBConnects = new();
               cosmosDBConnects.Add(("Default", context.Configuration.GetSection("Azure:Cosmos:ConnectionString").Get<string>()));
               cosmosDBConnects.Add(("CoreServiceV1", context.Configuration.GetSection("CoreServiceV1:Cosmos:ConnectionString").Get<string>())); //CoreService V1 read only
               cosmosDBConnects.Add(("CoreServiceV2", context.Configuration.GetSection("CoreServiceV2:Cosmos:ConnectionString").Get<string>())); //CoreService V2
               cosmosDBConnects.Add(("CoreServiceV2CnRead", context.Configuration.GetSection("CoreServiceV2:CosmosCnRead:ConnectionString").Get<string>())); //CoreService V2 CN read only
               services.AddMultipleAzureCosmos(cosmosDBConnects);
               //services.AddAzureCosmos(context.Configuration.GetSection("Azure:Cosmos:ConnectionString").Get<string>());
               services.AddAzureServiceBus(context.Configuration.GetSection("Azure:ServiceBus:ConnectionString").Get<string>());
              
               services.AddAzureRedis(context.Configuration.GetSection("Azure:Redis:ConnectionString").Get<string>());
               services.AddSingleton(typeof(IConverter), new SynchronizedConverter(new PdfTools()));
               services.AddSnowflakeId(Convert.ToInt64(context.Configuration.GetSection("Option:LocationNum").Get<Int32>()), 1);
               List<(string name, string connectionString)> storageConnects = new();

               storageConnects.Add(("Default", context.Configuration.GetSection("Azure:Storage:ConnectionString").Get<string>()));      
               string LogStorageConnectionString = context.Configuration.GetSection("Azure:LogStorage:ConnectionString").Get<string>(); //防火墙日志统计文件
               if (!string.IsNullOrWhiteSpace(LogStorageConnectionString)) {
                   storageConnects.Add(("LogStorage", LogStorageConnectionString));   //防火墙日志统计文件
               }
               services.AddMultipleAzureStorage(storageConnects);
               //services.AddHostedService<BlobRootServiceBusSub>();
               services.AddSingleton<BackgroundWorkerQueue>();
               services.AddHostedService<LongRunningService>();
            //  services.AddIPSearcher("");
           })
           .Build();
            await host.RunAsync();
        }

        public class BackgroundWorkerQueue
        {
            private readonly ConcurrentQueue<Func<CancellationToken, Task>> _workItems = new();
            private readonly SemaphoreSlim _signal = new(0);

            public async Task<Func<CancellationToken, Task>> DequeueAsync(CancellationToken cancellationToken)
            {
                await _signal.WaitAsync(cancellationToken);
                _workItems.TryDequeue(out var workItem);

                return workItem;
            }

            public void QueueBackgroundWorkItem(Func<CancellationToken, Task> workItem)
            {
                ArgumentNullException.ThrowIfNull(workItem);

                _workItems.Enqueue(workItem);
                _signal.Release();
            }
        }

        public class LongRunningService : BackgroundService
        {

            public LongRunningService(BackgroundWorkerQueue queue)
            {
                _queue = queue;
            }
            private readonly BackgroundWorkerQueue _queue;

            protected override async Task ExecuteAsync(CancellationToken stoppingToken)
            {
                while (!stoppingToken.IsCancellationRequested)
                {
                    var workItem = await _queue.DequeueAsync(stoppingToken);

                    await workItem(stoppingToken);
                }
            }
        }
    }
}