何时调用asyncio的loop.close()才安全？相关问题解析

嘿，看得出来你已经跟asyncio打交道挺久了，这个loop.close()的时机问题确实是个容易踩的坑——我当初也在这里卡过一阵！咱们一步步来解决你的三个问题：

1. 你遗漏了什么？如何修改代码避免异常？ #

你遇到的RuntimeError: Event loop is closed，大概率是因为在事件循环还没完全停止、或者还有未彻底清理的任务/回调时就调用了loop.close()。

核心问题点： #

• 取消任务是异步操作：当你调用task.cancel()时，任务不会立刻终止，它需要在遇到下一个await点时响应取消，执行完清理逻辑才算真正结束。
• loop.close()只能在循环停止后调用：如果循环还在运行（比如run_until_complete还没执行完），直接调用close()就会触发异常。

修改后的示例代码： #

import asyncio
import time
import logging

logging.basicConfig(level=logging.INFO)

async def blocking_call(task_id):
    try:
        # 模拟阻塞式调用
        await asyncio.get_event_loop().run_in_executor(None, lambda: time.sleep(1))
        logging.info(f"Task {task_id} completed")
        # 模拟任务出错场景
        if task_id == 2:
            raise RuntimeError("Intentional failure for task 2")
    except Exception as e:
        logging.error(f"Task {task_id} failed: {str(e)}")
        raise  # 抛出异常触发全局取消

async def main():
    tasks = [asyncio.create_task(blocking_call(i)) for i in range(5)]
    try:
        await asyncio.gather(*tasks)
    except Exception as e:
        logging.info("Error detected, canceling all remaining tasks")
        # 取消所有未完成的任务
        for task in tasks:
            if not task.done():
                task.cancel()
        # 关键：等待所有任务彻底完成（包括被取消的）
        # return_exceptions=True避免因取消抛出的异常中断流程
        await asyncio.gather(*tasks, return_exceptions=True)
        logging.info("All tasks cleaned up")

if __name__ == "__main__":
    loop = asyncio.get_event_loop()
    try:
        # 运行主协程直到完成，此时所有任务都已处理完毕
        loop.run_until_complete(main())
    finally:
        # 此时循环已停止，调用close()是安全的
        loop.close()
        logging.info("Event loop closed safely")

关键修改说明： #

• 将loop.close()移到run_until_complete()执行完毕后的finally块中：run_until_complete会等待main协程彻底结束，此时所有任务（包括被取消的）都已经完成清理。
• 取消任务后必须等待它们完成：用await asyncio.gather(*tasks, return_exceptions=True)确保所有被取消的任务都执行完收尾逻辑。

2. 不调用loop.close()会有什么后果？生产环境隐患？ #

短时间运行的脚本可能看不出问题，但生产环境长期运行的服务会有这些隐患：

• 资源泄漏：run_in_executor默认使用的线程池，如果不关闭loop，线程池可能不会被销毁，导致线程资源一直占用，累积下来会消耗大量系统内存和CPU。
• 未释放的异步资源：比如打开的网络连接、文件句柄、异步IO缓冲区等，虽然Python程序退出时会自动清理，但如果是持续运行的服务，这些泄漏会逐渐累积，导致性能下降甚至服务崩溃。
• 兼容性风险：asyncio的官方文档明确推荐使用完loop后关闭它，未来Python版本可能会对未关闭的loop增加更严格的检查，导致你的代码出现兼容性问题。
• 多次创建loop的场景问题：如果你的程序需要动态创建和销毁loop（比如多线程场景），不关闭旧loop会导致资源无法释放，每次创建新loop都会占用新资源。

3. 为何会抛出异常？任务已退出，loop仍不“满意”？ #

事件循环关闭的条件比你想象的更严格：它要求所有任务、回调、I/O操作都彻底完成或取消，且循环已经停止。你遇到异常的原因可能是：

• 循环仍在运行时调用close：如果在run_until_complete执行过程中（比如main协程内部）调用loop.close()，此时循环还处于运行状态，直接触发RuntimeError。
• 任务未彻底清理：即使你取消了任务，有些任务可能卡在阻塞的run_in_executor调用中（线程还在运行），此时任务处于pending状态，循环还有未完成的操作，关闭时就会报错。
• 隐藏的pending回调：asyncio内部可能有一些延迟回调（比如日志打印、资源清理）还在等待执行，这些回调没完成前，循环不会认为自己可以被安全关闭。

简单来说：你看到任务“退出”只是表面现象，asyncio内部还有一些收尾工作没做完，这时候强行关闭循环就会被它“拒绝”。

内容的提问来源于stack exchange，提问作者Marco Massenzio