python - 为什么我的主线程中没有正确处理 signal.SIGTERM？

我有连续运行的 python 代码(收集传感器数据)。它应该在启动时使用 start-stop-daemon 启动。但是，我希望能够优雅地终止进程，所以我从 How to process SIGTERM signal gracefully? 帖子中的建议开始，并将我的主循环放在一个单独的线程中。我希望能够在它作为守护进程运行时(start-stop-daemon 将发送一个终止信号)以及当我自己在终端中短暂启动它以进行测试(我按下 ctrl-c )时能够优雅地关闭它。

但是，如果我终止进程，似乎不会调用信号处理程序(即使不使用线程，“done (killed) ”也永远不会出现在我重定向到的文件中)。当我按下 ctrl-c 时，收集会继续并继续在终端(或我重定向到的文件)中打印数据。

我在以下代码中做错了什么？

from threading import Thread
import time, sys, signal

shutdown_flag = False #used for gracefull shutdown

def main_loop():
    while not shutdown_flag:
        collect_data() # contains some print "data" statements
        time.sleep(5)
    print "done (killed)"

def sighandler(signum, frame):
    print 'signal handler called with signal: %s ' % signum
    global shutdown_flag
    shutdown_flag = True

def main(argv=None):
    signal.signal(signal.SIGTERM, sighandler) # so we can handle kill gracefully
    signal.signal(signal.SIGINT, sighandler) # so we can handle ctrl-c
    try:
        Thread(target=main_loop, args=()).start()
    except Exception, reason:
        print reason

if __name__ == '__main__':
    sys.exit(main(sys.argv))

最佳答案

您正在使用以下语句终止主线程:

if __name__ == '__main__':
    sys.exit(main(sys.argv))

所以你的信号处理程序永远不会运行。信号处理程序是主线程的一部分，而不是您创建的 main_loop 线程。所以一旦主线程退出，就没有信号处理函数可以调用了。

你需要这样的东西:

def sighandler(signum, frame):
    print 'signal handler called with signal: %s ' % signum
    global shutdown_flag
    shutdown_flag = True
    sys.exit() # make sure you add this so the main thread exits as well.

if __name__ == '__main__':
    main(sys.argv)
    while 1:  # this will force your main thread to live until you terminate it.
       time.sleep(1)

一个简单的测试来查看你的程序中有多少线程正在运行:

def main_loop():
    while not shutdown_flag:
        collect_data() # contains some print "data" statements
        time.sleep(5)
        import threading
        print threading.enumerate()
    print "done (killed)"

关于python - 为什么我的主线程中没有正确处理 signal.SIGTERM？，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/19988726/