首页 > 解决方案 > 协程从未被期待

问题描述

我正在使用一个简单的上下文管理器,里面有一个 asyncio 循环:

class Runner:
    def __init__(self):
        self.loop = asyncio.get_event_loop()

    def __enter__(self):
        return self

    def __exit__(self, *args):
        self.loop.close()

    def do_work(self):
        ...
        return self.loop.run_until_complete(asyncio.gather(*futures))

当我使用两个 Runner 对象时,出现“从未等待协程”错误。

with Runner() as r:
    r.do_work()

with Runner() as r2:
    r2.do_work()

因为在第一个 Runner(r) 中关闭了一个循环。如果我不在exit中关闭循环,一切正常,但我不需要保持打开状态。我知道我们在一个线程中只能有一个循环,但为什么它不等待 run_until_complete 呢?

标签: pythonpython-asyncio

解决方案


为什么它不等待 run_until_complete

大概会发生这样的事情:

import asyncio


async def test():
    await asyncio.sleep(0.1)


if __name__ ==  '__main__':
    loop = asyncio.get_event_loop()

    loop.run_until_complete(test())
    loop.close()

    loop.run_until_complete(test())

结果:

RuntimeError: Event loop is closed
sys:1: RuntimeWarning: coroutine 'test' was never awaited

如何解决这个问题?

由于您以这种方式使用事件循环,因此每次都可以使用新的事件循环:

class Runner:
    def __init__(self):
        self.loop = asyncio.new_event_loop()  # *new*_event_loop


    def do_work(self):
        # Make sure all futures are created 
        # with relevant event loop been set as current
        asyncio.set_event_loop(self.loop)

        # ...
        return self.loop.run_until_complete(asyncio.gather(*futures))

推荐阅读