Jak z gracją przetwarzać sygnał SIGTERM?
Załóżmy, że mamy taki trywialny Demon napisany w Pythonie:
def mainloop():
while True:
# 1. do
# 2. some
# 3. important
# 4. job
# 5. sleep
mainloop()
I demonizujemy go używając start-stop-daemon
, który domyślnie wysyła SIGTERM
(TERM
) sygnał włączony --stop
.
Załóżmy, że aktualnie wykonywanym krokiem jest #2
. I właśnie w tym momencie wysyłamy TERM
sygnał.
Zdarza się, że egzekucja kończy się natychmiast.
Odkryłem, że mogę obsłużyć Zdarzenie sygnału za pomocą signal.signal(signal.SIGTERM, handler)
ale rzecz w tym, że nadal przerywa bieżące wykonanie i przekazuje kontrolę do handler
.
Moje pytanie brzmi - czy jest możliwe, aby nie przerywać bieżącego wykonania, ale obsłużyć sygnał TERM
w oddzielonym wątku (? tak, że byłem w stanie ustawić shutdown_flag = True
tak, że mainloop()
miał szansę zatrzymać się z wdziękiem?
7 answers
Rozwiązanie klasy clean to use:
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 :)")
Warning: date(): Invalid date.timezone value 'Europe/Kyiv', we selected the timezone 'UTC' for now. in /var/www/agent_stack/data/www/doraprojects.net/template/agent.layouts/content.php on line 54
2020-05-21 06:55:53
Po pierwsze, nie jestem pewien, czy potrzebujesz drugiego wątku, aby ustawić shutdown_flag
.
Dlaczego nie ustawić go bezpośrednio w programie obsługi SIGTERM?
Alternatywą jest wywołanie wyjątku z obsługi SIGTERM
, który będzie propagowany w górę stosu. Zakładając, że masz odpowiednią obsługę wyjątków (np. z with
/contextmanager
i try: ... finally:
bloki) powinno to być dość wdzięczne zamknięcie, podobne do tego, jeśli masz Ctrl + C Twój program.
Przykładowy program 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"
Teraz zobacz Ctrl + C zachowanie:
$ ./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
Tym razem wysyłam SIGTERM
Po 4 iteracjach z 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
Tym razem włączam mój Custom SIGTERM
handler i wysyłam go SIGTERM
:
$ ./signals-test.py handle_signal
Hello
Iteration #1
Iteration #2
Iteration #3
Iteration #4
Goodbye
$ echo $?
0
Warning: date(): Invalid date.timezone value 'Europe/Kyiv', we selected the timezone 'UTC' for now. in /var/www/agent_stack/data/www/doraprojects.net/template/agent.layouts/content.php on line 54
2020-01-24 22:47:47
Myślę, że jesteś blisko możliwego rozwiązania.
Wykonaj mainloop
w osobnym wątku i rozszerz go o Właściwość shutdown_flag
. Sygnał może być przechwytywany z signal.signal(signal.SIGTERM, handler)
w głównym wątku (nie w oddzielnym wątku). Funkcja obsługi sygnału powinna ustawić shutdown_flag
Na True i czekać, aż wątek zakończy się znakiem thread.join()
Warning: date(): Invalid date.timezone value 'Europe/Kyiv', we selected the timezone 'UTC' for now. in /var/www/agent_stack/data/www/doraprojects.net/template/agent.layouts/content.php on line 54
2014-04-14 21:50:58
Oto prosty przykład bez wątków i klas.
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
Warning: date(): Invalid date.timezone value 'Europe/Kyiv', we selected the timezone 'UTC' for now. in /var/www/agent_stack/data/www/doraprojects.net/template/agent.layouts/content.php on line 54
2017-01-19 23:21:17
Na podstawie poprzednich odpowiedzi utworzyłem menedżera kontekstu, który chroni przed sigint i 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")
Warning: date(): Invalid date.timezone value 'Europe/Kyiv', we selected the timezone 'UTC' for now. in /var/www/agent_stack/data/www/doraprojects.net/template/agent.layouts/content.php on line 54
2018-05-04 11:44:13
Znalazłem najprostszy sposób dla mnie. Oto przykład z widelcem dla jasności, że ten sposób jest przydatny do kontroli przepływu.
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)
Warning: date(): Invalid date.timezone value 'Europe/Kyiv', we selected the timezone 'UTC' for now. in /var/www/agent_stack/data/www/doraprojects.net/template/agent.layouts/content.php on line 54
2019-02-07 23:10:03
Najprostszym rozwiązaniem, jakie znalazłem, inspirując się powyższymi odpowiedziami jest
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)
Warning: date(): Invalid date.timezone value 'Europe/Kyiv', we selected the timezone 'UTC' for now. in /var/www/agent_stack/data/www/doraprojects.net/template/agent.layouts/content.php on line 54
2019-11-11 17:47:17