python - 每个任务完成后立即处理异步任务列表

Question

如何在任务完成后立即处理 async 任务列表的结果？

例如，以下应显示首先加载的页面:

urls = ['stackoverflow.com', 'google.com']
tasks = [asyncio.create_task(fetch_page(x)) for x in urls]

for page in asyncio.give_me_results_ASAP(tasks):
    print(page.url)

由于 google 加载速度更快，我希望它打印:

google.com
stackoverflow.com

Answer 1

asyncio.as_completed正是为此而设计的，并按照任务完成的顺序返回协同程序的迭代器。从迭代器返回的第一个协程将对应于完成的第一个任务，您可以在每个协程上等待以获取任务的结果。

# With Python 3.8+
import asyncio
import time

async def fetch_page(url):
    reponse_time = 0.1 if url == 'google.com' else 0.8
    await asyncio.sleep(reponse_time)
    return url

async def main():
    urls = ['stackoverflow.com', 'google.com']
    tasks = [asyncio.create_task(fetch_page(x)) for x in urls]

    for coro in asyncio.as_completed(tasks):
        print(f"{time.time():.3f}", await coro)

if __name__ == '__main__':
    asyncio.run(main())

产生:

1600821288.178 google.com
1600821288.280 stackoverflow.com