在异步模型中,执行由事件循环调度和协调。要取消当前暂停任务的执行,您基本上只需要不恢复它。虽然这在实践中有点不同,但很明显这使得取消暂停的任务在理论上很简单。
单独的超时当然是可能的:每当您暂停协程以等待结果时,您都想提供一个超时值。事件循环将确保在达到超时且任务尚未完成时取消等待任务。
一些具体示例:
>>> import asyncio
>>> loop = asyncio.get_event_loop()
>>> task = asyncio.ensure_future(asyncio.sleep(5))
>>> task.cancel()
>>> loop.run_until_complete(task)
Traceback (most recent call last):
...
concurrent.futures._base.CancelledError
在实践中,这可能会使用以下方式实现:
class Foo:
task = None
async def sleeper(self):
self.task = asyncio.sleep(60)
try:
await self.task
except concurrent.futures.CancelledError:
raise NotImplementedError
当这个方法处于休眠状态时,其他人可以调用foo.task.cancel() 来唤醒协程并让它处理取消。或者,任何调用sleeper() 的人都可以直接取消它,而不给它清理的机会。
设置超时同样简单:
>>> loop.run_until_complete(asyncio.wait_for(asyncio.sleep(60), 5))
[ ... 5 seconds later ... ]
Traceback (most recent call last):
...
concurrent.futures._base.TimeoutError
特别是在 HTTP 请求超时的情况下,请参阅aiohttp:
async def fetch_page(session, url):
with aiohttp.Timeout(10):
async with session.get(url) as response:
assert response.status == 200
return await response.read()
with aiohttp.ClientSession(loop=loop) as session:
content = loop.run_until_complete(fetch_page(session, 'http://python.org'))
显然,对fetch_page 的每次调用都可以决定自己的aiohttp.Timeout 值,并且每个单独的实例都会在达到该超时时抛出自己的异常。