问题现象描述
在使用Python的asyncio.wait()方法时，开发者经常会遇到任务取消异常(asyncio.CancelledError)。这种情况通常发生在以下几种场景：
- 当使用
wait()的timeout参数时，超时后未完成的任务会被自动取消 - 当事件循环关闭时，所有未完成的任务都会被取消
- 显式调用
task.cancel()方法后
问题根本原因
asyncio.wait()方法的默认行为会在超时或显式取消时，向未完成的任务发送取消请求。这些任务在下一个await点会抛出CancelledError异常。如果不妥善处理这些异常，会导致：
- 程序意外终止
- 资源泄露（如未关闭的文件或网络连接）
- 任务状态不一致
解决方案
方案1：捕获CancelledError异常
async def my_task():
try:
# 任务逻辑代码
await some_async_operation()
except asyncio.CancelledError:
# 清理资源
await cleanup_resources()
raise # 重新抛出异常方案2：使用return_when参数
通过设置return_when=asyncio.FIRST_COMPLETED可以避免自动取消其他任务：
done, pending = await asyncio.wait(tasks, return_when=asyncio.FIRST_COMPLETED)方案3：使用asyncio.shield保护关键任务
对于不希望被取消的关键任务，可以使用asyncio.shield()：
protected_task = asyncio.shield(important_task())
done, pending = await asyncio.wait([protected_task, other_task], timeout=5)最佳实践
结合上述方案，我们推荐以下最佳实践：
- 总是为任务实现清理逻辑
- 根据业务需求选择合适的
return_when策略 - 对关键任务使用
shield保护 - 在任务取消时记录日志以便调试
性能考量
处理任务取消时需要注意：
- 异常处理会增加少量性能开销
shield会阻止任务取消，可能导致资源占用时间延长- 过多的未处理取消异常会影响事件循环性能
调试技巧
当遇到取消异常问题时，可以使用以下方法调试：
- 启用
asyncio的调试模式：asyncio.get_event_loop().set_debug(True) - 检查任务创建堆栈：
task.get_stack() - 使用
asyncio.all_tasks()检查所有运行中的任务
总结
正确处理asyncio.wait()的任务取消异常是编写健壮异步应用的关键。通过理解取消机制、合理处理异常并采用最佳实践，可以显著提高异步代码的可靠性和可维护性。