C# 取消一个不带CancellationToken的任务？

在异步函数中，一般使用CancellationToken来控制函数的执行。这个Token需要作为参数传递到异步函数中：

public staic Task<T> DoAsync(CancellationToken token)
{...
}

那么如果一个异步函数没有这个Token参数，如何取消呢?

之前看到一个博客：How to Cancel a Non-Cancellable Task in C#

1.构建取消异步函数的扩展方法

先上代码：

    public static class TaskExtensions{public static async Task<T> WithCancellation<T> (this Task<T> task,CancellationToken cancellationToken){var tcs=new TaskCompletionSource<object>(TaskCreationOptions.RunContinuationsAsynchronously);using (cancellationToken.Register(state =>{((TaskCompletionSource<object>)state!).TrySetResult(null!);}, tcs)){var resultTask = await Task.WhenAny(task, tcs.Task);if(resultTask==tcs.Task){throw new OperationCanceledException(cancellationToken);}return await task;}}}

显然这是一个扩展方法，旨在为Task<T>类型扩展一个名叫WithCancellation函数。这个函数会有一个CancellationToken,但这个Token不是传递给任务参数task(也就是我们要取消的函数)的，而是用于外部控制。

代码中首先构建了一个异步任务，利用TaskCompletionSource<T>，可以构建一个用于控制原任务的异步任务，这里的TaskCompletionSource不多解释了，可以参考博客：here。然后通过传入的Token注册一个回调函数，回调函数传入的参数就是刚刚创立的TaskCompletionSource<T>对象，回调函数会调用成员函数TrySetResult()给任务赋值。而回调函数的执行则是在Token被取消时触发。

2. 测试

然后我们使用这个扩展方法构建实例：

 internal class CancelTaskWithoutCancellationToken{private static readonly CancellationTokenSource cancellationTokenSource = new CancellationTokenSource();public static async Task Test(){try{var result = await Task.Run(async () =>{await Task.Delay(TimeSpan.FromSeconds(5));Console.WriteLine("操作完成");await Task.Delay(300);Console.WriteLine("还继续吗？");return 7;}).WithCancellation(cancellationTokenSource.Token);Console.WriteLine(result.ToString());}catch (Exception ex){Console.WriteLine(ex.Message);Console.WriteLine("任务取消了");}finally{cancellationTokenSource.Dispose();}}public static void Cancel(){cancellationTokenSource?.Cancel();}}

测试代码：

var t1=CancelTaskWithoutCancellationToken.Test();
var t2 = Task.Run(() =>
{Thread.Sleep(1000);CancelTaskWithoutCancellationToken.Cancel();
});
await Task.WhenAll(t1, t2);

运行结果如下：

好像结果很符合预期。

假设测试代码后面还有一些任务要运行，也就是主线程没那么快结束呢？让我们在测试代码后面加一行：

var t1=CancelTaskWithoutCancellationToken.Test();
var t2 = Task.Run(() =>
{Thread.Sleep(1000);CancelTaskWithoutCancellationToken.Cancel();
});
await Task.WhenAll(t1, t2);
Console.WriteLine("----------------------");
Console.ReadLine();

运行结果：