T type logging

2025-01-15 15:06:54 +00:00
parent f9f108448a
commit 8b3305efca
100 changed files with 1229 additions and 761 deletions
@@ -1,8 +1,60 @@
-using Core.Annotations;
+using System.Net.WebSockets;
+using System.Text;
+using Core.Annotations;
+using Core.Models.Utils;
+using Core.Servers.Ws;
+using Core.Utils;

 namespace Core.Servers;

 [Injectable(InjectionType.Singleton)]
 public class WebSocketServer
 {
-}
+    protected IEnumerable<IWebSocketConnectionHandler> _webSocketConnectionHandler;
+    protected ISptLogger<WebSocketServer> _logger;
+    protected JsonUtil _jsonUtil;
+
+    public WebSocketServer(
+        IEnumerable<IWebSocketConnectionHandler> webSocketConnectionHandlers,
+        ISptLogger<WebSocketServer> logger,
+        JsonUtil jsonUtil
+    )
+    {
+        _webSocketConnectionHandler = webSocketConnectionHandlers;
+        _logger = logger;
+        _jsonUtil = jsonUtil;
+    }
+
+    public Task OnConnection(HttpContext httpContext)
+    {
+        return httpContext.WebSockets.AcceptWebSocketAsync()
+            .ContinueWith(task => HandleCommunication(httpContext, task.Result));
+    }
+
+    private void HandleCommunication(HttpContext context, WebSocket webSocket)
+    {
+        var socketHandlers = _webSocketConnectionHandler
+            .Where(wsh => context.Request.Path.Value.Contains(wsh.GetHookUrl()))
+            .ToList();
+        if (socketHandlers.Count == 0)
+        {
+            var message =
+                $"Socket connection received for url {context.Request.Path.Value}, but there is not websocket handler configured for it";
+            _logger.Warning(message);
+            webSocket.SendAsync(
+                    Encoding.UTF8.GetBytes(_jsonUtil.Serialize(new { error = message })),
+                    WebSocketMessageType.Text,
+                    true,
+                    CancellationToken.None
+                )
+                .Wait();
+            webSocket.CloseAsync(WebSocketCloseStatus.ProtocolError, message, CancellationToken.None).Wait();
+        }
+
+        foreach (var wsh in socketHandlers)
+        {
+            wsh.OnConnection(webSocket, context).Wait();
+            _logger.Info($"WebSocketHandler \"{wsh.GetSocketId()}\" connected");
+        }
+    }
+}