Question

我有一段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

真正的代码远比这个代码片段复杂，但结构是相同的：即主函数保持无限循环。

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

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

Answer 1

嗯，根据documentation atexit处理程序如果程序被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。

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

1 个答案: