Python `async def` 函数中使用 `yield` 和 `return` 的区别

在 Python 中，async def 函数用于定义异步函数，这些函数可以在执行过程中暂停和恢复，通常与 await 关键字一起使用。在异步函数中，yield 和 return 的使用方式与普通函数有所不同，本文将详细探讨它们之间的区别，并通过示例代码进行说明。

1. return 的使用

在 async def 函数中，return 语句用于返回一个值，并且一旦执行到 return 语句，函数将立即终止，不会继续执行后续代码。return 语句返回的值可以通过 await 获取。

示例代码

import asyncioasync def async_return_example():print("Start of async_return_example")await asyncio.sleep(1)  # 模拟异步操作print("Before return")return "Returned value"async def main():result = await async_return_example()print(f"Result: {result}")# 运行异步主函数
asyncio.run(main())

输出结果

Start of async_return_example
Before return
Result: Returned value

解释

• async_return_example 是一个异步函数，它在执行到 return 语句时返回字符串 "Returned value"。
• main 函数通过 await 调用 async_return_example，并获取返回值。
• return 语句执行后，函数立即终止，不会执行 return 语句之后的代码。

2. yield 的使用

在 async def 函数中，yield 语句用于生成一个值，并且函数不会立即终止。yield 语句通常用于生成器函数中，但在异步函数中使用时，它会将函数转换为异步生成器。异步生成器可以与 async for 循环一起使用。

示例代码

import asyncioasync def async_yield_example():print("Start of async_yield_example")await asyncio.sleep(1)  # 模拟异步操作print("Before yield")yield "Yielded value 1"await asyncio.sleep(1)  # 模拟异步操作print("Before second yield")yield "Yielded value 2"print("After second yield")async def main():async for value in async_yield_example():print(f"Received: {value}")# 运行异步主函数
asyncio.run(main())

输出结果

Start of async_yield_example
Before yield
Received: Yielded value 1
Before second yield
Received: Yielded value 2
After second yield

解释

• async_yield_example 是一个异步生成器函数，它在执行到 yield 语句时生成一个值，并且函数不会立即终止。
• main 函数通过 async for 循环遍历 async_yield_example 生成的值。
• yield 语句执行后，函数会暂停，等待下一次迭代，直到所有值都被生成。

3. 总结

• return：在 async def 函数中，return 用于返回一个值，并且函数会立即终止。返回的值可以通过 await 获取。
• yield：在 async def 函数中，yield 用于生成一个值，并且函数不会立即终止。yield 语句通常用于异步生成器中，可以通过 async for 循环遍历生成的值。