让我们假设我们用python编写了这样一个简单的守护进程:
def mainloop():
while True:
# 1. do
# 2. some
# 3. important
# 4. job
# 5. sleep
mainloop()
我们使用start-stop-daemon守护它，默认情况下发送SIGTERM (TERM)信号在——stop上。
让我们假设当前执行的步骤是#2。此时，我们发送TERM信号。
结果是执行立即终止。
我发现我可以使用signal.signal(信号。SIGTERM，处理程序)，但问题是它仍然中断当前执行并将控制传递给处理程序。
所以，我的问题是-是否有可能不中断当前的执行，而是在一个分离的线程中处理TERM信号(?)，以便我能够设置shutdown_flag = True，以便mainloop()有机会优雅地停止?
当前回答
您可以设置线程。事件，当捕获信号。
线程。事件是线程安全的，可以使用和传递，可以等待，同样的事件可以从其他地方设置和清除。
import signal, threading
quit_event = threading.Event()
signal.signal(signal.SIGTERM, lambda *_args: quit_event.set())
while not quit_event.is_set():
print("Working...")
其他回答
一个基于类的清洁使用解决方案:
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 :)")
我认为你已经接近一个可能的解决方案了。
在一个单独的线程中执行mainloop，并使用属性shutdown_flag扩展它。这个信号可以用signal(信号)捕获。SIGTERM，处理器)在主线程(而不是在一个单独的线程中)。信号处理程序应该将shutdown_flag设置为True，并等待线程以thread.join()结束。
我的代码示例如何使用信号:
#! /usr/bin/env python
import signal
def ctrl_handler(signum, frm):
print "You can't cannot kill me"
print "Installing signal handler..."
signal.signal(signal.SIGINT, ctrl_handler)
print "done"
while True:
# do something
pass
找到了对我来说最简单的方法。 这里有一个带有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, signal.SIGTERM)
os.waitpid(p, 0)
根据前面的回答，我创建了一个上下文管理器，它可以保护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")
