ReconnectDelayCalculator.cs

이 파일의 문서화 페이지로 가기

using Dreamine.Communication.Abstractions.Options;
 
namespace Dreamine.Communication.Core.Resilience;

public sealed class ReconnectDelayCalculator
{
    private readonly ReconnectPolicy _policy;

    public ReconnectDelayCalculator(ReconnectPolicy? policy = null)
    {
        _policy = policy ?? new ReconnectPolicy();
    }

    public TimeSpan Calculate(int attemptCount)
    {
        if (attemptCount <= 1)
        {
            return NormalizeDelay(_policy.InitialDelay);
        }
 
        var initialMilliseconds = Math.Max(1, _policy.InitialDelay.TotalMilliseconds);
        var factor = Math.Max(1, _policy.BackoffFactor);
        var calculatedMilliseconds = initialMilliseconds * Math.Pow(factor, attemptCount - 1);
        var maxMilliseconds = Math.Max(
            initialMilliseconds,
            _policy.MaxDelay.TotalMilliseconds);
 
        var finalMilliseconds = Math.Min(calculatedMilliseconds, maxMilliseconds);
 
        return TimeSpan.FromMilliseconds(finalMilliseconds);
    }

    private static TimeSpan NormalizeDelay(TimeSpan delay)
    {
        if (delay <= TimeSpan.Zero)
        {
            return TimeSpan.FromMilliseconds(1);
        }
 
        return delay;
    }
}