wabbajack/Wabbajack.Common/CircuitBreaker/WithAutoRetry.cs

using System;
using System.Threading;
using System.Threading.Tasks;

namespace Wabbajack.Common
{
    public static class CircuitBreaker
    {
        public static TimeSpan DEFAULT_DELAY = TimeSpan.FromMilliseconds(100);
        public static int DEFAULT_DELAY_MULTIPLIER = 2;
        public static int DEFAULT_RETRIES = 5;

        public static async ValueTask<TR> WithAutoRetryAsync<TR, TE>(Func<ValueTask<TR>> f, TimeSpan? delay = null, int? multipler = null, int? maxRetries = null) where TE : Exception
        {
            int retries = 0;
            delay ??= DEFAULT_DELAY;
            multipler ??= DEFAULT_DELAY_MULTIPLIER;
            maxRetries ??= DEFAULT_RETRIES;

            TOP:
            try
            {
                return await f();
            }
            catch (TE ex)
            {
                retries += 1;
                if (retries > maxRetries)
                    throw;
                Utils.Log($"(Retry {retries} of {maxRetries}), got exception {ex.Message}, waiting {delay!.Value.TotalMilliseconds}ms");
                await Task.Delay(delay.Value);
                delay = delay * multipler;
                goto TOP;
            }
        }
        
        public static async ValueTask WithAutoRetryAsync<TE>(Func<ValueTask> f, TimeSpan? delay = null, int? multipler = null, int? maxRetries = null) where TE : Exception
        {
            int retries = 0;
            delay ??= DEFAULT_DELAY;
            multipler ??= DEFAULT_DELAY_MULTIPLIER;
            maxRetries ??= DEFAULT_RETRIES;

            TOP:
            try
            {
                await f();
            }
            catch (TE ex)
            {
                retries += 1;
                if (retries > maxRetries)
                    throw;
                Utils.Log($"(Retry {retries} of {maxRetries}), got exception {ex.Message}, waiting {delay!.Value.TotalMilliseconds}ms");
                await Task.Delay(delay.Value);
                delay = delay * multipler;
                goto TOP;
            }
        }
        
        
        public static async ValueTask WithAutoRetryAllAsync(Func<ValueTask> f, TimeSpan? delay = null, int? multipler = null, int? maxRetries = null)
        {
            int retries = 0;
            delay ??= DEFAULT_DELAY;
            multipler ??= DEFAULT_DELAY_MULTIPLIER;
            maxRetries ??= DEFAULT_RETRIES;

            TOP:
            try
            {
                await f();
            }
            catch (Exception ex)
            {
                retries += 1;
                if (retries > maxRetries)
                    throw;
                Utils.Log($"(Retry {retries} of {maxRetries}), got exception {ex.Message}, waiting {delay!.Value.TotalMilliseconds}ms");
                await Task.Delay(delay.Value);
                delay = delay * multipler;
                goto TOP;
            }
        }
        public static async ValueTask<T> WithAutoRetryAllAsync<T>(Func<ValueTask<T>> f, TimeSpan? delay = null, int? multipler = null, int? maxRetries = null)
        {
            int retries = 0;
            delay ??= DEFAULT_DELAY;
            multipler ??= DEFAULT_DELAY_MULTIPLIER;
            maxRetries ??= DEFAULT_RETRIES;

            TOP:
            try
            {
                return await f();
            }
            catch (Exception ex)
            {
                retries += 1;
                if (retries > maxRetries)
                    throw;
                Utils.Log($"(Retry {retries} of {maxRetries}), got exception {ex.Message}, waiting {delay!.Value.TotalMilliseconds}ms");
                await Task.Delay(delay.Value);
                delay = delay * multipler;
                goto TOP;
            }
        }
        
        public static void WithAutoRetry<TE>(Action f, TimeSpan? delay = null, int? multipler = null, int? maxRetries = null) where TE : Exception
        {
            int retries = 0;
            delay ??= DEFAULT_DELAY;
            multipler ??= DEFAULT_DELAY_MULTIPLIER;
            maxRetries ??= DEFAULT_RETRIES;

            TOP:
            try
            {
                f();
            }
            catch (TE ex)
            {
                retries += 1;
                if (retries > maxRetries)
                    throw;
                Utils.Log($"(Retry {retries} of {maxRetries}), got exception {ex.Message}, waiting {delay!.Value.TotalMilliseconds}ms");
                Thread.Sleep(delay.Value);
                delay = delay * multipler;
                goto TOP;
            }
        }

    }
}