如何正确关闭持久的 System.Net.WebSockets.ClientWebSocket 连接？

Question

我正在System.Net.WebSockets.ClientWebSocket创建一个 C# WebSocket 客户端来连接到服务器，但我不拥有该服务器的源代码。到目前为止一切正常，但我希望“正确”断开我的客户端。这是我的源代码的缩写版本：

public class Client : IDisposable
{
    private ClientWebSocket socket;
    private string endpoint;
    private Task receiveTask;

    public Client(string endpoint)
    {
        this.endpoint = endpoint;
        this.socket = new ClientWebSocket();
    }

    public async Task Initialize()
    {
        byte[] contentBuffer = Encoding.UTF8.GetBytes("notify message");

        // Connect to the server, and send a message to notify
        // it of the client's availability to receive data.
        await OpenConnection();
        await socket.SendAsync(new ArraySegment<byte>(contentBuffer), WebSocketMessageType.Text, true, CancellationToken.None);
    }

    private async Task OpenConnection()
    {
        if (socket.State != WebSocketState.Open)
        {
            await socket.ConnectAsync(new Uri(endpoint), CancellationToken.None);
            receiveTask = Task.Run(async () => await Receive());
        }
    }

    private async Task Receive()
    {
        while (socket.State == WebSocketState.Open)
        {
            byte[] buffer = new byte[1024];
            var result = await m_sessionSocket.ReceiveAsync(new ArraySegment<byte>(buffer), cancellationToken);

            if (result.MessageType == WebSocketMessageType.Close)
            {
                break;
            }
            else
            {
                using (var stream = new MemoryStream())
                {
                    stream.Write(buffer, 0, result.Count);
                    while (!result.EndOfMessage)
                    {
                        result = await socket.ReceiveAsync(new ArraySegment<byte>(buffer), CancellationToken.None);
                        stream.Write(buffer, 0, result.Count);
                    }

                    stream.Seek(0, SeekOrigin.Begin);
                    using (var reader = new StreamReader(stream, Encoding.UTF8))
                    {
                        string message = reader.ReadToEnd();
                        // Do stuff with received message
                    }
                }
            }
        }
    }

    public void Dispose()
    {
        // NOTE: This is a gross oversimplification. Assume in the
        // actual project that a proper implementation of the Dispose
        // pattern has been created.
        if (socket.State == WebSocketState.Open)
        {
            // How do I notify the server of disconnection?
            // The below has some sort of race condition, whereby
            // it hangs the client.
            Task.Run(async () => await socket.CloseAsync(WebSocketCloseStatus.NormalClosure, string.Empty, CancellationToken.None)).Wait();
        }

        if (receiveTask != null)
        {
            receiveTask.Dispose();
        }

        socket.Dispose()
    }
}

现在我面临的挑战。Task在完成之前我无法处理正在运行的数据接收。该任务被阻止等待从服务器接收。CloseAsync尝试从内部调用Dispose似乎会遇到某种僵局，因此这可能不是正确的方法。

显然，我可以使用 aCancellationTokenSource将 a 传递CancellationToken给该Receive方法，然后将其传递给ReceiveAsyncWebSocket 的方法。如果我这样做并告诉令牌源标记该令牌，它将释放该块，并且我可以IsCancellationRequested在接收方法内检查该令牌的标志，并在看到该标志时退出。但是，如果我这样做，它会抛出异常。我真的需要在 Dispose 方法中捕获异常（哎呀！）只是为了取消任务吗？另外，如果我取消任务，是否会正确通知服务器断开连接？或者我只是通过长时间运行（非等待）Task运行该ReceiveAsync方法而一开始就做错了？

Answer 1

在这种情况下，您不应使用 IDisposable。优雅地关闭 websocket 连接涉及异步 I/O。它需要一个干净的“关闭握手”，即发送方发送 CLOSE 帧并且接收方发回 CLOSE 帧。只有这样，连接才会转换为干净的套接字关闭（使用 FIN 而不是 RST）。

如果您需要“IDisposable”语义，您可以查看最新 .NET Core 中引入的新 IAsyncDisposable 模式。但一般来说，使用同步 Dispose() 模式并不是一个好主意。