中断当前正在执行的所有 asyncio.sleep

where

这是在 Linux、Python 3.5.1 上。

what

我正在开发一个监控流程asyncio，他们在不同地方的任务await on asyncio.sleep不同时长的呼叫。

有时我希望能够打断所有所说的话asyncio.sleep调用并让所有任务正常进行，但我找不到如何做到这一点。一个示例是监控进程的正常关闭。

如何（失败的假设）

我以为我可以发送一个警报信号来达到这种效果，但进程终止了。我尝试用以下命令捕获警报信号：

def sigalrm_sent(signum, frame):
    tse.logger.info("got SIGALRM")

signal.signal(signal.SIGALRM, sigalrm_sent)

然后我得到了关于捕获 SIGALRM 的日志行，但是asyncio.sleep通话不会中断。

如何（拼凑）

此时我已经全部替换了asyncio.sleep调用此协程的调用：

async def interruptible_sleep(seconds):
    while seconds > 0 and not tse.stop_requested:
        duration = min(seconds, tse.TIME_QUANTUM)
        await asyncio.sleep(duration)
        seconds -= duration

所以我只需要选择一个TIME_QUANTUM不算太小，也不算太大。

but

有没有办法中断所有运行asyncio.sleep打电话而我错过了？

中断所有正在运行的调用asyncio.sleep似乎有点危险，因为它可以用于代码的其他部分，用于其他目的。相反，我会专门做一个sleep跟踪其正在运行的调用的协程。然后可以通过取消相应的任务来中断它们：

def make_sleep():
    async def sleep(delay, result=None, *, loop=None):
        coro = asyncio.sleep(delay, result=result, loop=loop)
        task = asyncio.ensure_future(coro)
        sleep.tasks.add(task)
        try:
            return await task
        except asyncio.CancelledError:
            return result
        finally:
            sleep.tasks.remove(task)

    sleep.tasks = set()
    sleep.cancel_all = lambda: sum(task.cancel() for task in sleep.tasks)
    return sleep

Example:

async def main(sleep, loop):
    for i in range(10):
        loop.create_task(sleep(i))
    await sleep(3)
    nb_cancelled = sleep.cancel_all()
    await asyncio.wait(sleep.tasks)
    return nb_cancelled

sleep = make_sleep()
loop = asyncio.get_event_loop()
result = loop.run_until_complete(main(sleep, loop)) 
print(result)  # Print '6'

出于调试目的，loop.time = lambda: float('inf')也有效。