Python asyncio中的任务取消机制是异步编程里保障程序可控性的重要部分，它允许开发者在任务执行过程中主动终止不需要继续运行的协程任务，同时规范了取消过程中的异常处理和资源回收逻辑。
任务取消的基本触发方式
在asyncio中，任务（Task）是对协程的封装对象，取消任务需要通过Task对象提供的方法实现。最常用的触发方式是调用Task的cancel()方法，该方法会向任务发送取消请求，不会立即终止任务执行，而是标记任务为取消状态，等待任务在合适的时机响应取消信号。
以下是一个简单的任务取消示例：
import asyncio
async def worker():
try:
print("任务开始执行")
await asyncio.sleep(5)
print("任务执行完成")
except asyncio.CancelledError:
print("任务被取消")
async def main():
# 创建任务
task = asyncio.create_task(worker())
# 等待1秒后取消任务
await asyncio.sleep(1)
task.cancel()
# 等待任务处理完取消逻辑
try:
await task
except asyncio.CancelledError:
print("主函数捕获到任务取消异常")
asyncio.run(main())
取消信号的传递流程
当调用task.cancel()方法时，asyncio内部会执行以下流程：
- 首先标记Task对象的内部状态为取消待处理状态，此时任务不会立即停止运行。
- 当任务执行到下一个可中断的await点时，asyncio会在当前协程的上下文里抛出
asyncio.CancelledError异常。 - 如果任务内部的协程没有捕获这个异常，异常会向上传递，最终标记任务为取消完成状态。
- 如果任务内部的协程捕获了
asyncio.CancelledError异常，开发者可以在异常处理块中执行资源释放、状态保存等收尾操作，之后可以选择不再向上抛出异常，此时任务会被标记为正常完成状态。
可中断的await点说明
并不是所有的代码位置都能响应取消信号，只有遇到可中断的await点时，取消异常才会被抛出。常见的可中断await点包括：
- 调用
asyncio.sleep()等asyncio内置的异步函数时。 - 等待另一个Task或Future对象完成时。
- 调用支持取消的异步IO操作（如异步网络请求、文件读写）时。
如果任务中执行的是纯CPU密集型计算，没有await可中断的异步操作，那么即使调用了cancel()方法，任务也会一直运行直到计算完成，无法被取消。
任务取消后的状态变化
任务被取消后，其状态会按照以下规则变化：
| 操作场景 | 任务最终状态 | 说明 |
|---|---|---|
| 调用cancel()后，任务未捕获CancelledError | CANCELLED | 任务被标记为取消状态，await任务时会抛出CancelledError |
| 调用cancel()后，任务捕获CancelledError且未重新抛出 | FINISHED | 任务被标记为正常完成状态，返回值由异常处理逻辑决定 |
| 任务已经完成后再调用cancel() | 无变化 | cancel()方法返回False，不会触发任何取消逻辑 |
取消多个任务的批量处理
在实际开发中，经常需要批量取消多个任务，此时可以使用asyncio.gather()配合取消逻辑实现，也可以通过循环调用每个任务的cancel()方法实现。以下是一个批量取消任务的示例：
import asyncio
async def task_func(task_id):
try:
print(f"任务{task_id}开始执行")
await asyncio.sleep(10)
print(f"任务{task_id}执行完成")
except asyncio.CancelledError:
print(f"任务{task_id}被取消")
raise # 重新抛出异常，让任务状态变为CANCELLED
async def main():
tasks = []
for i in range(3):
task = asyncio.create_task(task_func(i))
tasks.append(task)
# 等待2秒后批量取消所有任务
await asyncio.sleep(2)
for task in tasks:
task.cancel()
# 等待所有任务处理完取消逻辑
await asyncio.gather(*tasks, return_exceptions=True)
print("所有任务处理完成")
asyncio.run(main())
注意事项
不要在asyncio.CancelledError的异常处理块中执行耗时的操作，否则会阻塞事件循环，影响其他任务的执行。如果任务中使用了第三方异步库，需要确认该库是否支持任务取消，部分库可能没有正确处理取消信号，导致取消失效。
另外，在Python 3.11及以上版本中，asyncio的任务取消机制做了部分优化，取消了任务组的支持，使用asyncio.TaskGroup可以更方便地管理一组任务的取消和异常处理，其底层依然遵循上述的任务取消实现逻辑。