为 WebSocket 响应选择缓冲区大小

Question

我正在编写一个连接到websocket服务器的C#应用程序，并收到未知大小的JSON响应。为此，我正在使用 ClientWebSocket 类。

从客户端接收数据的唯一方法似乎是使用 ReceiveAsync 方法，该方法将ArraySegment<byte>作为参数：

var client = new ClientWebSocket();
client.ConnectAsync(new Uri($"ws://localhost:{port}"), token).Wait();
var result = new ArraySegment<byte>(new byte[1000]);
client.ReceiveAsync(result, token).Wait();

问题是，由于我不知道 JSON 响应会有多大，我不知道使支持该 ArraySegment 的缓冲区有多大。在本例中，我指定了 1000 个字节，这太小了，并且响应被截断。但是，我担心如果我将缓冲区大小设置为任意大（1,000,000 字节？），我会占用比我需要的更多的内存。

如何在不知道响应大小的情况下选择缓冲区大小？

Answer 1

如果我正确理解 API，如有必要，它会分多个部分为您提供 websocket 消息。

这意味着，如果从服务器发送的消息是 2048 字节，并且您使用 1000 字节的缓冲区并执行以下操作：

var buffer = new ArraySegment<byte>(new byte[1000]);
var result = await client.ReceiveAsync(buffer, token);

然后在第一次调用中，result.Count将设置为 1000，result.EndOfMessage将设置为 false 。这意味着您需要继续读取，直到EndOfMessage设置为 true，这意味着此示例有 3 次读取。

如果您需要在一个缓冲区中所有内容，并且无法忍受单独处理消息片段，则可以从一个小缓冲区开始，如果结果告诉您更多数据传入，则可以调整接收缓冲区的大小。因此，您还可以检查如果超过最大大小，接收是否停止。

int bufferSize = 1000;
var buffer = new byte[bufferSize];
var offset = 0;
var free = buffer.Length;
while (true)
{
    var result = await client.ReceiveAsync(new ArraySegment<byte>(buffer, offset, free), token);
    offset += result.Count;
    free -= result.Count;
    if (result.EndOfMessage) break;
    if (free == 0)
    {
        // No free space
        // Resize the outgoing buffer
        var newSize = buffer.Length + bufferSize;
        // Check if the new size exceeds a limit
        // It should suit the data it receives
        // This limit however has a max value of 2 billion bytes (2 GB)
        if (newSize > maxFrameSize)
        {
            throw new Exception ("Maximum size exceeded");
        }
        var newBuffer = new byte[newSize];
        Array.Copy(buffer, 0, newBuffer, 0, offset);
        buffer = newBuffer;
        free = buffer.Length - offset;
    }
}

当然，您还应该检查接收结果中的其他字段，例如 MessageType .