协程的取消和异常 Part 3 - 异常

Android

2023-11-24 06:38:09

协程中的异常处理

在协程中处理异常与在常规 Python 函数中处理异常非常相似。可以使用try/except块来捕获异常，并可以使用raise来引发异常。

异常的传播

当协程中发生异常时，该异常将沿着协程调用链向上传播，直到找到一个处理该异常的try/except块。如果未找到这样的块，则异常将传播到协程的调用者，依次类推，直到异常到达应用程序的根协程。

协程的取消

协程可以被取消，这意味着协程将立即停止执行，并且不会再有机会恢复执行。协程可以通过多种方式取消，包括：

使用asyncio.cancel()方法。
当协程的父协程被取消时。
当协程执行的操作被取消时，例如当协程正在等待一个未来的结果，并且该未来的结果被取消时。

当协程被取消时，将引发asyncio.CancelledError异常。协程可以通过捕获此异常来处理取消。

示例

以下示例演示了如何在协程中处理异常和取消：

import asyncio

async def my_coroutine():
    try:
        # 执行一些可能引发异常的操作
        result = await asyncio.sleep(1)
    except asyncio.CancelledError:
        # 处理取消
        pass
    except Exception as e:
        # 处理其他异常
        raise

async def main():
    # 创建一个协程
    coroutine = my_coroutine()

    # 等待协程完成
    try:
        await coroutine
    except asyncio.CancelledError:
        # 处理取消
        pass
    except Exception as e:
        # 处理其他异常
        raise

# 创建事件循环并运行主协程
asyncio.run(main())

在上面的示例中，my_coroutine()协程将等待一秒钟，然后返回一个结果。如果协程在等待期间被取消，则将引发asyncio.CancelledError异常。协程将捕获此异常并处理取消。如果协程在等待期间发生其他类型的异常，则将引发该异常，并由调用协程处理。

main()协程将创建my_coroutine()协程，然后等待其完成。如果my_coroutine()协程被取消，则main()协程也将捕获asyncio.CancelledError异常并处理取消。如果my_coroutine()协程发生其他类型的异常，则main()协程将引发该异常，以便由调用协程处理。