在 asyncio 中实现并行的 IO 等待可以通过使用 asyncio.gather()
函数来完成。asyncio.gather()
函数接受一个或多个可等待对象(coroutine 或者 Future)作为参数,并返回一个新的 Future 对象。这个新的 Future 对象将在所有可等待对象完成后完成,且其结果是一个由所有可等待对象的结果组成的列表。
下面是一个示例代码,演示了如何使用 asyncio.gather()
并行等待多个 IO 操作的完成:
import asyncio
async def fetch_data(url):
# 模拟一个 IO 操作
await asyncio.sleep(1)
print(f'Fetching data from {url}')
return f'Data from {url}'
async def main():
# 创建一个任务列表
urls = ['http://example.com', 'http://example.org', 'http://example.net']
tasks = [fetch_data(url) for url in urls]
# 并行等待任务完成
results = await asyncio.gather(*tasks)
# 处理结果
for result in results:
print(result)
asyncio.run(main())
在上面的示例中,fetch_data()
是一个模拟的 IO 操作,使用了 asyncio.sleep()
来模拟延迟。main()
函数创建了一个任务列表 tasks
,其中每个任务都是一个 fetch_data()
的调用。asyncio.gather(*tasks)
并行等待所有任务的完成,并返回一个包含所有结果的列表。最后,我们可以使用 for
循环遍历结果并进行处理。
运行上述代码,将输出类似于以下内容:
Fetching data from http://example.com
Fetching data from http://example.org
Fetching data from http://example.net
Data from http://example.com
Data from http://example.org
Data from http://example.net
可以看到,三个 IO 操作被并行执行,并最终返回了各自的结果。