Implementing WebSocket client in .net core 3.1 using ClientWebSocket class

Question

I'm implementing a simple client in C#/.NET Core 3.1, using ClientWebSocket library provided by Microsoft. While receiving data, I'm passing a cancellation token with 10 sec delay. If I don't receive any message in that duration, websocket goes to Aborted State and I have to connect again.

Can anyone help in this regard?

public async Task Receive()
{
    CancellationTokenSource tokenSource = null;
    WebSocketReceiveResult result = null;
    while(true)
    {
        try
        {
            if (webSocket.State == WebSocketState.Open)
            {
                tokenSource = new CancellationTokenSource(10000);
                var buffer = new ArraySegment<byte>(new Byte[8192]);
                using(var stream = new MemoryStream())
                {
                    do
                    {
                        result = await webSocket.ReceiveAsync(buffer, _cancellationTokenSource.Token);
                        stream.Write(buffer.Array, buffer.Offset, result.Count);
                    } while(!result.EndOfMessage);
                }
            }
            else
            {
                //Need to Handle this scenario
            }
        }
        catch (OpearationCancelledException ex)
        {
            //Handle Exception
        }
    }
}