python - 从 `call_soon`回调函数执行协程

Question

我有以下情况:

• 某些内部类(我无法控制)正在使用 call_soon 执行我的 callback 函数.
• 在我的回调中，我想调用另一个 courotune，但最终以“卡住”回调结束。

我将使用修改后的 Hello World with call_soon()证明这一点:

import asyncio

def hello_world(loop):
    print('Hello')
    # Call some coroutine.
    yield from asyncio.sleep(5, loop=loop)
    print('World')
    loop.stop()

loop = asyncio.get_event_loop()

# Schedule a call to hello_world()
loop.call_soon(hello_world, loop)

# Blocking call interrupted by loop.stop()
loop.run_forever()
loop.close()

当我运行它时，没有打印任何东西，程序也永远不会结束。

Ctrl+C

Traceback (most recent call last):
  File "../soon.py", line 15, in <module>
    loop.run_forever()
  File "/usr/lib/python3.4/asyncio/base_events.py", line 276, in run_forever
    self._run_once()
  File "/usr/lib/python3.4/asyncio/base_events.py", line 1136, in _run_once
    event_list = self._selector.select(timeout)
  File "/usr/lib/python3.4/selectors.py", line 432, in select
    fd_event_list = self._epoll.poll(timeout, max_ev)
KeyboardInterrupt

实际发生了什么，为什么？

有什么正确的方法吗？

Answer 1

您提到的示例演示了如何安排回调。

如果您使用yield from 语法，该函数实际上是一个coroutine。并且必须进行相应的装饰:

@asyncio.coroutine
def hello_world(loop):
    print('Hello')
    yield from asyncio.sleep(5, loop=loop)
    print('World')
    loop.stop()

然后您可以使用 ensure_future 将协程安排为任务:

loop = asyncio.get_event_loop()
coro = hello_world(loop)
asyncio.ensure_future(coro)
loop.run_forever()
loop.close()

或者等效地，使用 run_until_complete :

loop = asyncio.get_event_loop()
coro = hello_world(loop)
loop.run_until_complete(coro)

---

两周后，python 3.5将正式发布，您将可以使用新的 async/await语法:

async def hello_world(loop):
    print('Hello')
    await asyncio.sleep(5, loop=loop)
    print('World')

---

编辑:它有点难看，但没有什么能阻止您创建一个回调来安排您的协程:

loop = asyncio.get_event_loop()
coro = hello_world(loop)
callback = lambda: asyncio.ensure_future(coro)
loop.call_soon(callback)
loop.run_forever()
loop.close()