Question

Contrived example，但假设我在异步方法中有以下内容：

var cts = new CancellationTokenSource();
cts.CancelAfter(2000);
cts.Token.Register(Callback);
SomethingThatMightThrow();
await Task.Delay(10000, cts.Token);

只要在几秒钟后调用Callback，就可以正常工作。但是，我想在Task.Delay之后处理注册，所以假设我做了以下修改：

var cts = new CancellationTokenSource();
cts.CancelAfter(2000);
using (cts.Token.Register(Callback))
{
    SomethingThatMightThrow();
    await Task.Delay(10000, cts.Token);
}

在这种情况下，Callback not 被调用，可能是因为await Task.Delay...中的异常导致注册在被调用之前被处理掉。

确保取消时Callback被取消并且注册始终被处置的最佳方式是什么？

我确实想到了以下内容，但我不确定它有多强大：

var cts = new CancellationTokenSource();
cts.CancelAfter(2000);

var ctr = cts.Token.Register(Callback);
try
{
    SomethingThatMightThrow();
    await Task.Delay(10000, cts.Token);
}
finally
{
    if (!cts.Token.IsCancellationRequested)
        ctr.Dispose();
}

Answer 1

CancellationToken.Register通常用于将新的CancellationToken系统与使用其他类型的通知进行取消的旧系统进行互操作。它不能用作通用＆＃34;取消回调＆＃34;。

如果您希望在取消操作时以某种方式响应，那么您只需捕获相应的异常：

using (var cts = new CancellationTokenSource())
{
  cts.CancelAfter(2000);
  SomethingThatMightThrow();
  try
  {
    await Task.Delay(10000, cts.Token);
  }
  catch (OperationCanceledException)
  {
    Callback();
  }
}

Answer 2

并不是说该方法可能不会抛出异常，因为它可能不会像我想的那样快速地抛出它。我注意到了一些例如，Azure SDK异步方法可能需要相当长的时间回应在令牌上发出信号的取消

根据您的评论，您可以选择创建自己的计时器，以根据您的标准明确指定方法“运行时间过长”。为此，您可以使用Task.WhenAny：

using (var cts = new CancellationTokenSource())
{
    try
    {
        var cancellationDelayTask = Task.Delay(2000, cts.Token);
        var taskThatMightThrow = SomethingThatMightThrowAsync(cts.Token);

        if ((await Task.WhenAny(taskThatMightThrow, cancellationDelayTask)) 
             == cancellationDelayTask)
        {
            // Task.Delay "timeout" finished first.
        }
    }
    catch (OperationCanceledException)
    {
        Callback();
    }
}

处理CancellationTokenRegistration

2 个答案: