python - asyncio - 如何在信号处理程序中使用协程？

Question

我正在开发一个使用 python3.4 中的 asyncio 进行联网的应用程序。当此应用程序干净地关闭时，节点需要与集线器“断开连接”。此断开连接是一个需要网络连接的事件过程，因此循环需要在关闭之前等待此连接完成。

我的问题是，使用协程作为信号处理程序会导致应用程序无法关闭。请考虑以下示例:

import asyncio
import functools
import os
import signal

@asyncio.coroutine
def ask_exit(signame):
    print("got signal %s: exit" % signame)
    yield from asyncio.sleep(10.0)
    loop.stop()

loop = asyncio.get_event_loop()
for signame in ('SIGINT', 'SIGTERM'):
    loop.add_signal_handler(getattr(signal, signame),
                                        functools.partial(ask_exit, signame))

print("Event loop running forever, press CTRL+c to interrupt.")
print("pid %s: send SIGINT or SIGTERM to exit." % os.getpid())
loop.run_forever()

如果您运行此示例然后按 Ctrl+C，则什么也不会发生。问题是，如何使用信号和协程实现这种行为？

Answer 1

python >=3.5 的语法

loop = asyncio.get_event_loop()
for signame in ('SIGINT', 'SIGTERM'):
    loop.add_signal_handler(getattr(signal, signame),
                            lambda: asyncio.ensure_future(ask_exit(signame)))