如何在 Python 中优雅地处理 SIGTERM 信号以顺利终止进程？

处理 SIGTERM 信号以实现进程正常终止

在 Python 中管理守护进程的上下文中，优雅地处理 SIGTERM 信号对于确保有序关闭至关重要。当收到 SIGTERM 信号时，程序的默认行为是立即终止，这可能会中断关键操作。

为了防止突然终止，必须正确处理 SIGTERM 信号。一种方法涉及使用 signal.signal 函数来注册信号的处理程序。但是，该处理程序仍会中断当前执行流程并取得控制权。

要进行更精细的信号处理，请考虑使用单独的线程或进程来管理信号。这将处理过程与主执行分离，允许程序在异步处理信号的同时继续其操作。

一种常见的技术涉及创建一个标志（例如 shutdown_flag），当 SIGTERM 发生时该标志设置为 True处理程序被触发。主循环可以定期检查标志，以便在设置 shutdown_flag 时正常终止进程。

下面是一个示例，演示如何使用类实现干净的解决方案：

import signal
import time

class GracefulKiller:
    kill_now = False  # Flag to indicate termination

    def __init__(self):
        # Register SIGTERM and SIGINT handlers to set kill_now
        signal.signal(signal.SIGINT, self.exit_gracefully)
        signal.signal(signal.SIGTERM, self.exit_gracefully)

    def exit_gracefully(self, signum, frame):
        self.kill_now = True

if __name__ == '__main__':
    killer = GracefulKiller()
    # Main loop that continues until kill_now is set to True
    while not killer.kill_now:
        time.sleep(1)
        print("Doing something in a loop...")
    
    print("End of the program. I was killed gracefully :)")

此脚本创建一个处理 SIGTERM 和 SIGINT 信号的 GracefulKiller 类。当收到任一信号时，kill_now 标志设置为 True。主循环定期检查此标志，并在kill_now 变为 True 时正常终止。

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