如何在 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中文網其他相關文章！