在Asp.netcore中实现websocket通信

2024-04-02 19:55

短信预约 -IT技能 免费直播动态提醒

今天小试了一下在Asp.net core中使用websocket，这里记录一下：

在 Startup 类的 Configure 方法中添加 WebSocket 中间件。

app.UseWebSockets();

它也可以传入一些参数

app.UseWebSockets(new WebSocketOptions()
{
    KeepAliveInterval = TimeSpan.FromSeconds(120),
    ReceiveBufferSize = 4 * 1024
});

添加WebSocket中间件的处理函数：

async Task WebSocketMiddleware(HttpContext context, Func<Task> next)
{
    //这里监听的是 /ws 作为地址
    if (context.Request.Path != "/ws")
    {
        await next();
        return;
    }
                                                                                        
    if (!context.WebSockets.IsWebSocketRequest)
    {
        context.Response.StatusCode = 400;
        return;
    }

    WebSocket webSocket = await context.WebSockets.AcceptWebSocketAsync();
    await Echo(context, webSocket);
}

在Startup 类的 Configure函数中应用该中间件。

app.Use(WebSocketMiddleware);

最后再来完成我们的WebScoket功能函数Echo

async Task Echo(HttpContext context, 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);
}

到此一个简单的websocket应用就完成了，我们可以随便找一个测试webscoket的程序或在线网页测试一下了。