在使用asyncio.wait_for()函数时,需要注意当等待超时时,此函数会取消被等待的协程任务,同时抛出asyncio.TimeoutError异常。如果被等待的协程任务没有被取消,那么可能会出现一些问题,比如资源无法释放,任务无法终止等。
为了解决这个问题,可以在使用asyncio.wait_for()函数等待协程任务之前,先使用asyncio.shield()函数将协程任务包装起来。当使用asyncio.shield()函数包装协程任务时,即使等待超时,协程任务也不会被取消,而是继续执行,直到协程任务完成或手动取消。
示例代码如下:
import asyncio
async def my_coro(): try: await asyncio.sleep(10) except asyncio.CancelledError: print("Coro task was cancelled.")
async def main(): coro_task = asyncio.create_task(my_coro()) try: await asyncio.wait_for(asyncio.shield(coro_task), timeout=5) except asyncio.TimeoutError: print("Wait timed out, but task was not cancelled.") else: print("Task completed before timeout.")
coro_task.cancel()
await coro_task
asyncio.run(main())