C# Task 取消执行的简单封装

我让DeepSeek帮我写了一段使用 CancellationTokenSource 取消任务的简单示例如下：
取消任务的简单示例

1. using System.Threading.Tasks;
2. using System.Threading;
4. public async Task Test()
5. {
6.     var cts = new CancellationTokenSource();
7.     var task = Task.Run(() =>
8.     {
9.         while (true)
10.         {
11.             cts.Token.ThrowIfCancellationRequested();
12.             Thread.Sleep(200);
13.         }
14.     });
15.     await Task.Delay(1000);
16.     cts.Cancel();
17.     try
18.     {
19.         await task;
20.     }
21.     catch (OperationCanceledException)
22.     {
23.         // task is cancelled
24.     }
25.     finally
26.     {
27.         cts.Dispose();
28.     }
29. }

复制代码

相对于不可取消的普通任务多了很多代码，并且终止任务的方式 ThrowIfCancellationRequested 内部是通过抛出异常实现的
ThrowIfCancellationRequested 内部代码

1. public void ThrowIfCancellationRequested()
2. {
3.     if (IsCancellationRequested)
4.     {
5.         throw new OperationCanceledExceptionEx(this);
6.     }
7. }

复制代码

于是我就想自己封装一个简单的包装类，代码如下：
CancelableTask

1. namespace System.Threading.Tasks
2. {
3.     public sealed class CancelableTask : Task
4.     {
5.         readonly object cancelLock = new object();
6.         CancellationTokenSource cts;
8.         public bool IsCancellationRequested => cts != null && cts.IsCancellationRequested;
10.         private CancelableTask(Action action, CancellationToken token) : base(action, token) { }
12.         protected override void Dispose(bool disposing)
13.         {
14.             if (disposing)
15.             {
16.                 cts?.Dispose();
17.             }
18.             base.Dispose(disposing);
19.         }
21.         public void Cancel()
22.         {
23.             lock (cancelLock)
24.             {
25.                 if (!IsCompleted && !IsCancellationRequested)
26.                 {
27.                     cts.Cancel();
28.                 }
29.             }
30.         }
32.         public static CancelableTask StartNew(Action action)
33.         {
34.             var cts = new CancellationTokenSource();
35.             var task = new CancelableTask(action, cts.Token);
36.             task.cts = cts;
37.             task.Start();
38.             return task;
39.         }
41.         public static void Restart(ref CancelableTask task, Action action)
42.         {
43.             var oldTask = Interlocked.Exchange(ref task, null);
44.             oldTask?.Cancel();
45.             var newTask = StartNew(action);
46.             Interlocked.Exchange(ref task, newTask);
47.         }
48.     }
49. }

复制代码

此类公开了 Cancel 用来取消任务，静态的 StartNew 用来创建新实例，以及静态的 Restart 用来取消上个任务并重启任务，
还公开了一个 IsCancellationRequested 属性用来获取是否请求过取消任务，相对于调用 ThrowIfCancellationRequested 抛出异常来终止任务的方式性能更好也更加友好
使用此类重新编写上面的示例就简单很多了
CancelableTask 使用示例

1. using System.Threading.Tasks;
3. CancelableTask task;
5. public async Task Test()
6. {
7.     task = CancelableTask.StartNew(() =>
8.     {
9.         while (!task.IsCancellationRequested)
10.         {
11.             Thread.Sleep(200);
12.         }
13.     });
14.     await Task.Delay(1000);
15.     task.Cancel();
16.     await task;
17. }

复制代码

来源：程序园用户自行投稿发布，如果侵权，请联系站长删除
免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！