yawaflua/WebSockets
1
0
Fork 0
mirror of https://github.com/yawaflua/WebSockets.git synced 2026-02-04 14:04:11 +02:00
Code Issues Packages Projects Releases Wiki Activity

Adds more confs for handlers and try to fix error with swagger when using attribute

Browse Source
This commit is contained in:
Dmitri Shimanski
2025-04-01 18:54:55 +03:00
parent 54779bb793
commit 59ddda6077
8 changed files with 68 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
yawaflua.WebSockets/Core/WebSocketConfig.cs
View File

@@ -6,5 +6,7 @@ namespace yawaflua.WebSockets.Core;
public class WebSocketConfig
{
public Func<IWebSocket, HttpContext, Task>? OnOpenHandler { get; set; } = null;
public Func<Exception, IWebSocket, HttpContext, Task>? OnErrorHandler { get; set; } = null;
public Func<Exception, HttpContext, Task>? OnConnectionErrorHandler { get; set; } = null;
}

17
yawaflua.WebSockets/Core/WebSocketRouter.cs
View File

@@ -140,21 +140,22 @@ public class WebSocketRouter
var path = context.Request.Path.Value;
if (Routes.TryGetValue(path, out var handler))
if (path != null && Routes.TryGetValue(path, out var handler))
{
var webSocket = await context.WebSockets.AcceptWebSocketAsync();
await Task.Run(async () =>
{
IWebSocketClient client = null!;
var webSocketManager = new WebSocketManager();
try
{
var webSocketManager = new WebSocketManager();
var client = new WebSocketClient(context, webSocket, path);
client = new WebSocketClient(context, webSocket, path);
Clients.Add(client);
await Task.Run(async () =>
{
if (_webSocketConfig?.OnOpenHandler != null)
await _webSocketConfig.OnOpenHandler(new WebSocket(webSocket, client, webSocketManager)!, context);
await _webSocketConfig.OnOpenHandler(new WebSocket(webSocket, client, webSocketManager), context);
}, cts);
var buffer = new byte[1024 * 4];
@@ -179,6 +180,11 @@ public class WebSocketRouter
catch (Exception ex)
{
_logger.LogError(message:"Error with handling request: ",exception: ex);
await Task.Run(async () =>
{
if (_webSocketConfig?.OnErrorHandler != null)
await _webSocketConfig.OnErrorHandler(ex, new WebSocket(webSocket, client, webSocketManager), context);
}, cts);
}
}, cts);
@@ -186,11 +192,14 @@ public class WebSocketRouter
else
{
context.Response.StatusCode = 404;
throw new KeyNotFoundException("Path not found");
}
}
catch (Exception ex)
{
_logger.LogError(ex, $"Error when handle request {context.Connection.Id}: ");
if (_webSocketConfig!.OnConnectionErrorHandler != null)
await _webSocketConfig.OnConnectionErrorHandler(ex, context);
}
}
}