OA2023/OASystem/OASystem.Api/Program.cs

using Microsoft.AspNetCore.Http.Connections;
using Microsoft.AspNetCore.Http.Features;
using Microsoft.AspNetCore.ResponseCompression;
using Microsoft.AspNetCore.Server.Kestrel.Core;
using Microsoft.Extensions.DependencyInjection.Extensions;
using OASystem.API.Middlewares;
using OASystem.API.OAMethodLib;
using OASystem.API.OAMethodLib.AMapApi;
using OASystem.API.OAMethodLib.APNs;
using OASystem.API.OAMethodLib.DeepSeekAPI;
using OASystem.API.OAMethodLib.GenericSearch;
using OASystem.API.OAMethodLib.Hub.Hubs;
using OASystem.API.OAMethodLib.JuHeAPI;
using OASystem.API.OAMethodLib.QiYeWeChatAPI;
using OASystem.API.OAMethodLib.Quartz.Jobs;
using OASystem.API.OAMethodLib.SignalR.HubService;
using OASystem.Infrastructure.Logging;
using Quartz;
using Quartz.Impl;
using Quartz.Spi;
using QuzrtzJob.Factory;
using Serilog.Events;
using System.IO.Compression;
using static OASystem.API.Middlewares.RateLimitMiddleware;

Console.Title = $"FMGJ OASystem Server";
var builder = WebApplication.CreateBuilder(args);
var basePath = AppContext.BaseDirectory;

//引入配置文件
var _config = new ConfigurationBuilder()
                 .SetBasePath(basePath)
                 .AddJsonFile("appsettings.json", optional: true, reloadOnChange: true)
                 .AddJsonFile("appsettings.Development.json", optional: true, reloadOnChange: true)
                 .AddEnvironmentVariables()
                 .Build();
builder.Services.AddSingleton(new AppSettingsHelper(_config));

//设置请求参数可不填
builder.Services.AddControllers(options => options.SuppressImplicitRequiredAttributeForNonNullableReferenceTypes = true);

//设置请求参数错误默认返回格式
builder.Services.AddControllers()
    .ConfigureApiBehaviorOptions(options =>
    {
        options.InvalidModelStateResponseFactory = context =>
        {
            var errors = context.ModelState
                .Where(e => e.Value.Errors.Count > 0)
                .ToDictionary(
                    kvp => kvp.Key,
                    kvp => kvp.Value.Errors.Select(e => e.ErrorMessage).ToArray()
                );

            var result = new JsonView
            {
                Code = 400,
                Msg = errors.FirstOrDefault().Value.FirstOrDefault() ?? "",
                Data = errors
            };

            return new BadRequestObjectResult(result);
        };
    });

// Add services to the container.
builder.Services.AddControllersWithViews();
builder.Services.AddControllers()
    .AddJsonOptions(options =>
    {
        //空字段不响应Response
        //options.JsonSerializerOptions.DefaultIgnoreCondition = JsonIgnoreCondition.WhenWritingNull;

        options.JsonSerializerOptions.Converters.Add(new NullJsonConverter());

        //时间格式化响应
        options.JsonSerializerOptions.Converters.Add(new DateTimeJsonConverter("yyyy-MM-dd HH:mm:ss"));

        //decimal 四位小数
        //options.JsonSerializerOptions.Converters.Add(new DecimalConverter(_decimalPlaces)); // 将保留小数位数参数传递给自定义序列化器
    });

builder.Services.TryAddSingleton<IHttpContextAccessor, HttpContextAccessor>();

#region 添加限流中间件服务注册

// 添加内存缓存（限流需要）
builder.Services.AddMemoryCache();

// 配置限流设置
builder.Services.Configure<RateLimitConfig>(
    builder.Configuration.GetSection("RateLimiting"));
#endregion

#region Gzip

builder.Services.AddResponseCompression(options =>
{
    options.EnableForHttps = true;
    options.Providers.Add<GzipCompressionProvider>();
});

builder.Services.Configure<GzipCompressionProviderOptions>(options =>
{
    options.Level = CompressionLevel.Optimal;
});

#endregion

