给定超时后，CancellationTokenSource不会取消任务

Question

我正在学习TAP，并且我喜欢使用单元测试来探索TPL数据流。 我有以下我无法理解的内容：

var cts = new CancellationTokenSource(500);
var tcs = new TaskCompletionSource<bool>(cts.Token);
var agent = new ActionBlock<FakeMessage>( async evt =>
{
    await Task.Delay(5000);
    tcs.SetResult(true);
});

agent.Post(new FakeMessage());
try
{
    var result = await tcs.Task;
    Assert.Fail();
}
catch (OperationCanceledException ex)
{
    Assert.IsTrue(true);
}
catch (Exception e)
{
    Assert.Fail();
}

我期望它提高第一行中定义的超时并捕获OperationCanceledException ，但是在await tcs.Task之后，我总是以Assert.Fail await tcs.Task 。 有人可以解释一下我的假设进行得不好吗？

Answer 1

TaskCompletionSource不接受CancellationToken 。

它确实接受Object状态，并且您可以从技术上将CancellationToken传递给它，但是它不会做任何事情，尤其是不要取消TaskCompletionSource 。

如果要取消TaskCompletionSource ，则可以通过简单的超时来完成：

Task.Delay(500).ContinueWith(t => tcs.SetCancelled());

您还可以创建一个TaskCompletionSource ，它接受CancellationToken并在取消TaskCompletionSource时TaskCompletionSource取消：

class TaskCompletionSourceWithCancellation<T> : TaskCompletionSource<T>
{
    public CancellationToken CancellationToken { get; }

    public TaskCompletionSourceWithCancellation(CancellationToken cancellationToken)
    {
        CancellationToken = cancellationToken;
        var cancellationTokenRegistration =
            cancellationToken.Register(
                _ => ((TaskCompletionSourceWithCancellation<TResult>)_).TrySetCanceled(),
                this);
        Task.ContinueWith(_ => cancellationTokenRegistration.Dispose());
    }
}