Python graceful shutdown 的多框架实现

1次阅读

asyncio.run()无法捕获sigint因其内置信号处理器直接调用loop.stop();优雅关闭需手动创建事件循环、注册信号处理器后运行,windows需特殊处理;fastapi lifespan需显式启用;flask+gevent需spawn greenlet执行异步清理;多进程下各worker须独立实现shutdown。

Python graceful shutdown 的多框架实现

asyncio.run() 里没法直接捕获 SIGINT?

因为 asyncio.run() 内部会自动安装信号处理器并调用 loop.stop(),此时你写的 try/except KeyboardInterrupt 根本不会触发。想做优雅关闭,得绕过它——改用手动创建事件循环 + 显式运行。

  • 正确姿势是:调用 asyncio.new_event_loop(),设置 loop.add_signal_handler() 绑定 SIGINTSIGTERM 到自定义清理函数
  • 必须在 loop.run_forever() 前注册信号处理器,否则收不到信号
  • Windows 不支持 add_signal_handler(),得用 loop.add_reader() 配合 os.set_handle_inheritable() 模拟,但更推荐直接用 asyncio.run() + atexit 做有限兜底

FastAPI 的 lifespan 为什么没生效?

lifespan 只在用 uvicorn.run() 启动、且显式传入 lifespan 参数时才起作用;如果用 uvicorn main:app 命令行启动,默认不启用 lifespan,startup/shutdown 事件根本不会触发。

  • 确保启动方式是 uvicorn.run("main:app", lifespan="on")
  • lifespan 异步生成器里,yield 之前是 startup 阶段,之后是 shutdown 阶段,中间任何异常都会中断整个流程
  • 别在 yield 后面写耗时操作(比如等数据库连接池清空),Uvicorn 默认只给 5 秒超时,超时后强制 kill,得配 timeout-graceful-shutdown

Flask + gevent 怎么安全关掉 greenlet?

gevent 的 monkey patch 会让标准库的 socket、Thread 等变成协程安全,但信号处理仍走原生路径,signal.signal() 注册的 handler 里不能直接 await,否则报 RuntimeError: await outside async function

  • gevent.spawn() 把异步清理逻辑扔进新的 greenlet,再用 gevent.wait() 等它结束
  • 务必调用 gevent.killall() 清理残留 greenlet,否则主进程可能卡住
  • 如果用了 flask.cli 启动(比如 flask run),它自带 reloader,会 fork 子进程,SIGINT 只发给父进程,子进程得靠心跳或共享 flag 感知退出信号

多进程模型下 shutdown 为什么只杀了一个 worker?

gunicorn 这类多进程服务器,主进程收到 SIGTERM 后会转发给所有 worker,但每个 worker 是独立 python 进程,有自己的事件循环和资源句柄。你在单个 worker 里写的 shutdown 逻辑,不会自动同步到其他 worker。

立即学习Python免费学习笔记(深入)”;

  • 每个 worker 必须单独实现自己的清理逻辑,不能依赖“主进程统一关”
  • 避免在 shutdown 阶段访问共享资源(如 redis 连接池),不同 worker 可能同时 close 同一个连接,引发 ConnectionResetError
  • 若需协调(比如等所有 worker 完成任务再退出),得用外部机制:文件锁、Redis pub/sub 或进程间通信(multiprocessing.Queue),但会增加复杂度

真正麻烦的不是怎么写 shutdown 函数,而是确认每个组件——数据库连接、http 客户端、消息队列消费者、后台 task——都注册了对应的 cleanup,并且它们的关闭顺序和超时时间互相兼容。漏掉一个,就可能卡住整个进程。

text=ZqhQzanResources