ApiClientTokenRefreshRetryPolicy.cs

Go to the documentation of this file.

using System;
using System.Threading;
 
using Microsoft.AspNetCore.SignalR.Client;
 
namespace Tgstation.Server.Client
{
    sealed class ApiClientTokenRefreshRetryPolicy : IRetryPolicy
    {
        readonly ApiClient apiClient;
 
        readonly IRetryPolicy wrappedPolicy;
 
        public ApiClientTokenRefreshRetryPolicy(ApiClient apiClient, IRetryPolicy wrappedPolicy)
        {
            this.apiClient = apiClient ?? throw new ArgumentNullException(nameof(apiClient));
            this.wrappedPolicy = wrappedPolicy ?? throw new ArgumentNullException(nameof(wrappedPolicy));
        }
 
        public TimeSpan? NextRetryDelay(RetryContext retryContext)
        {
            if (retryContext == null)
                throw new ArgumentNullException(nameof(retryContext));
 
            if (retryContext.PreviousRetryCount == 0)
                AttemptTokenRefresh();
 
            return wrappedPolicy.NextRetryDelay(retryContext);
        }
 
        async void AttemptTokenRefresh()
        {
            try
            {
                await apiClient.RefreshToken(CancellationToken.None);
            }
            catch
            {
                // intentionally ignored
            }
        }
    }
}