如何使用任务并行库(TPL)实现重试逻辑

时间:2011-05-22 18:54:53

标签: c# .net task-parallel-library

  

可能重复:
  Retry a task multiple times based on user input in case of an exception in task

我正在寻找一种在TPL中实现重试逻辑的方法。我想有一个泛型函数/类,它将能够返回一个将执行给定操作的Task,并且在异常的情况下将重试该任务,直到给定的重试计数。我尝试使用ContinueWith进行播放,并在异常的情况下让回调创建一个新任务,但它似乎只适用于固定的重试次数。有什么建议吗?

    private static void Main()
    {
        Task<int> taskWithRetry = CreateTaskWithRetry(DoSometing, 10);
        taskWithRetry.Start();
        // ...

    }

    private static int DoSometing()
    {
        throw new NotImplementedException();
    }

    private static Task<T> CreateTaskWithRetry<T>(Func<T> action, int retryCount)
    {

    }

2 个答案:

答案 0 :(得分:9)

有什么理由与TPL做一些特别的事吗?为什么不只为Func<T>本身做一个包装?

public static Func<T> Retry(Func<T> original, int retryCount)
{
    return () =>
    {
        while (true)
        {
            try
            {
                return original();
            }
            catch (Exception e)
            {
                if (retryCount == 0)
                {
                    throw;
                }
                // TODO: Logging
                retryCount--;
            }
        }
    };
}

请注意,您可能需要添加ShouldRetry(Exception)方法,以便在不重试的情况下中止某些例外(例如取消)。

答案 1 :(得分:0)

private static Task<T> CreateTaskWithRetry<T>(Func<T> action, int retryCount)
{
    Func<T> retryAction = () =>
    {
        int attemptNumber = 0;
        do
        {
            try
            {
                attemptNumber++;
                return action();
            }
            catch (Exception exception) // use your the exception that you need
            {
                // log error if needed
                if (attemptNumber == retryCount)
                    throw;
            }
        }
        while (attemptNumber < retryCount);

        return default(T);
    };

    return new Task<T>(retryAction);
}