.net Core 中间件实现WebSocket通讯

以下是一个简单的示例，展示如何在ASP.NET Core中使用中间件来处理WebSocket请求，并实现基本的通讯功能：

public class WebSocketMiddleware
{
    private readonly RequestDelegate _next;
 
    public WebSocketMiddleware(RequestDelegate next)
    {
        _next = next;
    }
 
    public async Task Invoke(HttpContext context)
    {
        if (context.WebSockets.IsWebSocketRequest)
        {
            WebSocket webSocket = await context.WebSockets.AcceptWebSocketAsync();
            await Echo(webSocket);
        }
        else
        {
            await _next(context);
        }
    }
 
    private async Task Echo(WebSocket webSocket)
    {
        var buffer = new byte[1024 * 4];
        WebSocketReceiveResult result = await webSocket.ReceiveAsync(new ArraySegment<byte>(buffer), CancellationToken.None);
        while (!result.CloseStatus.HasValue)
        {
            await webSocket.SendAsync(new ArraySegment<byte>(buffer, 0, result.Count), result.MessageType, result.EndOfMessage, CancellationToken.None);
            result = await webSocket.ReceiveAsync(new ArraySegment<byte>(buffer), CancellationToken.None);
        }
        await webSocket.CloseAsync(result.CloseStatus.Value, result.CloseStatusDescription, CancellationToken.None);
    }
}
 
public static class WebSocketMiddlewareExtensions
{
    public static IApplicationBuilder UseWebSocketMiddleware(this IApplicationBuilder builder)
    {
        return builder.UseMiddleware<WebSocketMiddleware>();
    }
}

然后在Startup.cs中配置中间件：

public void Configure(IApplicationBuilder app, IWebHostEnvironment env)
{
    // 其他配置...
 
    app.UseWebSocketMiddleware();
 
    // 其他配置...
}

这个示例中，WebSocketMiddleware类实现了中间件，用于处理WebSocket请求。它接受任何WebSocket请求，并在有消息到达时，将接收到的消息原样发送回客户端。Echo方法处理WebSocket连接的生命周期。

WebSocketMiddlewareExtensions类提供了一个扩展方法，用于在IApplicationBuilder上注册中间件。在Startup.cs的Configure方法中，通过调用UseWebSocketMiddleware来使中间件生效。