在Asp.net core項目中使用WebSocket

更新時間：2022年08月27日 16:58:42 作者：天方

這篇文章介紹了在Asp.net core項目中使用WebSocket的方法，文中通過示例代碼介紹的非常詳細。對大家的學習或工作具有一定的參考借鑒價值，需要的朋友可以參考下

今天小試了一下在Asp.net core中使用websocket，這里記錄一下：

在 Startup 類的 Configure 方法中添加 WebSocket 中間件。

app.UseWebSockets();

它也可以傳入一些參數(shù)

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

添加WebSocket中間件的處理函數(shù)：

async Task WebSocketMiddleware(HttpContext context, Func<Task> next)
{
    //這里監(jiān)聽的是 /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函數(shù)中應用該中間件。

app.Use(WebSocketMiddleware);

最后再來完成我們的WebScoket功能函數(shù)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的程序或在線網(wǎng)頁測試一下了。