Python Asyncio强制超时

问题描述

使用Asyncio可以在超时的情况下执行协程，因此在超时后会取消协程：

@asyncio.coroutine
def coro():
    yield from asyncio.sleep(10)

loop = asyncio.get_event_loop()
loop.run_until_complete(asyncio.wait_for(coro(), 5))

以上示例按预期工作(5秒后超时)。

但是，当协同例程不使用asyncio.sleep()(或其他异步协同例程)时，它似乎不会超时。示例：

@asyncio.coroutine
def coro():
    import time
    time.sleep(10)

loop = asyncio.get_event_loop()
loop.run_until_complete(asyncio.wait_for(coro(), 1))

这需要10秒以上的时间才能运行，因为time.sleep(10)未取消。在这种情况下可以强制取消协程吗？

如果应该使用Asyncio来解决此问题，我该如何解决？

---

解决方案

否，您不能中断协程，除非它将控制权交还给事件循环，这意味着它需要在yield from调用中。asyncio是单线程的，因此当您阻塞第二个示例中的time.sleep(10)调用时，事件循环无法运行。这意味着当您使用wait_for设置的超时到期时，事件循环将无法对其执行操作。在coro退出之前，事件循环没有机会再次运行，此时为时已晚。

这就是为什么通常情况下，您应该始终避免任何非异步的阻塞调用；任何时候，在没有屈服于事件循环的情况下阻塞调用，程序中的其他任何东西都不能执行，这可能不是您想要的。如果您确实需要执行较长的阻塞操作，则应该尝试使用BaseEventLoop.run_in_executor在线程或进程池中运行它，这样可以避免阻塞事件循环：

import asyncio
import time
from concurrent.futures import ProcessPoolExecutor

@asyncio.coroutine
def coro(loop):
    ex = ProcessPoolExecutor(2)
    yield from loop.run_in_executor(ex, time.sleep, 10)  # This can be interrupted.

loop = asyncio.get_event_loop()
loop.run_until_complete(asyncio.wait_for(coro(loop), 1))