如何在 Python 中优雅地处理 SIGTERM 信号？

在 Python 中优雅地处理 SIGTERM 信号

在使用长时间运行的 Python 守护进程时，优雅地处理诸如 SIGTERM 之类的终止信号对于确保干净关闭，不会丢失数据。这涉及暂停当前执行、执行必要的清理任务，然后正常终止进程的能力。以下是实现此目的的方法：

使用信号处理程序

Python 提供了 signal.signal 函数来为特定信号注册处理程序。通过注册 SIGTERM 的处理程序，我们可以在信号终止执行之前拦截该信号。然而，正如问题中提到的，处理程序仍然会中断当前执行并将控制权传递给自身。

使用单独的线程进行信号处理

以避免中断当前执行时，我们可以创建一个单独的线程来处理信号事件。这使我们能够在后台执行与关闭相关的任务，而不会中断主循环。下面是一个使用简单类的示例：

解决方案：

import signal
import threading
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, signum, frame):
        self.kill_now = True

def mainloop():
    killer = GracefulKiller()
    while not killer.kill_now:
        # Perform necessary tasks in the main loop
        time.sleep(1)
        print("doing something in a loop ...")

if __name__ == '__main__':
    mainloop()

    # When the main loop exits, we have gracefully handled the termination signal
    print("End of the program. I was killed gracefully :)")

在此解决方案中，GracefulKiller 类为 SIGTERM 注册一个信号处理程序。当收到信号时，kill_now 标志设置为 True。主循环检查该标志并继续直到它为 True。这允许主循环在正常终止之前完成任何关键任务。

通过使用单独的线程或基于类的解决方案，我们可以确保处理终止信号而不会过早终止主执行。这种方法允许我们执行正常关闭，确保数据完整性和守护进程干净退出。

以上是如何在 Python 中优雅地处理 SIGTERM 信号？的详细内容。更多信息请关注PHP中文网其他相关文章！