Python中取消异步任务需调用Task.cancel()触发协作式取消,协程必须在await点响应CancelledError并重抛,否则取消无效;纯CPU计算或阻塞IO需转为异步执行以支持取消。

Python中取消异步任务,核心在于协程的可取消性设计与asyncio.Task的生命周期控制。协程本身不支持“中途强制终止”,但通过Task.cancel()触发取消信号,配合await点上的异常传播(CancelledError),才能实现安全、协作式的取消。
协程为何不能直接“杀掉”?
Python的asyncio基于事件循环和协作式多任务,没有抢占式调度。协程只有在await表达式处才会让出控制权——这意味着取消请求不会立即生效,必须等到协程下一次挂起时,事件循环才能注入CancelledError。若协程长时间运行且无await(比如纯CPU密集计算或死循环),它将完全忽略取消信号。
- 取消是协作式(cooperative)的,不是强制式(preemptive)
-
Task.cancel()只是设置取消标记,并调度异常,不中断当前执行 - 协程需在
await点响应CancelledError,否则取消无效
如何正确取消一个正在运行的Task?
使用asyncio.create_task()启动任务后,调用其cancel()方法即可发起取消。关键是要等待任务真正结束(处理完异常或退出),通常用asyncio.wait_for(task, timeout=None)或await task捕获结果。
- 调用
task.cancel()后,task.done()仍为False,直到协程处理完取消异常 - 推荐用
try/except CancelledError在协程内做清理(如关闭连接、释放资源) - 避免忽略
CancelledError:不捕获、或捕获后未重新抛出,会导致取消失败
常见取消失效场景与修复方式
以下情况容易导致取消“看起来没用”:
立即学习“Python免费学习笔记(深入)”;
-
协程中吞掉了CancelledError:例如
except Exception:兜底捕获,应显式重抛或使用except BaseException:并谨慎处理 -
阻塞式IO或CPU密集操作:改用
asyncio.to_thread()(Python 3.9+)或loop.run_in_executor()包装,确保可被取消 -
未await取消后的Task:仅调
cancel()不await,任务可能仍在后台运行;建议await asyncio.shield(task)保护关键逻辑,或用asyncio.wait([task], return_when=asyncio.FIRST_COMPLETED)统一管理
实用取消模式示例
一个健壮的可取消协程通常结构如下:
import asyncioasync def fetch_with_timeout(url, timeout=5): try: async with asyncio.timeout(timeout):
模拟异步HTTP请求
await asyncio.sleep(3) return f"Data from {url}" except asyncio.TimeoutError: print("Timeout hit") raise except asyncio.CancelledError: print("Fetch was cancelled") # 清理资源(如关闭client session) raise # 必须重新抛出,否则取消被吞掉async def main(): task = asyncio.create_task(fetch_with_timeout("https://www.php.cn/link/b05edd78c294dcf6d960190bf5bde635")) await asyncio.sleep(1) task.cancel() # 发起取消 try: await task # 等待完成并捕获异常 except asyncio.CancelledError: print("Task cleanly cancelled")
asyncio.run(main())
不复杂但容易忽略。