#region Cors
builder.Services.AddCors(policy =>
{
    //policy.AddPolicy("Cors", opt => opt
    //.AllowAnyOrigin()
    //.AllowAnyHeader()
    //.AllowAnyMethod()
    //.WithExposedHeaders("X-Pagination"));

    policy.AddPolicy("Cors", opt => opt
            .SetIsOriginAllowed(origin =>
            {
                // 定义允许的来源列表
                var allowedOrigins = new List<string>
                    {
                       "http://132.232.92.186:9002"

                    };

                // 检查请求的来源是否在允许的列表中
                return allowedOrigins.Contains(origin);
            })

          //.AllowAnyOrigin()
          .AllowAnyHeader()
          .WithMethods("GET", "POST", "HEAD", "PUT", "DELETE", "OPTIONS")
          .AllowCredentials());

});
#endregion

#region 上传文件 
builder.Services.AddCors(policy =>
{
    policy.AddPolicy("Cors", opt => opt
    .AllowAnyOrigin()
    .AllowAnyHeader()
    .AllowAnyMethod()
    .WithExposedHeaders("X-Pagination"));
});

builder.Services.Configure<FormOptions>(options =>
{
    options.KeyLengthLimit = int.MaxValue;
    options.ValueLengthLimit = int.MaxValue;
    options.MultipartBodyLengthLimit = int.MaxValue;
    options.MultipartHeadersLengthLimit = int.MaxValue;
});

builder.Services.Configure<KestrelServerOptions>(options =>
{
    options.Limits.MaxRequestBodySize = int.MaxValue;
    options.Limits.MaxRequestBufferSize = int.MaxValue;
});

#endregion

#region 接口分组
var groups = new List<Tuple<string, string>>
{
    //new Tuple<string, string>("Group1","分组一"),
    //new Tuple<string, string>("Group2","分组二")
};
#endregion

#region 注入数据库

#region old

builder.Services.AddScoped(options =>
{
    return new SqlSugarClient(new List<ConnectionConfig>()
    {
        new() {
            ConfigId = DBEnum.OA2023DB,
            ConnectionString = _config.GetConnectionString("OA2023DB"),
            DbType = DbType.SqlServer,
            IsAutoCloseConnection = true,

        },
        new()
        {
            ConfigId = DBEnum.OA2014DB,
            ConnectionString = _config.GetConnectionString("OA2014DB"),
            DbType = DbType.SqlServer,
            IsAutoCloseConnection = true },
    }
    , db =>
    {
        // SQL执行完
        db.Aop.OnLogExecuted = (sql, pars) =>
        {
            // 超过1秒
            if (db.Ado.SqlExecutionTime.TotalSeconds > 1)
            {
                var fileName = db.Ado.SqlStackTrace.FirstFileName;
                var fileLine = db.Ado.SqlStackTrace.FirstLine;
                var FirstMethodName = db.Ado.SqlStackTrace.FirstMethodName;
                //执行完了可以输出SQL执行时间 (OnLogExecutedDelegate) 
                Console.WriteLine("NowTime:" + DateTime.UtcNow.ToString("yyyy-MM-dd HH:mm:ss"));
                Console.WriteLine("MethodName:" + FirstMethodName);
                Console.WriteLine("ElapsedTime:" + db.Ado.SqlExecutionTime.ToString());
                Console.WriteLine("ExecuteSQL:" + sql);
            }
        };
        //SQL执行前
        db.Aop.OnLogExecuting = (sql, pars) =>
        {
        };
        //SQL报错
        db.Aop.OnError = (exp) =>
        {
            //获取原生SQL推荐 5.1.4.63  性能OK
            //UtilMethods.GetNativeSql(exp.Sql, exp.Parametres);
            //获取无参数SQL对性能有影响，特别大的SQL参数多的，调试使用
            //UtilMethods.GetSqlString(DbType.SqlServer, exp.sql, exp.parameters);

        };
        //修改SQL和参数的值
        db.Aop.OnExecutingChangeSql = (sql, pars) =>
        {
            //sql=newsql
            //foreach(var p in pars) //修改
            return new KeyValuePair<string, SugarParameter[]>(sql, pars);
        };
    }
    );
});

