.NET中WebSocket接收多消息/多分片的处理疑问及最佳实践

我正在实现一个WebSocket服务器，采用二进制数据（而非文本）进行服务端与客户端间的通信。当使用以下代码等待并处理消息时，是否需要考虑消息被拆分为多个分片接收，以及同时接收两条消息的情况？

var buffer = new byte[256];
var result = await webSocket.ReceiveAsync(new ArraySegment<byte>(buffer), CancellationToken.None);

*我确认所有合法客户端/服务端发送的消息都能容纳在256字节的缓冲区中，且调用webSocket.SendAsync()时已将EndOfMessage标志设为true；
若确实需要处理上述情况，恳请提供能正确接收消息的最佳实践代码，我有一个仅能单次处理一条消息的await Process(byte[] data)方法。
我曾使用以下代码处理多分片消息，但不确定是否属于过度设计：

while (socket.State == WebSocketState.Open)
{
    var buffer = new byte[256];
    await using var dataStream = new MemoryStream();
    WebSocketReceiveResult result;
    do
    {
        result = await socket.ReceiveAsync(new ArraySegment<byte>(buffer), CancellationToken.None);
        if (result.MessageType == WebSocketMessageType.Close)
        {
            await socket.CloseAsync(WebSocketCloseStatus.NormalClosure, "", CancellationToken.None);
            return;
        }
        await dataStream.WriteAsync(buffer);
    } while (!result.EndOfMessage);
    await Process(dataStream.ToArray()); //Process functions do not detect multiple messages, and just can process one message at a time
    dataStream.SetLength(0); //Clear memory stream data
}

核心问题解答 #

• 必须处理消息分片：即便单条消息不超过256字节，WebSocket协议或底层TCP仍可能将消息拆分为多个分片传输。单次ReceiveAsync调用无法保证拿到完整消息，必须通过result.EndOfMessage标志判断接收是否完成。
• 无需额外处理多消息混接：WebSocket协议会严格维护消息边界，只有当一条消息的所有分片接收完毕（EndOfMessage=true）后，才会开始接收下一条消息的分片，不会出现两条消息内容混在一起的情况。

最佳实践代码 #

你的代码思路正确，但存在一个关键问题：写入MemoryStream时没有使用result.Count控制有效字节长度，会导致多余的无效字节被写入。修正后的可靠实现如下：

while (socket.State == WebSocketState.Open)
{
    var buffer = new byte[256];
    await using var dataStream = new MemoryStream();
    WebSocketReceiveResult result;
    
    do
    {
        result = await socket.ReceiveAsync(new ArraySegment<byte>(buffer), CancellationToken.None);
        
        // 处理关闭请求
        if (result.MessageType == WebSocketMessageType.Close)
        {
            await socket.CloseAsync(WebSocketCloseStatus.NormalClosure, "正常关闭", CancellationToken.None);
            return;
        }
        
        // 仅写入本次接收的有效字节
        await dataStream.WriteAsync(buffer, 0, result.Count);
    } while (!result.EndOfMessage);
    
    // 传递完整消息给处理方法
    await Process(dataStream.ToArray());
}

对原代码的点评 #

你的实现不属于过度设计，完全符合WebSocket消息接收的标准流程。唯一的瑕疵是未处理有效字节数，修正后即可作为生产级代码使用。

内容的提问来源于stack exchange，提问作者Iman Navidi