Task.WaitAll 抛出 OperationCanceledException

Task.WaitAll throws OperationCanceledException

我有 运行 个具有相同 CancellationTokenSource 任务的列表。

我希望当前线程等到所有任务完成直到任务被取消。

Task.WaitAll(tasks.ToArray(), searchCencellationTokenSource.Token);
System.Console.WriteLine("Done !");

即使当前线程处于等待状态,任务也可能被另一个任务取消。这是正常行为。

但是,当当前线程处于等待状态并且另一个任务取消任务时,WaitAll 抛出 CancellationTokenSource 消息:"The operation was canceled.".

我知道它被取消了,我是故意的。我只是希望它在任务取消或完成后继续执行下一个代码,而不抛出异常。

我知道我可以用 try & catch 包装这段代码,但抛出异常是繁重的操作,我不希望它发生在像这样的正常行为上。

这个阻塞机制可以改写为:

Task.WhenAll(taskA, taskB, taskC).Wait()

这为您提供了一个我们可以等待但也可以取消的任务。因此,要忽略取消异常,您可以执行以下操作:

Task.WhenAll(taskA, taskB, taskC).ContinueWith(t => { }, TaskContinuationOptions.OnlyOnCanceled).Wait();

不会抛出 OperationCancelledException

然后可以将其包装到扩展方法中,如下所示:

public static class TaskExtensions
{
    public static Task IgnoreCancellation(this Task task)
    {
        return task.ContinueWith(t => { }, TaskContinuationOptions.OnlyOnCanceled);
    }
}

这将允许您再次编写以下内容而不会遇到 OperationCancelledException:

Task.WhenAll(taskA, taskB, taskC).IgnoreCancellation().Wait();

这是一个显示该方法有效的测试夹具:

public class IgnoreTaskCancellation
{
    [Fact]
    public void ShouldThrowAnAggregateException()
    {
        CancellationTokenSource cts = new CancellationTokenSource(10);

        Task taskA = Task.Delay(20, cts.Token);
        Task taskB = Task.Delay(20, cts.Token);
        Task taskC = Task.Delay(20, cts.Token);

        Assert.Throws<AggregateException>(() => Task.WhenAll(taskA, taskB, taskC).Wait());
    }

    [Fact]
    public void ShouldNotThrowAnException()
    {
        CancellationTokenSource cts = new CancellationTokenSource(10);

        Task taskA = Task.Delay(20, cts.Token);
        Task taskB = Task.Delay(20, cts.Token);
        Task taskC = Task.Delay(20, cts.Token);

        Task.WhenAll(taskA, taskB, taskC).ContinueWith(t => { }, TaskContinuationOptions.OnlyOnCanceled).Wait();
    }

    [Fact]
    public void ShouldNotThrowAnExceptionUsingIgnore()
    {
        CancellationTokenSource cts = new CancellationTokenSource(10);

        Task taskA = Task.Delay(20, cts.Token);
        Task taskB = Task.Delay(20, cts.Token);
        Task taskC = Task.Delay(20, cts.Token);

        Task.WhenAll(taskA, taskB, taskC).IgnoreCancellation().Wait();
    }
}

希望对您有所帮助。