如何优雅地处理SIGTERM信号？

Question

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

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

mainloop()

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

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

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

我发现我可以使用＜code＞signal.signal（signal.SIGTERM，handler）。

所以，我的问题是——有没有可能不中断当前的执行，而是在一个单独的线程中处理< code>TERM信号(？)以便我能够设置< code>shutdown_flag = True以便< code>mainloop()有机会正常停止？

Answer 1

下面是一个没有线程或类的简单示例。

import signal

run = True

def handler_stop_signals(signum, frame):
    global run
    run = False

signal.signal(signal.SIGINT, handler_stop_signals)
signal.signal(signal.SIGTERM, handler_stop_signals)

while run:
    pass # do stuff including other IO stuff

Answer 2

首先，我不确定您是否需要第二个线程来设置shutdown_flag。
为什么不直接在 SIGTERM 处理程序中设置它呢？

另一种方法是从 SIGTERM 处理程序中引发异常，该异常将沿堆栈向上传播。假设您已经进行了适当的异常处理（例如，使用/contextmanager并尝试：...最后：块）这应该是一个相当优雅的关闭，类似于你要Ctrl C你的程序。

示例程序< code>signals-test.py:

#!/usr/bin/python

from time import sleep
import signal
import sys


def sigterm_handler(_signo, _stack_frame):
    # Raises SystemExit(0):
    sys.exit(0)

if sys.argv[1] == "handle_signal":
    signal.signal(signal.SIGTERM, sigterm_handler)

try:
    print "Hello"
    i = 0
    while True:
        i += 1
        print "Iteration #%i" % i
        sleep(1)
finally:
    print "Goodbye"

现在查看Ctrl C行为：

$ ./signals-test.py default
Hello
Iteration #1
Iteration #2
Iteration #3
Iteration #4
^CGoodbye
Traceback (most recent call last):
  File "./signals-test.py", line 21, in <module>
    sleep(1)
KeyboardInterrupt
$ echo $?
1

这一次，我在使用＜code＞kill$（ps aux|grep signals test| awk'/python/{print$2}）进行4次迭代后将其发送＜code＞SIGTERM＜/code＞。：

$ ./signals-test.py default
Hello
Iteration #1
Iteration #2
Iteration #3
Iteration #4
Terminated
$ echo $?
143

这次我启用了我的自定义 SIGTERM 处理程序并将其发送到 SIGTERM：

$ ./signals-test.py handle_signal
Hello
Iteration #1
Iteration #2
Iteration #3
Iteration #4
Goodbye
$ echo $?
0

Answer 3

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

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 :)")