在 C# 中使用 Polly，我可以等到时间跨度过去或任务终止后再重试吗？

Question

我正在尝试在 C# 中使用Polly包。 我想运行一些代码，然后，如果失败，请等待并重试。 目前我的循环看起来类似于：

var successful = false
while (!successful){
    // Try to perform operation.
    successful = TryToDoStuff()
    if (!successful){
        // Wait, then retry.
        await Task.WhenAny(
            taskCompletionSource1.Task,
            taskCompletionSource1.Task,
            Task.Delay(TimeSpan.FromSeconds(10)));
    }
}

即：等待 10 秒或直到这些任务完成源之一收到信号并终止。 然后重试。

我想做的是这样的事情（Polly API 不支持）：

Policy
    .Handle<RetryException>()
    .WaitAndRetryForever(
        Task.WhenAny(
            taskCompletionSource1.Task,
            taskCompletionSource1.Task,
            Task.Delay(TimeSpan.FromSeconds(10))))
    .Execute(TryToDoStuff); // Method TryToDoStuff will throw RetryException if it fails 

有可能用 Polly 做这样的事情吗？ 我可以等待 TimeSpan 以外的任何东西吗？

关于我在上面的例子中等待的两个任务：一个任务是一个取消，指示整个事情应该关闭。 另一个是“唤醒连接尝试”任务，其终止指示“此对象的状态已更改；尝试再次调用它”。 在这两种情况下，我都希望我的循环立即继续下一次迭代，而不是等待超时结束。

目前等待超时还不错，因为它只有 10 秒，但如果我将其更改为指数退避，那么突然超时可能会很长。 因此，我希望中断超时并直接进行下一次迭代。

注意：我的重试循环遵循 async-await 模式并不是必须的。 如果等待部分是同步和阻塞的就可以了。 我只想能够使用任务完成源取消等待。

Answer 1

所有 Polly 策略和执行都可以响应CancellationToken以发出取消信号。

如果我理解正确，有两个要求：

1. 如果发出取消信号，则立即重试（不再延迟）
2. 如果发生RetryException ，则重试延时（10 秒或指数）

您可以使用以下各项的 Polly 策略来表达这一点：

var retryImmediatelyOnCancellation = Policy
    .Handle<OperationCanceledException>()
    .RetryForever();

var retryWithDelay = Policy
    .Handle<RetryException>()
    .WaitAndRetryForever(/* specify your desired delay or delay sequence for retries */);

然后以嵌套方式执行两个重试策略。

就像是：

retryImmediatelyOnCancellation.Execute(() => 
{
    CancellationTokenSource externalCancellation = ... // Get the CancellationTokenSource signalling external cancellation.
    CancellationTokenSource wakeUpConnectionAttemptCancellation = ... // Get the CancellationTokenSource signalling "wake up connection attempt".
    CancellationTokenSource combinedCancellationSource = CancellationTokenSource.CreateLinkedTokenSource(externalCancellation.Token, wakeUpConnectionAttemptCancellation.Token);

    retryWithDelay.Execute(ct => TryDoStuff(), combinedCancellationSource.Token);
});

如果发出CancellationToken信号，则retryWithDelay策略造成的延迟会立即CancellationToken 。

---

CancellationToken是比TaskCompletionSource更自然的取消信号。 但是，如果您需要（无论出于何种原因）坚持使用TaskCompletionSource作为这些事件的信号，您可以将其转换为取消CancellationTokenSource ，例如：

taskCompletionSource.Task.ContinueWith(t => cancellationTokenSource.Cancel());

---

请注意， TaskCompletionSource和CancellationToken都是一次性的：一旦完成或取消，它们就无法重置（未完成或取消）。 上面的代码示例在CancellationTokenSource重试循环内移动了CancellationTokenSource的创建，以便您在每个取消信号后获得新的CancellationTokenSource 。

---

如果您切换到异步，所有这些也适用于异步 Polly 策略。

Answer 2

我能想出的最佳解决方案是：

var successful = false
while (!successful){

    // Create cancellation token that gets cancelled when one of the tasks terminates.
    var cts = new CancellationTokenSource();
    _ = Task.Run(async () =>
    {
        await Task.WhenAny(
            taskCompletionSource1.Task,
            taskCompletionSource1.Task);
        cts.Cancel();
    });

    // Try to perform operation.
    Policy
        .Handle<RetryException>()
        .WaitAndRetryForever(
            TimeSpan.FromSeconds(10))
        .Execute(
            // Method TryToDoStuff will throw RetryException if it fails 
            ct => TryToDoStuff(), 
            // Pass in cancellation token.
            cts.Token);
}

这似乎有效。 但如果没有波莉，我最终可能会这样做。