区分屏蔽任务取消和当前TA

Question

阅读时： https://docs.python.org/3/library/asyncio-task.html#asyncio.Task.cancel 捕捉cancelederror似乎有两个目的。在

一是可能会阻止任务被取消。在

另一个是确定有什么东西取消了你正在等待的任务。 如何区别？在

async def cancel_me():
    try:
        await asyncio.sleep(3600)
    except asyncio.CancelledError:
        raise
    finally:
        print('cancel_me(): after sleep')

async def main():
    task = asyncio.create_task(cancel_me())
    await asyncio.sleep(1)
    task.cancel()
    try:
        await task
    except asyncio.CancelledError:
        # HERE: How do I know if `task` has been cancelled, or I AM being cancelled?
        print("main(): cancel_me is cancelled now")

Answer 1

上下文

首先，让我们考虑更广泛的背景：

caller() > your_coro() > callee()

您可以控制协同程序，但不能控制调用者，只能部分控制被调用者。在

默认情况下，“向上传播”和“向下传播”都是有效的：

(1)
caller1()          +    (2)
                            + > callee()
caller2()  > your_coro()  +
(4)          (3)

在这个图中，语义和非常松散地，如果caller1()被主动取消，那么{}被取消，然后你的协程被取消，然后{}被取消。如果caller2()被主动取消，也会发生大致相同的情况。在

（callee()是共享的，因此不是一个普通的协程，而是Task或Future）

你想要什么样的替代行为？在

屏蔽

如果您希望callee()继续，即使caller2()被取消，shield它：

^{pr2}$

反向屏蔽

如果您允许callee()死亡，但希望您的协同程序继续，请转换异常：

async def reverse_shield(awaitable):
    try:
        return await awaitable
    except asyncio.CancelledError:
        raise Exception("custom")

async def your_coro():
    await reverse_shield(callee_f)
    # handle custom exception

保护自己

这个问题值得怀疑-通常你应该允许你的来电者取消你的合作计划。在

一个值得注意的例外是，如果你的调用者是一个框架，并且它是不可配置的。在

^{4}$