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

15
yawaflua.WebSockets/Attributes/WebSocketAttribute.cs
View File

@@ -1,5 +1,7 @@
using JetBrains.Annotations;
using Microsoft.AspNetCore.Mvc;
using Microsoft.AspNetCore.Mvc.ApiExplorer;
using Microsoft.AspNetCore.Mvc.Routing;
namespace yawaflua.WebSockets.Attributes;
@@ -18,13 +20,17 @@ namespace yawaflua.WebSockets.Attributes;
/// When applied to methods, defines specific sub-routes (requires class-level base path).
/// </remarks>
[AttributeUsage(AttributeTargets.Method | AttributeTargets.Class)]
public class WebSocketAttribute : RouteAttribute
[ApiExplorerSettings(IgnoreApi = true)]
public class WebSocketAttribute : RouteAttribute, IRouteTemplateProvider, IApiDescriptionVisibilityProvider
{
/// <summary>
/// Original route template specified in attribute
/// </summary>
public string Template { get; }
public new string Template { get; }
public new int? Order { get; } = 0;
public new string? Name { get; }
/// <summary>
/// Creates WebSocket route definition
/// </summary>
@@ -36,5 +42,8 @@ public class WebSocketAttribute : RouteAttribute
public WebSocketAttribute([RouteTemplate]string path) : base(path)
{
Template = path;
Name = path;
}
public bool IgnoreApi => true;
}

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);
}
}
}

9
yawaflua.WebSockets/Models/WebSocketClient.cs
View File

@@ -9,14 +9,15 @@ internal class WebSocketClient : IWebSocketClient
private HttpContext HttpContext { get; set; }
public Guid Id { get; } = Guid.NewGuid();
public string Path { get; }
public ConnectionInfo ConnectionInfo { get => HttpContext.Connection; }
public IDictionary<object, object> Items
public ConnectionInfo ConnectionInfo => HttpContext.Connection;
public IDictionary<object, object>? Items
{
get => HttpContext.Items;
set => HttpContext.Items = (value);
set => HttpContext.Items = value;
}
public HttpRequest HttpRequest { get => HttpContext.Request; }
public HttpRequest HttpRequest => HttpContext.Request;
public WebSocket webSocket { get; }
internal WebSocketClient(HttpContext httpContext, WebSocket webSocket, string path)

12
yawaflua.WebSockets/ServiceBindings.cs
View File

@@ -8,7 +8,8 @@ namespace yawaflua.WebSockets;
public static class ServiceBindings
{
public static IServiceCollection SettingUpWebSockets(this IServiceCollection isc, WebSocketConfig? socketOptions = null)
public static IServiceCollection SettingUpWebSockets(this IServiceCollection isc,
WebSocketConfig? socketOptions = null)
{
isc.AddSingleton<WebSocketRouter>();
if (socketOptions != null) isc.AddSingleton(socketOptions);
@@ -19,10 +20,11 @@ public static class ServiceBindings
return isc;
}
public static IApplicationBuilder ConnectWebSockets(this IApplicationBuilder iab)
public static IApplicationBuilder UseWebSocketWithSwagger(this IApplicationBuilder app)
{
iab.UseWebSockets();
iab.UseMiddleware<WebSocketMiddleware>();
return iab;
app.UseWebSockets();
app.UseMiddleware<WebSocketMiddleware>();
return app;
}
}

10
yawaflua.WebSockets/yawaflua.WebSockets.csproj
View File

@@ -19,11 +19,11 @@
</PropertyGroup>
<ItemGroup>
<PackageReference Include="JetBrains.Annotations" Version="(2023.3.0,)"/>
<PackageReference Include="Microsoft.AspNetCore.WebSockets" Version="(2.1.7,)" />
<PackageReference Include="Microsoft.Extensions.Hosting.Abstractions" Version="(6.0.0,)" />
<PackageReference Include="System.Net.WebSockets" Version="(4.0.0,)"/>
<PackageReference Include="Microsoft.AspNetCore.Mvc.Core" Version="(2.2.0,)" PrivateAssets="all">
<PackageReference Include="JetBrains.Annotations" Version=">= (2023.3.0,)"/>
<PackageReference Include="Microsoft.AspNetCore.WebSockets" Version=">= (2.1.7,)" />
<PackageReference Include="Microsoft.Extensions.Hosting.Abstractions" Version=">= (6.0.0,)" />
<PackageReference Include="System.Net.WebSockets" Version=">= (4.0.0,)"/>
<PackageReference Include="Microsoft.AspNetCore.Mvc.Core" Version=">= (2.2.2,)" PrivateAssets="all">
<Private>False</Private>
</PackageReference>
<None Include="..\README.md" Pack="true" PackagePath="\"/>