#endregion

#endregion

//#region Identity 配置
//builder.Services.AddDataProtection();
////不要用 AddIdentity ， AddIdentity 是于MVC框架中的
//builder.Services.AddIdentityCore<User>(opt =>
//{
//    opt.Password.RequireDigit = false; //数字
//    opt.Password.RequireLowercase = false;//小写字母
//    opt.Password.RequireNonAlphanumeric = false;//特殊符号 例如 ￥#@！ 
//    opt.Password.RequireUppercase = false; //大写字母
//    opt.Password.RequiredLength = 6;//密码长度 6 
//    opt.Password.RequiredUniqueChars = 1;//相同字符可以出现几次
//    opt.Lockout.MaxFailedAccessAttempts = 5; //允许最多输入五次用户名/密码错误
//    opt.Lockout.DefaultLockoutTimeSpan = new TimeSpan(0, 5, 0);//锁定五分钟
//    opt.Tokens.PasswordResetTokenProvider = TokenOptions.DefaultEmailProvider; // 修改密码使用邮件【验证码模式】
//    opt.Tokens.EmailConfirmationTokenProvider = TokenOptions.DefaultEmailProvider;  //// 
//});
//var idBuilder = new IdentityBuilder(typeof(User), typeof(UserRole), services);
//idBuilder.AddEntityFrameworkStores<swapDbContext>().AddDefaultTokenProviders().AddRoleManager<RoleManager<UserRole>>().AddUserManager<UserManager<User>>();
//#endregion

#region 注入Swagger注释(启用)

if (AppSettingsHelper.Get("UseSwagger").ToBool())
{
    builder.Services.AddSwaggerGen(a =>
    {
        a.SwaggerDoc("v1", new OpenApiInfo
        {
            Version = "v1",
            Title = "Api",
            Description = "Api接口文档"
        });
        foreach (var item in groups)
        {
            a.SwaggerDoc(item.Item1, new OpenApiInfo { Version = item.Item1, Title = item.Item2, Description = $"{item.Item2}接口文档" });
        }
        a.DocumentFilter<SwaggerApi>();
        a.IncludeXmlComments(Path.Combine(basePath, "OASystem.Api.xml"), true);
        a.IncludeXmlComments(Path.Combine(basePath, "OASystem.Domain.xml"), true);
        a.AddSecurityDefinition("Bearer", new OpenApiSecurityScheme
        {
            Description = "Value: Bearer {token}",
            Name = "Authorization",
            In = ParameterLocation.Header,
            Type = SecuritySchemeType.ApiKey,
            Scheme = "Bearer"
        });
        a.AddSecurityRequirement(new OpenApiSecurityRequirement()
        {{
            new OpenApiSecurityScheme
            {
                Reference = new OpenApiReference
                {
                    Type = ReferenceType.SecurityScheme,
                    Id = "Bearer"
                }, Scheme = "oauth2", Name = "Bearer", In = ParameterLocation.Header }, new List<string>()
            }
        });

    });
}
#endregion

#region 添加校验

builder.Services.AddTransient<OASystemAuthentication>();
builder.Services.AddAuthentication(JwtBearerDefaults.AuthenticationScheme)
    .AddJwtBearer(options =>
        {
            options.TokenValidationParameters = new TokenValidationParameters
            {
                ValidateIssuer = true,
                ValidateAudience = true,
                ValidateLifetime = true,
                ValidateIssuerSigningKey = true,
                ValidAudience = "OASystem.com",
                ValidIssuer = "OASystem.com",
                IssuerSigningKey = new SymmetricSecurityKey(Encoding.UTF8.GetBytes(_config["JwtSecurityKey"])),
                ClockSkew = TimeSpan.FromSeconds(30), //过期时间容错值，解决服务器端时间不同步问题（秒）
                RequireExpirationTime = true,
            };
            options.Events = new JwtBearerEvents
            {
                OnMessageReceived = context =>
                {
                    var path = context.HttpContext.Request.Path;
                    //如果是signalr请求，需要将token转存，否则JWT获取不到token。OPTIONS请求需要过滤到，因为OPTIONS请求获取不到Token，用NGINX过滤掉OPTION请求.
                    if (path.StartsWithSegments("/ChatHub"))
                    {
                        string accessToken = context.Request.Query["access_token"].ToString();
                        if (string.IsNullOrWhiteSpace(accessToken))
                        {
                            accessToken = context.Request.Headers["Authorization"].ToString();
                        }
                        context.Token = accessToken.Replace("Bearer ", "").Trim();
                    }
                    return Task.CompletedTask;
                }
            };
        });

