using System;

using System.Threading.Tasks;

using System.Threading;

using Polly;

public class Program

{

    public static async Task Main()

    {

        var service = new DummyService();

        await service.StartAsync();

        Console.ReadLine();

    }

}

​

class DummyService

{

    private DummyWebSocket _webSocket = new();

    public async Task StartAsync()

    {

        await _webSocket.StartAsync();

        _= Task.Run(async () =>

        {

            try

            {

                await CreateRetry().ExecuteAsync(

                    async () => await ReceiveLoopAsync());

            }

            catch (Exception ex)

            {

                Console.WriteLine("Failed after all retries");

            }

        });

    }

​

    public async Task ReceiveLoopAsync()

    {

        while(true)

        {

            var result = await _webSocket.ReadAync();

            Console.WriteLine(result);

        }

    }

​

    private IAsyncPolicy CreateRetry()

        => Policy.Handle<InvalidOperationException>()

        .WaitAndRetryAsync(3, _ => TimeSpan.FromMilliseconds(100),

        onRetryAsync: async (ex, ts) =>

        {

            await _webSocket.StopAsync();

            await _webSocket.StartAsync();

        });

}

​

class DummyWebSocket

{

    Random rnd = new();

    public async Task StartAsync(CancellationToken token = default)

    {

        Console.WriteLine("Starting");

        await Task.Delay(100, token);

    }

    public async Task<string> ReadAync(CancellationToken token = default)

    {

        Console.WriteLine("Reading");

        await Task.Delay(100, token);

        return rnd.Next() % 2 == 1 ? "data" : throw new InvalidOperationException();

    }

    public async Task StopAsync(CancellationToken token = default)

    {

        Console.WriteLine("Stopping");

        await Task.Delay(100, token);

    }

}