时间戳

首页 > 解决方案 > 在 asyncio 中检索任务结果的正确方法

python-asyncio - 在 asyncio 中检索任务结果的正确方法

问题描述

我正在尝试检查waitasyncio 模块的方法。我有这个原始的测试应用程序作为游乐场：

import asyncio

async def foo():
    return 1  # datetime.datetime.now()

async def entry_point():
    coros = [foo()]*3
    done, pending = await asyncio.wait(coros, return_when=asyncio.FIRST_COMPLETED)
    for obj in done:
        print(obj._result)  # works, but seem dirty

asyncio.run(entry_point())

基本上，我的目标是获得第一个完成任务的结果。这是我在术语上有点困惑。文档说asyncio.wait

返回两组任务/期货：（完成，待处理）。

我如何知道对象是 atask还是 a future？主要问题是，如何提取成功结束任务的结果？一种方法是访问受保护的属性_result（如我的代码片段所示）。但我觉得有一种更清洁的方法可以做到这一点。实现这一目标的正确模式是什么？

https://docs.python.org/3/library/asyncio-task.html#waiting-primitives

标签： python-asyncio

解决方案

asyncio.wait的文档有以下注释：

自 3.8 版起已弃用，将在 3.11 版中删除：不建议将协程对象直接传递给 wait()。

因此，您应该使用也具有asyncio.Task.result方法的asyncio.Task：

测试.py：

import asyncio
import random


async def task(i):
    t = random.uniform(1, 5)
    print(f"START: {i} ({t:.3f}s)")
    await asyncio.sleep(t)
    print(f"END: {i}")

    return i


async def main():
    tasks = []

    for i in range(5):
        tasks.append(asyncio.create_task(task(i)))

    done, pending = await asyncio.wait(tasks, return_when=asyncio.FIRST_COMPLETED)

    for t in done:
        print(t.result())


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

测试：

$ python test.py
START: 0 (2.743s)
START: 1 (2.490s)
START: 2 (4.785s)
START: 3 (3.746s)
START: 4 (1.010s)
END: 4
4

如果要检索所有结果并首先获取最早的下一个结果，请改用asyncio.as_completed：

...
for t in asyncio.as_completed(tasks):
    print(await t)
...

测试 2：

$ python test.py
START: 0 (2.155s)
START: 1 (1.309s)
START: 2 (3.380s)
START: 3 (3.451s)
START: 4 (1.587s)
END: 1
1
END: 4
4
END: 0
0
END: 2
2
END: 3
3

推荐阅读