python - 协程从未被期待
问题描述
我正在使用一个简单的上下文管理器,里面有一个 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 呢?
解决方案
为什么它不等待 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))