#endregion

#region 初始化日志
var environment = Environment.GetEnvironmentVariable("ASPNETCORE_ENVIRONMENT");
Log.Logger = new LoggerConfiguration()
        //不记录定时访问API
        .Filter.ByIncludingOnly(logEvent =>
        {
            if (logEvent.Properties.TryGetValue("RequestPath", out var pathValue))
            {
                var path = pathValue.ToString().Trim('"');
                return !path.StartsWith("/api/System/PotsMessageUnreadTotalCount");
            }
            return true;
        })
       .MinimumLevel.Information()
       .MinimumLevel.Override("Microsoft", LogEventLevel.Warning)
       .MinimumLevel.Override("System", LogEventLevel.Warning)
       .Enrich.FromLogContext()
       .WriteTo.Console()
       .WriteTo.File(Path.Combine("Logs", @"Log.txt"), rollingInterval: RollingInterval.Day)
       .CreateLogger();

// 

#region 出入境费用明细 专用记录器

// 指定磁盘绝对路径（示例：D盘的AppLogs文件夹）
var logDirectory = @"D:\OASystem\Logs\EnterExitCost";

// 自动创建目录（如果不存在）
try
{
    Directory.CreateDirectory(logDirectory);
    Log.Information($"日志目录已创建/确认存在: {logDirectory}");
}
catch (Exception ex)
{
    Log.Fatal($"无法创建日志目录 {logDirectory}: {ex.Message}");
    throw;
}

var eec_TextLogger = new LoggerConfiguration()
    .MinimumLevel.Information()
    .WriteTo.File(Path.Combine(logDirectory, "text-records-.txt"), rollingInterval: RollingInterval.Month)
    .CreateLogger();

#endregion

#region 团组步骤操作 专用记录器

// 指定磁盘绝对路径（示例：D盘的AppLogs文件夹）
var groupLogDir = @"D:\OASystem\Logs\GroupStepOP";

// 自动创建目录（如果不存在）
try
{
    Directory.CreateDirectory(groupLogDir);
    Log.Information($"日志目录已创建/确认存在: {groupLogDir}");
}
catch (Exception ex)
{
    Log.Fatal($"无法创建日志目录 {groupLogDir}: {ex.Message}");
    throw;
}

var groupStepOP_TextLogger = new LoggerConfiguration()
    .MinimumLevel.Information()
    .WriteTo.File(Path.Combine(groupLogDir, "text-records-.txt"), rollingInterval: RollingInterval.Month)
    .CreateLogger();

#endregion

#region 任务分配操作 专用记录器

// 指定磁盘绝对路径（示例：D盘的AppLogs文件夹）
var taskLogDir = @"D:\OASystem\Logs\TaskAllocation";

// 自动创建目录（如果不存在）
try
{
    Directory.CreateDirectory(taskLogDir);
    Log.Information($"日志目录已创建/确认存在: {taskLogDir}");
}
catch (Exception ex)
{
    Log.Fatal($"无法创建日志目录 {taskLogDir}: {ex.Message}");
    throw;
}

var task_TextLogger = new LoggerConfiguration()
    .MinimumLevel.Information()
    .WriteTo.File(Path.Combine(taskLogDir, "text-records-.txt"), rollingInterval: RollingInterval.Month)
    .CreateLogger();

#endregion

// 配置Serilog为Log;
builder.Host.UseSerilog();

builder.Services.AddSingleton<ITextFileLogger>(new TextFileLogger(eec_TextLogger));
builder.Services.AddSingleton<IGroupTextFileLogger>(new GroupTextFileLogger(groupStepOP_TextLogger));
builder.Services.AddSingleton<ITaskTextFileLogger>(new TaskTextFileLogger(task_TextLogger));
#endregion

