Python协程如何处理异常

在Python中，协程（coroutine）是一种特殊的函数，可以在执行过程中暂停和恢复。处理协程中的异常与处理普通函数的异常类似，但有一些不同之处。当协程中发生异常时，它会被捕获并存储在协程对象的_exception属性中。要处理协程中的异常，可以使用asyncio.gather()函数，它会收集多个协程的结果，并将它们迭代器返回。如果在协程中发生异常，asyncio.gather()会立即停止执行其他协程，并将异常传递给return_exceptions参数指定的回调函数。

下面是一个处理协程异常的示例：

import asyncio

async def raise_exception():
    raise ValueError("An error occurred in the coroutine")

async def handle_exception(task):
    try:
        await task
    except Exception as e:
        print(f"Caught exception: {e}")

async def main():
    task1 = asyncio.create_task(raise_exception())
    task2 = asyncio.create_task(asyncio.sleep(1))  # This will not raise an exception

    results = await asyncio.gather(task1, task2, return_exceptions=True)

    for result in results:
        if isinstance(result, Exception):
            handle_exception(result)
        else:
            print(f"Result: {result}")

asyncio.run(main())

在这个示例中，我们定义了一个名为raise_exception的协程，它会引发一个ValueError异常。我们还定义了一个名为handle_exception的协程，用于处理异常。在main协程中，我们使用asyncio.gather()函数执行raise_exception和asyncio.sleep(1)协程。我们将return_exceptions参数设置为True，以便在发生异常时将它们传递给handle_exception协程。最后，我们遍历results列表，并根据结果类型调用相应的处理函数。