在asyncio.ensure_future中捕获错误

时间:2016-05-03 10:30:37

标签: python python-3.x exception python-asyncio

我有这段代码:

try:
    asyncio.ensure_future(data_streamer.sendByLatest())
except ValueError as e:
    logging.debug(repr(e))

data_streamer.sendByLatest()可以提升ValueError,但不会被抓住。

1 个答案:

答案 0 :(得分:9)

ensure_future - 只需创建Task并立即返回。你应该等待创建的任务得到它的结果(包括引发异常的情况):

import asyncio


async def test():
    await asyncio.sleep(0)
    raise ValueError('123')


async def main():    
    try:
        task = asyncio.ensure_future(test())  # Task aren't finished here yet 
        await task  # Here we await for task finished and here exception would be raised 
    except ValueError as e:
        print(repr(e))


if __name__ == '__main__':
    loop = asyncio.get_event_loop()
    loop.run_until_complete(main())

输出:

ValueError('123',)

如果您不打算在创建任务后立即等待任务,您可以稍后等待它(知道它是如何完成的):

async def main():    
    task = asyncio.ensure_future(test())
    await asyncio.sleep(1)
    # At this moment task finished with exception,
    # but we didn't retrieved it's exception.
    # We can do it just awaiting task:
    try:
        await task  
    except ValueError as e:
        print(repr(e)) 

输出相同:

ValueError('123',)