Python中信号处理程序和atexit处理程序的调用

问题描述

我有一段Python代码，如下所示：

import sys
import signal
import atexit

def release():
    print "Release resources..."

def sigHandler(signo, frame):
    release()
    sys.exit(0)

if __name__ == "__main__":
    signal.signal(signal.SIGTERM, sigHandler)
    atexit.register(release)

    while True:
        pass

实际代码比此代码片段复杂得多，但结构是相同的：即main函数维护无限循环。

我需要一个信号回调来释放占用的资源，比如DB句柄。 然后，我添加了一个SIGTERM处理程序，以防服务器终止，它只需调用Release函数，然后退出进程。 atexit目标是处理流程成功完成。

现在我遇到了一个问题，我只想在进程终止时只调用release一次。我的代码有什么改进吗？

---

解决方案

documentation

根据documentation，如果程序被未被Python处理的信号杀死，或者发生内部错误，或者如果调用了os._exit()，则不会执行退出处理程序。因此，我将使用如下内容(几乎复制了您的代码)：

import sys
import signal
import atexit

def release():
    print "Release resources..."

def sigHandler(signo, frame):
    sys.exit(0)

if __name__ == "__main__":
    atexit.register(release)
    signal.signal(signal.SIGTERM, sigHandler)

    while True:
        pass

我检查过release()被调用一次，且仅在TERM(外部发出)和INTR信号(Ctrl-C从键盘发出)的情况下调用一次。如果需要，您可以安装更多的信号处理程序(例如，用于HUP等)。如果您需要"更优雅的关闭"，您应该找到一种方法来优雅地中断循环和/或安装外部"关闭处理程序"(在SIGKILL的情况下，您将没有机会干净地释放资源)，或者简单地使您的应用程序ACID。