#region 引入注册Autofac Module
builder.Host.UseServiceProviderFactory(new AutofacServiceProviderFactory());
var hostBuilder = builder.Host.ConfigureContainer<ContainerBuilder>(builder =>
{
    try
    {
        builder.RegisterModule(new AutofacRegister());
    }
    catch (Exception ex)
    {
        throw new Exception(ex.Message + "\n" + ex.InnerException);
    }
});
#endregion

#region AutoMapper

AutoMapper.IConfigurationProvider config = new MapperConfiguration(cfg =>
{
    cfg.AddProfile<_baseMappingProfile>();
});
builder.Services.AddSingleton(config);
builder.Services.AddScoped<IMapper, Mapper>();

#endregion

#region DeepSeek AI 服务

// 配置HTTP客户端
builder.Services.AddHttpClient<IDeepSeekService, DeepSeekService>();

#endregion

#region 聚合API 服务
builder.Services.AddControllersWithViews();
builder.Services.AddSingleton<IJuHeApiService, JuHeApiService>();
builder.Services.AddHttpClient("PublicJuHeApi", c => c.BaseAddress = new Uri("http://web.juhe.cn"));
builder.Services.AddHttpClient("PublicJuHeTranslateApi", c => c.BaseAddress = new Uri("http://apis.juhe.cn"));
#endregion

#region 企业微信API 服务

builder.Services.AddControllersWithViews();
builder.Services.AddSingleton<IQiYeWeChatApiService, QiYeWeChatApiService>();
builder.Services.AddHttpClient("PublicQiYeWeChatApi", c => c.BaseAddress = new Uri("https://qyapi.weixin.qq.com"));

#endregion

#region 有道API 服务
//builder.Services.AddControllersWithViews();
//builder.Services.AddSingleton<IYouDaoApiService, YouDaoApiService>();
//builder.Services.AddHttpClient("PublicYouDaoApi", c => c.BaseAddress = new Uri("https://openapi.youdao.com"));
#endregion

#region 高德地图API 服务
builder.Services.AddHttpClient<GeocodeService>();
#endregion

#region 通用搜索服务
builder.Services.AddScoped(typeof(DynamicSearchService<>));

#endregion

#region Quartz
builder.Services.AddSingleton<ISchedulerFactory, StdSchedulerFactory>();
builder.Services.AddSingleton<QuartzFactory>();
builder.Services.AddSingleton<ALiYunPostMessageJob>();
builder.Services.AddSingleton<TaskJob>();
builder.Services.AddSingleton<TaskNewsFeedJob>();
builder.Services.AddSingleton<PerformanceJob>();
builder.Services.AddSingleton<GroupProcessNodeJob>();
//# new business
builder.Services.AddControllersWithViews();
builder.Services.AddSingleton<IAPNsService, APNsService>();

builder.Services.AddSingleton<IJobFactory, IOCJobFactory>();

#endregion

#region SignalR
builder.Services.AddSignalR()
                .AddJsonProtocol(options =>
                {
                    options.PayloadSerializerOptions.PropertyNamingPolicy = null;
                });
builder.Services.TryAddSingleton(typeof(CommonService));
#endregion

var app = builder.Build();

//// 1. 异常处理器应该在最早的位置（除了日志等）
//app.UseExceptionHandler(new ExceptionHandlerOptions
//{
//    ExceptionHandlingPath = "/Home/Error", 
//    AllowStatusCode404Response = true
//});

//自定义异常中间件
//app.UseMiddleware<ExceptionHandlingMiddleware>();

