给定一个和每个任务组文档中taskgroup
正在运行的任务数,如果任何任务出现错误,则组中的其余任务将被取消。
如果其中一些任务在取消时需要执行清理,那么如何在任务中检测被取消的任务?
希望在任务中引发一些异常,但事实并非如此:
脚本.py:
import asyncio
class TerminateTaskGroup(Exception):
"""Exception raised to terminate a task group."""
async def task_that_needs_to_cleanup_on_cancellation():
try:
await asyncio.sleep(10)
except Exception:
print('exception caught, performing cleanup...')
async def err_producing_task():
await asyncio.sleep(1)
raise TerminateTaskGroup()
async def main():
try:
async with asyncio.TaskGroup() as tg:
tg.create_task(task_that_needs_to_cleanup_on_cancellation())
tg.create_task(err_producing_task())
except* TerminateTaskGroup:
print('main() termination handled')
asyncio.run(main())
执行后,我们可以看到没有引发任何异常task_that_needs_to_cleanup_on_cancellation()
:
$ python3 script.py
main() termination handled