wabbajack/Wabbajack.Common/CircuitBreaker.cs

using System;
using System.Threading;
using System.Threading.Tasks;
using Microsoft.Extensions.Logging;

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>(ILogger logger, 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;
                logger.LogWarning("(Retry {retries} of {maxRetries}), got exception {message}, waiting {wait}ms", 
                    retries, maxRetries, ex.Message, delay!.Value.TotalMilliseconds);
                await Task.Delay(delay.Value);
                delay = delay * multipler;
                goto TOP;
            }
        }
        
        public static async ValueTask WithAutoRetryAsync<TE>(ILogger logger, 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;
                logger.LogWarning("(Retry {retries} of {maxRetries}), got exception {message}, waiting {wait}ms", 
                    retries, maxRetries, ex.Message, delay!.Value.TotalMilliseconds);
                await Task.Delay(delay.Value);
                delay = delay * multipler;
                goto TOP;
            }
        }
        
        
        public static async ValueTask WithAutoRetryAllAsync(ILogger logger, 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;
                logger.LogWarning("(Retry {retries} of {maxRetries}), got exception {message}, waiting {wait}ms", 
                    retries, maxRetries, ex.Message, delay!.Value.TotalMilliseconds);
                await Task.Delay(delay.Value);
                delay = delay * multipler;
                goto TOP;
            }
        }
        public static async ValueTask<T> WithAutoRetryAllAsync<T>(ILogger logger, 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;
                await Task.Delay(delay.Value);
                logger.LogWarning("(Retry {retries} of {maxRetries}), got exception {message}, waiting {wait}ms", 
                    retries, maxRetries, ex.Message, delay!.Value.TotalMilliseconds);
                delay = delay * multipler;
                goto TOP;
            }
        }
        
        public static void WithAutoRetry<TE>(ILogger logger, 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;
                logger.LogWarning("(Retry {retries} of {maxRetries}), got exception {message}, waiting {wait}ms", 
                    retries, maxRetries, ex.Message, delay!.Value.TotalMilliseconds);
                Thread.Sleep(delay.Value);
                delay = delay * multipler;
                goto TOP;
            }
        }

    }
}