如何优雅地处理 SIGTERM 信号？

Question

假设我们有一个用 python 编写的这样一个简单的守护进程：

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

mainloop()

and we daemonize it using start-stop-daemon which by default sends SIGTERM ( TERM ) signal on --stop .

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

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

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

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

原文由 zerkms 发布，翻译遵循 CC BY-SA 4.0 许可协议

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

原文由 Mayank Jaiswal 发布，翻译遵循 CC BY-SA 4.0 许可协议

Answer 2

首先，我不确定您是否需要第二个线程来设置 shutdown_flag 。

为什么不直接在 SIGTERM 处理程序中设置它？

另一种方法是从 SIGTERM 处理程序引发异常，该异常将向上传播到堆栈。假设你有适当的异常处理（例如 with / contextmanager 和 try: ... finally: 块）这个关闭应该是相当优雅的 Ctrl+C 你的程序。

示例程序 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

这次我发送它 SIGTERM 4 次迭代后 kill $(ps aux | grep signals-test | awk '/python/ {print $2}') ：

 $ ./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

原文由 Will Manley 发布，翻译遵循 CC BY-SA 4.0 许可协议