Retry.cs
2.96 KB
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);
}
}
}
}
}