取消异常 - 如何立即停止 Python 协程？

协程作为一种强大的异步编程工具，在 Python 中发挥着重要作用。协程的取消操作是不可或缺的一部分，因为它允许开发者在某些情况下停止协程的执行。然而，在使用 job.cancel() 方法取消协程时，可能会遇到延迟取消的问题，导致协程继续执行一段时间，甚至打印出意外结果。

那么，如何避免取消协程延迟的问题，确保协程程序能够及时取消呢？

理解取消异常

Python 中的 asyncio.CancelledError 异常是一种特殊的异常，当协程被取消时抛出。这个异常可以用来捕获取消事件，并采取相应的处理措施，比如清理资源或退出协程。

使用 try/except 块捕获取消异常

为了确保协程程序能够及时取消，我们可以使用 try/except 块来捕获取消异常。当取消异常被抛出时，我们可以立即执行必要的处理操作，并退出协程。

async def my_coroutine():
    try:
        # 协程代码
    except asyncio.CancelledError:
        # 取消异常处理代码

在上面的示例中，my_coroutine() 函数是一个协程函数，使用 try/except 块捕获了取消异常。当协程被取消时，asyncio.CancelledError 异常会被抛出，并执行异常处理代码。

使用 asyncio.ensure_future() 方法

asyncio.ensure_future() 方法可以用来创建一个协程对象，并立即启动它的执行。如果在协程执行过程中需要取消它，可以使用 cancel() 方法来取消它。

import asyncio

async def my_coroutine():
    # 协程代码

# 创建协程对象并启动执行
coroutine = asyncio.ensure_future(my_coroutine())

# 取消协程
coroutine.cancel()

在上面的示例中，我们使用 asyncio.ensure_future() 方法创建了一个协程对象，并立即启动它的执行。然后，我们使用 cancel() 方法取消了协程。

避免取消协程延迟的技巧

以下是一些避免取消协程延迟的技巧：

• 使用 try/except 块捕获取消异常。
• 使用 asyncio.ensure_future() 方法来创建协程对象并立即启动它的执行。
• 在协程代码中使用 await asyncio.sleep(0) 来让出控制权，以便其他协程有机会执行。

通过使用这些技巧，我们可以确保协程程序能够及时取消，并避免延迟取消的问题。