Python 并发程序如何优雅退出?

Python并发程序优雅退出需主动监听信号、及时清理资源、避免强制终止;用signal

捕获SIGINT设退出标志,以threading.Event或asyncio任务管理生命周期,统一关闭文件、连接、日志等资源,并规避sys.exit()、os._exit()等危险操作。

Python 并发程序优雅退出的核心是:主动监听退出信号、及时清理资源、避免强制终止导致状态不一致或数据丢失。

使用 signal 捕获中断信号

对于运行在终端的脚本,Ctrl+C 会发送 SIGINT,需捕获并触发有序关闭流程:

  • 注册 signal.signal(signal.SIGINT, handler),在 handler 中设置退出标志(如 event.set() 或修改全局布尔变量)
  • 避免在 handler 中执行耗时操作(如网络请求、文件写入),只做轻量标记
  • 若用 threading.Event 控制工作线程,主线程捕获信号后调用 event.set(),各工作线程循环中检查 event.is_set() 并安全退出

为线程和协程统一管理生命周期

不同并发模型退出方式略有差异,但都应避免 threading.Thread._stop()(已废弃且不安全)或 asyncio.Task.cancel() 后不等待:

  • 多线程:用 threading.Event 作为“退出开关”,工作线程定期检查;退出前调用 thread.join(timeout=3) 等待自然结束,超时则记录警告
  • asyncio:启动任务时保存 Task 对象;收到退出信号后调用 task.cancel(),再用 await asyncio.gather(*tasks, return_exceptions=True) 等待全部完成
  • concurrent.futures:调用 executor.shutdown(wait=True, cancel_futures=True) —— Python 3.9+ 支持 cancel_futures,可尝试取消未开始的任务

释放关键资源与确保数据一致性

退出前必须完成资源清理,否则可能引发泄漏或脏数据:

  • 关闭打开的文件句柄、数据库连接、网络 socket,优先使用 with 语句或显式 .close()
  • 若正在写入日志或数据库,确保最后一条“服务已停止”记录落盘(例如调用 logging.shutdown()connection.commit()
  • 对共享状态(如缓存、计数器),在退出前做一次原子性快照或持久化,避免重启后状态错乱

避免常见陷阱

一些看似合理但实际危险的做法:

  • 在信号 handler 中直接调用 sys.exit() —— 可能跳过 finally 和上下文管理器的 __exit__
  • os._exit() 强制终止 —— 绕过所有 Python 清理逻辑,极易丢失缓冲区数据
  • 忽略子进程 —— 若用 subprocess.Popen 启动了外部程序,退出前应调用 proc.terminate() + proc.wait()
  • 协程中未 await 清理函数 —— 如 async def cleanup(): ... 忘记 await cleanup(),导致异步资源未释放