Allow control over websocket accept.

Author: timostamm

example/Startup.cs

@@ -26,6 +26,13 @@ public void Configure(IApplicationBuilder app, IWebHostEnvironment env)
                 var handler = WebSocketHandler.CreateFor<ChatController>(app.ApplicationServices);
                 a.Run(handler.HandleRequest);
             });
+            
+            app.Map("/ws-my", a =>
+            {
+                var handler = WebSocketHandler.CreateFor<MyController>(app.ApplicationServices);
+                a.Run(handler.HandleRequest);
+            });
+            
         }
     }
 }

src/AWebsocketController.cs

@@ -1,10 +1,17 @@
 using System.Net.WebSockets;
 using System.Threading.Tasks;
+using Microsoft.AspNetCore.Http;
 
 namespace TimoStamm.WebSockets.Controller
 {
     public abstract class AWebsocketController : IWebsocketController
     {
+        public async Task<Client> OnWebSocketRequest(HttpContext context)
+        {
+            var socket = await context.WebSockets.AcceptWebSocketAsync();
+            return new Client(socket, context);
+        }
+
         public virtual Task OnOpen(Client client)
         {
             return Task.CompletedTask;

src/IWebsocketController.cs

@@ -1,24 +1,19 @@
 using System.Net.WebSockets;
 using System.Threading.Tasks;
+using Microsoft.AspNetCore.Http;
 
 namespace TimoStamm.WebSockets.Controller
 {
-    
-    
     public interface IWebsocketController
     {
+        Task<Client> OnWebSocketRequest(HttpContext context);
 
         Task OnOpen(Client client);
 
-
         Task OnTextMessage(Client client, string text);
 
-        
         Task OnBinaryMessage(Client client, byte[] bytes);
 
-
         Task OnClose(Client client, WebSocketCloseStatus closeStatus, string closeStatusDescription);
-
     }
-    
 }

src/WebSocketHandler.cs

@@ -45,8 +45,8 @@ public async Task HandleRequest(HttpContext context)
             }
 
             var cancel = context.RequestAborted;
-            var socket = await context.WebSockets.AcceptWebSocketAsync();
-            var client = new Client(socket, context);
+            var client = await _controller.OnWebSocketRequest(context);
+            var socket = client.WebSocket;
 
             _clients.TryAdd(context.TraceIdentifier, client);