python - 如何优雅地处理 SIGTERM 信号？

Question

假设我们有这样一个用 python 编写的琐碎的守护进程:

def mainloop():
    while True:
        # 1. do
        # 2. some
        # 3. important
        # 4. job
        # 5. sleep

mainloop()

我们使用 start-stop-daemon 对其进行守护，默认情况下在 --stop 上发送 SIGTERM (TERM) 信号.

假设当前执行的步骤是#2。而此时我们正在发送 TERM 信号。

发生的情况是执行立即终止。

我发现我可以使用 signal.signal(signal.SIGTERM, handler) 处理信号事件，但问题是它仍然会中断当前执行并将控制权传递给 处理程序。

所以，我的问题是 - 是否可以不中断当前执行，而是在单独的线程中处理 TERM 信号(？)，以便我能够设置 shutdown_flag = True 以便 mainloop() 有机会优雅地停止？

Answer 1

基于类的清洁使用解决方案:

import signal
import time

class GracefulKiller:
  kill_now = False
  def __init__(self):
    signal.signal(signal.SIGINT, self.exit_gracefully)
    signal.signal(signal.SIGTERM, self.exit_gracefully)

  def exit_gracefully(self, *args):
    self.kill_now = True

if __name__ == '__main__':
  killer = GracefulKiller()
  while not killer.kill_now:
    time.sleep(1)
    print("doing something in a loop ...")
   
  print("End of the program. I was killed gracefully :)")