Skip to main content
 首页 » 编程设计

python之Python中signal handler和atexit handler的调用

2025年05月04日9pengyingh

我有一段 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 函数保持一个无限循环。

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

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

请您参考如下方法:

嗯,根据documentation如果程序被 Python 未处理的信号杀死,或者出现内部错误,或者调用了 os._exit(),则不会执行 atexit 处理程序。所以我会使用这样的东西(几乎复制了你的代码):

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 .