Spotify.NET/SpotifyAPI.Web/RetryHandlers/SimpleRetryHandler.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Net;
using System.Threading;
using System.Threading.Tasks;
using SpotifyAPI.Web.Http;

namespace SpotifyAPI.Web
{
  public class SimpleRetryHandler : IRetryHandler
  {
    private readonly Func<TimeSpan, Task> _sleep;

    /// <summary>
    ///     Specifies after how many miliseconds should a failed request be retried.
    /// </summary>
    public TimeSpan RetryAfter { get; set; }

    /// <summary>
    ///     Maximum number of tries for one failed request.
    /// </summary>
    public int RetryTimes { get; set; }

    /// <summary>
    ///     Whether a failure of type "Too Many Requests" should use up one of the allocated retry attempts.
    /// </summary>
    public bool TooManyRequestsConsumesARetry { get; set; }

    /// <summary>
    ///     Error codes that will trigger auto-retry
    /// </summary>
    public IEnumerable<HttpStatusCode> RetryErrorCodes { get; set; }

    /// <summary>
    ///   A simple retry handler which retries a request based on status codes with a fixed sleep interval.
    ///   It also supports Retry-After headers sent by spotify. The execution will be delayed by the amount in
    ///   the Retry-After header
    /// </summary>
    /// <returns></returns>
    public SimpleRetryHandler() : this(Task.Delay) { }
    public SimpleRetryHandler(Func<TimeSpan, Task> sleep)
    {
      _sleep = sleep;
      RetryAfter = TimeSpan.FromMilliseconds(50);
      RetryTimes = 10;
      TooManyRequestsConsumesARetry = false;
      RetryErrorCodes = new[] {
        HttpStatusCode.InternalServerError,
        HttpStatusCode.BadGateway,
        HttpStatusCode.ServiceUnavailable
      };
    }

    private static TimeSpan? ParseTooManyRetries(IResponse response)
    {
      if (response.StatusCode != (HttpStatusCode)429)
      {
        return null;
      }
      if (
        (response.Headers.ContainsKey("Retry-After") && int.TryParse(response.Headers["Retry-After"], out int secondsToWait))
        || (response.Headers.ContainsKey("retry-after") && int.TryParse(response.Headers["retry-after"], out secondsToWait)))
      {
        return TimeSpan.FromSeconds(secondsToWait);
      }

      throw new APIException("429 received, but unable to parse Retry-After Header. This should not happen!");
    }

    public Task<IResponse> HandleRetry(IRequest request, IResponse response, IRetryHandler.RetryFunc retry, CancellationToken cancel = default)
    {
      Ensure.ArgumentNotNull(response, nameof(response));
      Ensure.ArgumentNotNull(retry, nameof(retry));

      return HandleRetryInternally(request, response, retry, RetryTimes, cancel);
    }

    private async Task<IResponse> HandleRetryInternally(
      IRequest request,
      IResponse response,
      IRetryHandler.RetryFunc retry,
      int triesLeft,
      CancellationToken cancel)
    {
      cancel.ThrowIfCancellationRequested();

      var secondsToWait = ParseTooManyRetries(response);
      if (secondsToWait != null && (!TooManyRequestsConsumesARetry || triesLeft > 0))
      {
        await _sleep(secondsToWait.Value).ConfigureAwait(false);
        response = await retry(request, cancel).ConfigureAwait(false);
        var newTriesLeft = TooManyRequestsConsumesARetry ? triesLeft - 1 : triesLeft;
        return await HandleRetryInternally(request, response, retry, newTriesLeft, cancel).ConfigureAwait(false);
      }

      while (RetryErrorCodes.Contains(response.StatusCode) && triesLeft > 0)
      {
        await _sleep(RetryAfter).ConfigureAwait(false);
        response = await retry(request, cancel).ConfigureAwait(false);
        return await HandleRetryInternally(request, response, retry, triesLeft - 1, cancel).ConfigureAwait(false);
      }

      return response;
    }
  }
}