Использование Polly для повторных подключений и тайм-аутов

У меня возникла проблема с использованием Polly при попытке выполнить следующее:

• Логика повторного подключения - я попытался создать политику Polly, которая работает, когда вы пытаетесь выполнить StartAsync без подключения к Интернету. Однако, когда он достигает ReceiveLoop, политика больше не влияет на этот метод, и если наше соединение прервется в этот момент, оно никогда не попытается снова подключиться. Он просто выдает следующее исключение: Disconnected: The remote party closed the WebSocket connection without completing the close handshake.. Возможно, у меня должно быть два полиса: один в StartAsync и один в ReceiveLoop, но мне это почему-то кажется неправильным, поэтому я и задаю вопрос.

• Тайм-ауты - я хочу добавить тайм-ауты для каждого вызова метода ClientWebSocket, например. ConnectAsync, SendAsync и т. д. Я не очень хорошо знаком с Polly, но думаю, что эта политика автоматически делает это за нас. Однако мне нужен кто-то, кто это подтвердит. Под таймаутом я подразумеваю аналогичную _webSocket.ConnectAsync(_url, CancellationToken.None).TimeoutAfter(timeoutMilliseconds) логику, реализацию TimeoutAfter можно найти здесь. Пример того, как это сделали другие репозитории, можно найти здесь.

Упрощенно, я хочу сделать этот класс устойчивым, что означает, что вместо того, чтобы безуспешно пытаться подключиться к мертвому серверу веб-сокетов в течение 30 секунд, независимо от причины, он должен быстро дать сбой -> повторить попытку через 10 секунд -> быстро сбой. -> повторить попытку и так далее. Этот подождите и повторите логику должен повторяться до тех пор, пока мы не вызовем StopAsync или не избавимся от экземпляра.

Вы можете найти класс WebSocketDuplexPipe на Гитхаб.

public sealed class Client : IDisposable
{
    private const int RetrySeconds = 10;
    private readonly WebSocketDuplexPipe _webSocketPipe;
    private readonly string _url;

    public Client(string url)
    {
        _url = url;
        _webSocketPipe = new WebSocketDuplexPipe();
    }

    public Task StartAsync(CancellationToken cancellationToken = default)
    {
        var retryPolicy = Policy
            .Handle<Exception>(e => !cancellationToken.IsCancellationRequested)
            .WaitAndRetryForeverAsync(_ => TimeSpan.FromSeconds(RetrySeconds),
                (exception, calculatedWaitDuration) =>
                {
                    Console.WriteLine($"{exception.Message}. Retry in {calculatedWaitDuration.TotalSeconds} seconds.");
                });

        return retryPolicy.ExecuteAsync(async () =>
        {
            await _webSocketPipe.StartAsync(_url, cancellationToken).ConfigureAwait(false);
            _ = ReceiveLoop();
        });
    }

    public Task StopAsync()
    {
        return _webSocketPipe.StopAsync();
    }

    public async Task SendAsync(string data, CancellationToken cancellationToken = default)
    {
        var encoded = Encoding.UTF8.GetBytes(data);
        var bufferSend = new ArraySegment<byte>(encoded, 0, encoded.Length);
        await _webSocketPipe.Output.WriteAsync(bufferSend, cancellationToken).ConfigureAwait(false);
    }

    private async Task ReceiveLoop()
    {
        var input = _webSocketPipe.Input;

        try
        {
            while (true)
            {
                var result = await input.ReadAsync().ConfigureAwait(false);
                var buffer = result.Buffer;

                try
                {
                    if (result.IsCanceled)
                    {
                        break;
                    }

                    if (!buffer.IsEmpty)
                    {
                        while (MessageParser.TryParse(ref buffer, out var payload))
                        {
                            var message = Encoding.UTF8.GetString(payload);

                            _messageReceivedSubject.OnNext(message);
                        }
                    }

                    if (result.IsCompleted)
                    {
                        break;
                    }
                }
                finally
                {
                    input.AdvanceTo(buffer.Start, buffer.End);
                }
            }
        }
        catch (Exception ex)
        {
            Console.WriteLine($"Disconnected: {ex.Message}");
        }
    }
}