//serilog日志 请求中间管道
app.UseSerilogRequestLogging(options =>
{
    //options.MessageTemplate = "HTTP {RequestMethod} {RequestPath} from {ClientIP} (UA: {UserAgent}, Referer: {Referer}) - {StatusCode} in {Elapsed} ms";

    options.MessageTemplate = "HTTP {RequestMethod} {RequestPath} from {ClientIP} (UA: {UserAgent}) - {StatusCode} in {Elapsed} ms";

    // 自定义日志级别
    options.GetLevel = (httpContext, elapsed, ex) =>
    {
        if (ex != null) return LogEventLevel.Error;
        if (httpContext.Response.StatusCode > 499) return LogEventLevel.Error;

        // 对健康检查等端点使用更低级别
        if (httpContext.Request.Path.StartsWithSegments("/health"))
            return LogEventLevel.Debug;

        return LogEventLevel.Information;
    };

    // 丰富日志上下文
    options.EnrichDiagnosticContext = (diagnosticContext, httpContext) =>
    {
        // 获取客户端IP（处理代理情况）
        var ipAddress = CommonFun.GetClientIpAddress(httpContext);
        var userAgent = CommonFun.DetectOS(httpContext.Request.Headers.UserAgent.ToString());

        // 添加IP和其他有用信息到日志上下文
        diagnosticContext.Set("ClientIP", ipAddress);
        diagnosticContext.Set("RequestHost", httpContext.Request.Host.Value);
        diagnosticContext.Set("UserAgent", userAgent);
        diagnosticContext.Set("Referer", httpContext.Request.Headers.Referer.ToString());

        // 对于API请求添加额外信息
        if (httpContext.Request.Path.StartsWithSegments("/api"))
        {
            diagnosticContext.Set("RequestContentType", httpContext.Request.ContentType);
            diagnosticContext.Set("RequestContentLength", httpContext.Request.ContentLength ?? 0);
        }
    };

});

AutofacIocManager.Instance.Container = app.UseHostFiltering().ApplicationServices.GetAutofacRoot();//AutofacIocManager

// Configure the HTTP request pipeline.
if (!app.Environment.IsDevelopment())
{
    app.UseExceptionHandler("/Home/Error");
}

app.UseStaticFiles();

app.UseRouting();

app.UseCors("Cors");  //Cors

//app.UseMiddleware<FixedPromptMiddleware>();

// 定义允许API的访问时间段
//var startTime = DateTime.Parse(_config["ApiAccessTime:StartTime"]);
//var endTime = DateTime.Parse(_config["ApiAccessTime:EndTime"]);
//app.UseMiddleware<TimeRestrictionMiddleware>(startTime, endTime);

//指定API操作记录信息
app.UseMiddleware<RecordAPIOperationMiddleware>();

app.UseAuthentication(); // 认证

app.UseMiddleware<RateLimitMiddleware>();

app.UseAuthorization();  // 授权

app.UseWhen(context =>
    context.Request.Path.StartsWithSegments("/api/MarketCustomerResources/QueryNewClientData"),
    branch => branch.UseResponseCompression());

// 授权路径
//app.MapGet("generatetoken", c => c.Response.WriteAsync(JWTBearer.GenerateToken(c)));

#region 启用swaggerUI
if (AppSettingsHelper.Get("UseSwagger").ToBool())
{
    app.UseSwagger();
    app.UseSwaggerUI(c =>
    {
        c.SwaggerEndpoint("/swagger/v1/swagger.json", "Ver0.1");
        foreach (var item in groups)
        {
            c.SwaggerEndpoint($"/swagger/{item.Item1}/swagger.json", item.Item2);
        }
        c.RoutePrefix = string.Empty;
        c.DocExpansion(Swashbuckle.AspNetCore.SwaggerUI.DocExpansion.None);
        c.DefaultModelsExpandDepth(-1);

        //c.EnableFilter();// 添加搜索功能
        //c.EnableDeepLinking(); // 启用深度链接
    });
}
#endregion

#region Quartz

//获取容器中的QuartzFactory
var quartz = app.Services.GetRequiredService<QuartzFactory>();
app.Lifetime.ApplicationStarted.Register(async () =>
{
    await quartz.Start();
});

app.Lifetime.ApplicationStopped.Register(() =>
{
    //Quzrtz关闭方法
    //quartz.Stop();
});

#endregion

#region SignalR

app.MapHub<ChatHub>("/ChatHub", options =>
{
    options.Transports =
        HttpTransportType.WebSockets |
        HttpTransportType.LongPolling;
});

#endregion

app.MapControllerRoute(
    name: "default",
    pattern: "{controller=Home}/{action=Index}/{id?}");



app.Run();