web-dev-qa-db-ja.com

SIGTERMシグナルを上品に処理するには？

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)を使ってシグナルイベントを処理できることがわかりましたが、それでも現在の実行が中断され、制御がhandlerに渡されるということです。

それで、私の質問は、現在の実行を中断せずにTERMシグナルを別のスレッド（？）で処理してshutdown_flag = Trueを設定してmainloop()を適切に停止できるようにすることは可能かどうかです。

pythondaemonsigtermstart-stop-daemon
162
zerkms

クラスベースのクリーンな使い方の解決策：

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,signum, frame):
    self.kill_now = True

if __== '__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 :)"
222
Mayank Jaiswal

まず、shutdown_flagを設定するために2番目のスレッドが必要かどうかはわかりません。 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

今回はkill $(ps aux | grep signals-test | awk '/python/ {print $2}')で4回繰り返した後にSIGTERMを送信します。

$ ./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
44
Will Manley

私はあなたが可能な解決策に近いと思います。

別のスレッドでmainloopを実行し、それをプロパティshutdown_flagで拡張します。シグナルはメインスレッドのsignal.signal(signal.SIGTERM, handler)で捉えることができます（別のスレッドではありません）。シグナルハンドラはshutdown_flagをTrueに設定し、スレッドがthread.join()で終了するのを待つべきです

27
moliware

これはスレッドやクラスのない簡単な例です。

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
22
thoughtarray

前の答えに基づいて、私はsigintとsigtermから保護するコンテキストマネージャを作成しました。

import logging
import signal
import sys


class TerminateProtected:
    """ Protect a piece of code from being killed by SIGINT or SIGTERM.
    It can still be killed by a force kill.

    Example:
        with TerminateProtected():
            run_func_1()
            run_func_2()

    Both functions will be executed even if a sigterm or sigkill has been received.
    """
    killed = False

    def _handler(self, signum, frame):
        logging.error("Received SIGINT or SIGTERM! Finishing this block, then exiting.")
        self.killed = True

    def __enter__(self):
        self.old_sigint = signal.signal(signal.SIGINT, self._handler)
        self.old_sigterm = signal.signal(signal.SIGTERM, self._handler)

    def __exit__(self, type, value, traceback):
        if self.killed:
            sys.exit(0)
        signal.signal(signal.SIGINT, self.old_sigint)
        signal.signal(signal.SIGTERM, self.old_sigterm)


if __== '__main__':
    print("Try pressing ctrl+c while the sleep is running!")
    from time import sleep
    with TerminateProtected():
        sleep(10)
        print("Finished anyway!")
    print("This only prints if there was no sigint or sigterm")
10
Okke

私にとって最も簡単な方法が見つかりました。ここではわかりやすくするためにforkを使った例を示します。

import signal
import time
import sys
import os

def handle_exit(sig, frame):
    raise(SystemExit)

def main():
    time.sleep(120)

signal.signal(signal.SIGTERM, handle_exit)

p = os.fork()
if p == 0:
    main()
    os._exit()

try:
    os.waitpid(p, 0)
except (KeyboardInterrupt, SystemExit):
    print('exit handled')
    os.kill(p, 15)
    os.waitpid(p, 0)
0
Kron