python – 调用ensure_future是否正确结束协同程序的结果？

发布时间：2020-12-20 11:05:35 所属栏目：Python 来源：网络整理

导读：我还不确定何时使用ensure_future,我想知道这是否是我想要/需要它的情况. 请考虑以下代码 import asyncioloop = asyncio.get_event_loop()async def do_work(count): for i in range(count): if i % 10000 == 0: await asyncio.sleep(0.1) print(count) retu

我还不确定何时使用ensure_future,我想知道这是否是我想要/需要它的情况.

请考虑以下代码

import asyncio


loop = asyncio.get_event_loop()


async def do_work(count):
    for i in range(count):
        if i % 10000 == 0:
            await asyncio.sleep(0.1)

    print(count)
    return i


async def do_batch_work():
    res1 = do_work(100000)
    res2 = do_work(100)
    await asyncio.wait([res1,res2])
    return res1,res2


def main():
    loop = asyncio.get_event_loop()
    res1,res2 = loop.run_until_complete(do_batch_work())
    print(res1.result())
    print(res2.result())
    loop.close()

main()

这打印：

100
100000
Traceback (most recent call last):
  File "more_async.py",line 30,in <module>
    main()
  File "more_async.py",line 26,in main
    print(res1.result())
AttributeError: 'coroutine' object has no attribute 'result'

异步代码按预期运行并按预期顺序打印出来,但loop.run_until_complete(…)不允许我访问底层结果,因为协程对象似乎无法获得结果.

我可以通过如下更改方法来解决这个问题

async def do_batch_work():
    res1 = asyncio.ensure_future(do_work(100000))
    res2 = asyncio.ensure_future(do_work(100))
    await asyncio.wait([res1,res2

通过调用asyncio.ensure_future(…),我确保将任务返回到我可以调用result()的位置.

我想知道,这是处理它的正确方法吗？如果我关心那个协程的结果,还是我还有另一种我不知道的方法,我是否需要使用ensure_future？

解决方法

async def do_batch_work():
    res1 = do_work(100000)
    res2 = do_work(100)
    await asyncio.wait([res1,res2

res1和res2只是该示例中的协同程序,而不是Future对象,因此没有结果属性.当您使用ensure_future()时,res1和res2现在是Future对象,这就是您可以访问result属性的原因. wait()在两种情况下的工作方式相同,并且两个代码中的结果都是同等生成的,只是你的函数没有返回正确的对象.

如果要修改第一个示例,则wait()将返回所需的Future对象作为第一个项目.

async def do_batch_work():
    res1 = do_work(100000)
    res2 = do_work(100)
    (res1,res2),_ = await asyncio.wait([res1,res2

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!