添加项目文件。

common/Tnb.WebSockets/Extensions/WebSocketApplicationBuilderExtensions.cs

@@ -0,0 +1,25 @@
+using JNPF.WebSockets;
+using Microsoft.AspNetCore.Http;
+
+namespace Microsoft.AspNetCore.Builder;
+
+/// <summary>
+/// WebSocket 中间件拓展.
+/// </summary>
+public static class WebSocketApplicationBuilderExtensions
+{
+    /// <summary>
+    /// 映射 WebSocket 管理.
+    /// </summary>
+    /// <param name="app"></param>
+    /// <param name="path"></param>
+    /// <param name="handler"></param>
+    /// <returns></returns>
+    public static IApplicationBuilder MapWebSocketManager(
+        this IApplicationBuilder app,
+        PathString path,
+        WebSocketHandler handler)
+    {
+        return app.Map(path, (_app) => _app.UseMiddleware<WebSocketMiddleware>(handler));
+    }
+}

common/Tnb.WebSockets/Extensions/WebSocketServiceCollectionExtensions.cs

@@ -0,0 +1,25 @@
+using JNPF.WebSockets;
+
+namespace Microsoft.Extensions.DependencyInjection;
+
+/// <summary>
+/// WebSocket服务集合拓展.
+/// </summary>
+public static class WebSocketServiceCollectionExtensions
+{
+    public static IServiceCollection AddWebSocketManager(this IServiceCollection services)
+    {
+        services.AddTransient<WebSocketConnectionManager>();
+
+        var types = AppDomain.CurrentDomain.GetAssemblies()
+              .SelectMany(a => a.GetTypes().Where(t => t.BaseType == typeof(WebSocketHandler)))
+              .ToArray();
+
+        foreach (var type in types)
+        {
+            services.AddSingleton(type);
+        }
+
+        return services;
+    }
+}