v3.4.6

apihost/Tnb.API.Entry/Extensions/ConfigureMvcControllerExtensions.cs

@@ -10,6 +10,7 @@ using Newtonsoft.Json;
 using OnceMi.AspNetCore.OSS;
 using JNPF.API.Entry.Handlers;
 using JNPF.Common.Cache;
+using Microsoft.AspNetCore.Authentication.JwtBearer;
 
 namespace Microsoft.Extensions.DependencyInjection;
 
@@ -18,60 +19,108 @@ namespace Microsoft.Extensions.DependencyInjection;
 /// </summary>
 public static class ConfigureMvcControllerExtensions
 {
-  /// <summary>
-  /// OSS服务配置.
-  /// </summary>
-  /// <param name="services"></param>
-  /// <returns></returns>
-  public static IServiceCollection ConfigureMvcController(this IServiceCollection services)
-  {
-    services.AddControllers()
-    .AddMvcFilter<RequestActionFilter>()
-    .AddInjectWithUnifyResult<RESTfulResultProvider>()
-    .AddJsonOptions(options => options.JsonSerializerOptions.PropertyNamingPolicy = null)
-        .AddNewtonsoftJson(options =>
+    /// <summary>
+    /// OSS服务配置.
+    /// </summary>
+    /// <param name="services"></param>
+    /// <returns></returns>
+    public static IServiceCollection ConfigureMvcController(this IServiceCollection services)
+    {
+        services.AddControllers()
+            .AddMvcFilter<RequestActionFilter>()
+            .AddInjectWithUnifyResult<RESTfulResultProvider>()
+            .AddJsonOptions(options => options.JsonSerializerOptions.PropertyNamingPolicy = null)
+            .AddNewtonsoftJson(options =>
+            {
+                // 默认命名规则
+                options.SerializerSettings.ContractResolver = new DefaultContractResolver();
+
+                // 设置时区为 UTC
+                options.SerializerSettings.DateTimeZoneHandling = DateTimeZoneHandling.Utc;
+
+                // 格式化json输出的日期格式
+                options.SerializerSettings.DateFormatString = "yyyy-MM-dd HH:mm:ss";
+
+                // 忽略空值
+                // options.SerializerSettings.NullValueHandling = NullValueHandling.Ignore;
+
+                // 忽略循环引用
+                options.SerializerSettings.ReferenceLoopHandling = ReferenceLoopHandling.Ignore;
+
+                // 格式化json输出的日期格式为时间戳
+                options.SerializerSettings.Converters.Add(new NewtonsoftDateTimeJsonConverter());
+            });
+
+        services.AddUnifyJsonOptions("special", new JsonSerializerSettings
         {
-          // 默认命名规则
-          options.SerializerSettings.ContractResolver = new DefaultContractResolver();
+            // 默认命名规则
+            ContractResolver = new DefaultContractResolver(),
 
-          // 设置时区为 UTC
-          options.SerializerSettings.DateTimeZoneHandling = DateTimeZoneHandling.Utc;
+            // 设置时区为 UTC
+            DateTimeZoneHandling = DateTimeZoneHandling.Utc,
 
-          // 格式化json输出的日期格式
-          options.SerializerSettings.DateFormatString = "yyyy-MM-dd HH:mm:ss";
-
-          // 忽略空值
-          // options.SerializerSettings.NullValueHandling = NullValueHandling.Ignore;
-
-          // 忽略循环引用
-          options.SerializerSettings.ReferenceLoopHandling = ReferenceLoopHandling.Ignore;
-
-          // 格式化json输出的日期格式为时间戳
-          options.SerializerSettings.Converters.Add(new NewtonsoftDateTimeJsonConverter());
+            // 格式化json输出的日期格式
+            DateFormatString = "yyyy-MM-dd HH:mm:ss",
         });
 
-    // 配置Nginx转发获取客户端真实IP
-    // 注1：如果负载均衡不是在本机通过 Loopback 地址转发请求的，一定要加上options.KnownNetworks.Clear()和options.KnownProxies.Clear()
-    // 注2：如果设置环境变量 ASPNETCORE_FORWARDEDHEADERS_ENABLED 为 True，则不需要下面的配置代码
-    services.Configure<ForwardedHeadersOptions>(options =>
-    {
-      options.ForwardedHeaders = ForwardedHeaders.All;
-      options.KnownNetworks.Clear();
-      options.KnownProxies.Clear();
-    });
+        // 配置Nginx转发获取客户端真实IP
+        // 注1：如果负载均衡不是在本机通过 Loopback 地址转发请求的，一定要加上options.KnownNetworks.Clear()和options.KnownProxies.Clear()
+        // 注2：如果设置环境变量 ASPNETCORE_FORWARDEDHEADERS_ENABLED 为 True，则不需要下面的配置代码
+        services.Configure<ForwardedHeadersOptions>(options =>
+        {
+            options.ForwardedHeaders = ForwardedHeaders.All;
+            options.KnownNetworks.Clear();
+            options.KnownProxies.Clear();
+        });
 
-    // Jwt处理程序
-    services.AddJwt<JwtHandler>(enableGlobalAuthorize: true);
+        // Jwt处理程序
+        services.AddJwt<JwtHandler>(enableGlobalAuthorize: true, jwtBearerConfigure: options =>
+        {
+            // 实现 JWT 身份验证过程控制
+            options.Events = new JwtBearerEvents
+            {
+                // 添加读取 Token 的方式
+                OnMessageReceived = context =>
+                {
+                    var httpContext = context.HttpContext;
 
-    // 跨域
-    services.AddCorsAccessor();
+                    // 判断请求是否包含 token 参数，如果有就设置给 Token
+                    if (httpContext.Request.Query.ContainsKey("token"))
+                    {
+                        // 设置 Token
+                        context.Token = httpContext.Request.Query["token"];
+                    }
 
-    services.AddConfigurableOptions<CacheOptions>();
-    services.AddSession();
-    services.AddMemoryCache(); // 使用本地缓存必须添加
+                    return Task.CompletedTask;
+                },
 
-    services.AddSingleton<IHttpContextAccessor, HttpContextAccessor>();
+                // Token 验证通过处理
+                OnTokenValidated = context =>
+                {
+                    return Task.CompletedTask;
+                },
+            };
+        });
 
-    return services;
-  }
+        // 跨域
+        services.AddCorsAccessor();
+
+        // 注册远程请求
+        services.AddRemoteRequest();
+
+        // 视图引擎
+        services.AddViewEngine();
+
+        // 脱敏词汇检测
+        services.AddSensitiveDetection();
+
+        // WebSocket服务
+        services.AddWebSocketManager();
+
+        services.AddSession();
+
+        services.AddSingleton<IHttpContextAccessor, HttpContextAccessor>();
+
+        return services;
+    }
 }