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

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

csharp 复制代码
using System.Threading.Tasks;
using System.Threading;

public async Task Test()
{
    var cts = new CancellationTokenSource();
    var task = Task.Run(() =>
    {
        while (true)
        {
            cts.Token.ThrowIfCancellationRequested();
            Thread.Sleep(200);
        }
    });
    await Task.Delay(1000);
    cts.Cancel();
    try
    {
        await task;
    }
    catch (OperationCanceledException)
    {
        // task is cancelled
    }
    finally
    {
        cts.Dispose();
    }
}

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

csharp 复制代码
public void ThrowIfCancellationRequested()
{
    if (IsCancellationRequested)
    {
        throw new OperationCanceledExceptionEx(this);
    }
}

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

csharp 复制代码
namespace System.Threading.Tasks
{
    public sealed class CancelableTask : Task
    {
        readonly object cancelLock = new object();
        CancellationTokenSource cts;

        public bool IsCancellationRequested => cts != null && cts.IsCancellationRequested;

        private CancelableTask(Action action, CancellationToken token) : base(action, token) { }

        protected override void Dispose(bool disposing)
        {
            if (disposing)
            {
                cts?.Dispose();
            }
            base.Dispose(disposing);
        }

        public void Cancel()
        {
            lock (cancelLock)
            {
                if (!IsCompleted && !IsCancellationRequested)
                {
                    cts.Cancel();
                }
            }
        }

        public static CancelableTask StartNew(Action action)
        {
            var cts = new CancellationTokenSource();
            var task = new CancelableTask(action, cts.Token);
            task.cts = cts;
            task.Start();
            return task;
        }

        public static void Restart(ref CancelableTask task, Action action)
        {
            var oldTask = Interlocked.Exchange(ref task, null);
            oldTask?.Cancel();
            var newTask = StartNew(action);
            Interlocked.Exchange(ref task, newTask);
        }
    }
}

此类公开了 Cancel 用来取消任务,静态的 StartNew 用来创建新实例,以及静态的 Restart 用来取消上个任务并重启任务,

还公开了一个 IsCancellationRequested 属性用来获取是否请求过取消任务,相对于调用 ThrowIfCancellationRequested 抛出异常来终止任务的方式性能更好也更加友好

使用此类重新编写上面的示例就简单很多了
CancelableTask 使用示例

csharp 复制代码
using System.Threading.Tasks;

CancelableTask task;

public async Task Test()
{
    task = CancelableTask.StartNew(() =>
    {
        while (!task.IsCancellationRequested)
        {
            Thread.Sleep(200);
        }
    });
    await Task.Delay(1000);
    task.Cancel();
    await task;
}
相关推荐
LateFrames1 小时前
用 【C# + Winform + MediaPipe】 实现人脸468点识别
python·c#·.net·mediapipe
追逐时光者7 小时前
精选 4 款开源免费、美观实用的 MAUI UI 组件库,助力轻松构建美观且功能丰富的应用程序!
后端·.net
R-G-B11 小时前
【14】C#实战篇——C++动态库dll 接口函数将char* strErr字符串 传给C# ,并且在winform的MessageBox和listbox中显示。C++ string 日志传给 C#
c++·c#·strerr字符串传给c#·动态库dll传递字符串给c#·string日志传给c#·c++ string传给 c#·c++底层函数日志传给c#显示
我是唐青枫12 小时前
深入掌握 FluentMigrator:C#.NET 数据库迁移框架详解
数据库·c#·.net
tiankongdeyige13 小时前
Unity学习之C#的反射机制
学习·unity·c#
绿荫阿广14 小时前
用纯.NET开发并制作一个智能桌面机器人(六):使用.NET开发一个跨平台功能完善的小智AI客户端
c#·.net·asp.net core·maui·winui
周杰伦fans16 小时前
c#设计模式—访问者模式
c#·.net
Nᴏsᴛᴀʟɢɪᴀ念20 小时前
多线程奇幻漂流:从单核到多核质变(一)
java·开发语言·jvm·多线程
疯狂的Alex1 天前
【C#避坑实战系列文章15】C# WinForm 上位机开发:解决串口粘包+LiveCharts卡顿+InfluxDB存储(免费代码+仿真工具)
sqlite·c#·上位机·串口通信·livechars·c#硬件对接
ajassi20001 天前
开源 C# 快速开发(十六)数据库--sqlserver增删改查
windows·开源·c#