SIGTERMシグナルを正常に処理する方法は?


197

Pythonで書かれたそのような簡単なデーモンがあるとしましょう:

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

mainloop()

そしてstart-stop-daemon、デフォルトでSIGTERMTERM)シグナルを送信するを使用してデーモン化します--stop

現在実行されているステップがであるとしましょう#2。そして、この瞬間にTERM信号を送信しています。

実行はすぐに終了します。

を使用してシグナルイベントを処理できることがわかりましたsignal.signal(signal.SIGTERM, handler)が、それでも現在の実行が中断され、コントロールがに渡されhandlerます。

だから、私の質問は-現在の実行を中断せずTERMに、別のスレッド(?)でシグナルを処理shutdown_flag = Truemainloop()て、正常に停止するように設定できるようにすることは可能ですか?


2
プロセスへののsignalfd配信を使用してマスクすることで、以前に求めていたことをSIGTERM実行しました。
Eric Urban

回答:


276

クラスベースのクリーンで使いやすいソリューション:

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

1
アイデアをありがとう!リブートガードで修正したアプローチを使用しました。github.com/ryran/reboot-guard/blob/master/rguard#L284:L304
rsaw

7
これは最良の答えであり(スレッドは不要)、推奨される最初の試みのアプローチです。
jose.angel.jimenez

2
@ Mausy5043 Pythonでは、クラスを定義するための括弧を使用できません。python 3.xの場合は完全に問題ありませんが、python 2.xの場合は、「クラスXYZ(object):」を使用することをお勧めします。理由:docs.python.org/2/reference/datamodel.html#newstyle
Mayank Jaiswal

2
フォローアップして、やる気を維持してください、ありがとう。いつも使っています。
chrisfauerbach 2016

2
最悪の場合、それは単に正常にシャットダウンする前に別の反復を実行することを意味します。False値は一度だけ設定され、かつ複数のアクセスが問題ではないので、それが唯一のFalseからTrueに行くことができます。
Alceste_

52

まず、を設定するために2番目のスレッドが必要かどうかはわかりませんshutdown_flag
SIGTERMハンドラで直接設定しないのはなぜですか?

別の方法は、SIGTERMハンドラーから例外を発生させることです。これは、スタックに伝達されます。適切な例外処理(with/ contextmanagertry: ... 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

3
「SIGTERMハンドラで直接設定しないでください」---ワーカースレッドがランダムな場所で中断するため。複数のステートメントをワーカーループに入れると、ソリューションがワーカーをランダムな位置で終了させ、ジョブが不明な状態のままになることがわかります。
zerkms 2014

私にとっても、Dockerのコンテキストでもうまく機能します。ありがとう!
マリアン

4
フラグを設定し、例外を発生させない場合、それはスレッドの場合と同じになります。したがって、ここではスレッドを使用する必要はありません。
2015年

28

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

mainloop別のスレッドで実行し、プロパティで拡張しますshutdown_flag。シグナルはsignal.signal(signal.SIGTERM, handler)、メインスレッドで(別のスレッドではなく)キャッチできます。シグナルハンドラはshutdown_flagTrueに設定し、スレッドが終了するのを待つ必要がありますthread.join()


4
はい、分離されたスレッドが私が最終的に解決した方法です。おかげで
zerkms '29

7
ここではスレッドは必要ありません。シングルスレッドプログラム自体では、最初にシグナルハンドラーを登録し(シグナルハンドラーの登録は非ブロッキング)、次にメインループを記述します。シグナルハンドラー関数はいつフラグを設定し、ループはこのフラグをチェックする必要があります。私は同じためのクラスベースのソリューションをここに貼り付けました。
Mayank Jaiswal 2015

2
セカンドスレッドを持っている必要はありません。シグナルハンドラを登録します。
oneloop


26

以下は、スレッドやクラスのない簡単な例です。

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

11

以前の回答に基づいて、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 __name__ == '__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")

4

私にとって最も簡単な方法を見つけました。ここでは、この方法がフロー制御に役立つことを明確にするためにフォークを使用した例を示します。

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

私が見つけた最も簡単な解決策は、上記の応答からインスピレーションを得ることです

class SignalHandler:

    def __init__(self):

        # register signal handlers
        signal.signal(signal.SIGINT, self.exit_gracefully)
        signal.signal(signal.SIGTERM, self.exit_gracefully)

        self.logger = Logger(level=ERROR)

    def exit_gracefully(self, signum, frame):
        self.logger.info('captured signal %d' % signum)
        traceback.print_stack(frame)

        ###### do your resources clean up here! ####

        raise(SystemExit)
弊社のサイトを使用することにより、あなたは弊社のクッキーポリシーおよびプライバシーポリシーを読み、理解したものとみなされます。
Licensed under cc by-sa 3.0 with attribution required.