Blame view

netcore/src/Infrastructure/NCC/FriendlyException/Retry.cs 2.96 KB
de2bd2f9   “wangming”   项目初始化
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
  using System;
  using System.Diagnostics;
  using System.Linq;
  using System.Threading.Tasks;
  
  namespace NCC.FriendlyException
  {
      /// <summary>
      /// 重试类
      /// </summary>
      public sealed class Retry
      {
          /// <summary>
          /// 重试有异常的方法,还可以指定特定异常
          /// </summary>
          /// <param name="action"></param>
          /// <param name="numRetries">重试次数</param>
          /// <param name="retryTimeout">重试间隔时间</param>
          /// <param name="finalThrow">是否最终抛异常</param>
          /// <param name="exceptionTypes">异常类型,可多个</param>
          public static void Invoke(Action action, int numRetries, int retryTimeout = 1000, bool finalThrow = true, Type[] exceptionTypes = default)
          {
              if (action == null) throw new ArgumentNullException(nameof(action));
  
              Invoke(async () =>
              {
                  action();
                  await Task.CompletedTask;
              }, numRetries, retryTimeout, finalThrow, exceptionTypes).GetAwaiter().GetResult();
          }
  
          /// <summary>
          /// 重试有异常的方法,还可以指定特定异常
          /// </summary>
          /// <param name="action"></param>
          /// <param name="numRetries">重试次数</param>
          /// <param name="retryTimeout">重试间隔时间</param>
          /// <param name="finalThrow">是否最终抛异常</param>
          /// <param name="exceptionTypes">异常类型,可多个</param>
          public static async Task Invoke(Func<Task> action, int numRetries, int retryTimeout = 1000, bool finalThrow = true, Type[] exceptionTypes = default)
          {
              if (action == null) throw new ArgumentNullException(nameof(action));
  
              // 不断重试
              while (true)
              {
                  try
                  {
                      await action();
                      break;
                  }
                  catch (Exception ex)
                  {
                      // 如果可重试次数小于或等于0,则终止重试
                      if (--numRetries < 0)
                      {
                          if (finalThrow) throw;
                          else return;
                      }
  
                      // 如果填写了 exceptionTypes 且异常类型不在 exceptionTypes 之内,则终止重试
                      if (exceptionTypes != null && exceptionTypes.Length > 0 && !exceptionTypes.Any(u => u.IsAssignableFrom(ex.GetType())))
                      {
                          if (finalThrow) throw;
                          else return;
                      }
  
                      if (Debugger.IsAttached)
                      {
                          Console.WriteLine($"You can retry {numRetries} more times.");
                      }
  
                      // 如果可重试异常数大于 0,则间隔指定时间后继续执行
                      if (retryTimeout > 0) await Task.Delay(retryTimeout);
                  }
              }
          }
      }
  }