// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
 
using System;
 
namespace Microsoft.AspNetCore.SignalR.Client.Internal;
 
internal sealed class DefaultRetryPolicy : IRetryPolicy
{
    internal static TimeSpan?[] DEFAULT_RETRY_DELAYS_IN_MILLISECONDS = new TimeSpan?[]
    {
            TimeSpan.Zero,
            TimeSpan.FromSeconds(2),
            TimeSpan.FromSeconds(10),
            TimeSpan.FromSeconds(30),
            null,
    };
 
    private readonly TimeSpan?[] _retryDelays;
 
    public DefaultRetryPolicy()
    {
        _retryDelays = DEFAULT_RETRY_DELAYS_IN_MILLISECONDS;
    }
 
    public DefaultRetryPolicy(TimeSpan[] retryDelays)
    {
        _retryDelays = new TimeSpan?[retryDelays.Length + 1];
 
        for (int i = 0; i < retryDelays.Length; i++)
        {
            _retryDelays[i] = retryDelays[i];
        }
    }
 
    public TimeSpan? NextRetryDelay(RetryContext retryContext)
    {
        return _retryDelays[retryContext.PreviousRetryCount];
    }
}