捕捉async/await中的异常

2023-09-14 09:56:35

使用async/await可以很好地处理异步任务，但是当出现异常时，情况就变得复杂了。在传统的同步编程中，我们可以使用try/catch块来捕获异常，但是在async/await中，事情变得更加微妙。

async/await中的异常处理

在async/await中，异常可以发生在两个地方：

回调函数中 ：当调用异步函数时，如果函数的执行过程中出现异常，则该异常将在回调函数中被捕获。
async方法中 ：如果async方法的执行过程中出现异常，则该异常将在async方法本身中被捕获。

如何捕获回调函数中的异常

要捕获回调函数中的异常，可以使用try/catch块将回调函数包裹起来。例如：

def callback(result):
    try:
        # 执行一些操作
    except Exception as e:
        # 捕获异常

# 调用异步函数
async def main():
    await asyncio.sleep(1)
    callback("result")

asyncio.run(main())

如何捕获async方法中的异常

要捕获async方法中的异常，可以使用try/catch块将async方法包裹起来。例如：

async def main():
    try:
        # 执行一些操作
    except Exception as e:
        # 捕获异常

asyncio.run(main())

如何同时捕获回调函数和async方法中的异常

如果需要同时捕获回调函数和async方法中的异常，可以使用以下方法：

使用 add_done_callback 方法将回调函数添加到异步任务中。当异步任务完成时，回调函数将被调用，并且可以捕获其中的异常。
使用 await 操作符等待异步任务完成。如果异步任务的执行过程中出现异常，则该异常将在 await 操作符处被捕获。

例如：

async def main():
    task = asyncio.create_task(asyncio.sleep(1))

    try:
        await task
    except Exception as e:
        # 捕获异常

asyncio.run(main())