場景設置:
你編寫了一個python服務程序,并且在命令行下啟動,而你的命令行會話又被終端所控制,python服務成了終端程序的一個子進程。因此如果你關閉了終端,這個命令行程序也會隨之關閉。
要使你的python服務不受終端影響而常駐系統,就需要將它變成守護進程。
守護進程就是Daemon程序,是一種在系統后臺執行的程序,它獨立于控制終端并且執行一些周期任務或觸發事件,通常被命名為"d"字母結尾,如常見的httpd、syslogd、systemd和dockerd等。
代碼實現
python可以很簡潔地實現守護進程,下面先給出代碼和相應注釋:
# coding=utf8import osimport sysimport atexitdef daemonize(pid_file=None): """ 創建守護進程 :param pid_file: 保存進程id的文件 :return: """ # 從父進程fork一個子進程出來 pid = os.fork() # 子進程的pid一定為0,父進程大于0 if pid: # 退出父進程,sys.exit()方法比os._exit()方法會多執行一些刷新緩沖工作 sys.exit(0) # 子進程默認繼承父進程的工作目錄,最好是變更到根目錄,否則回影響文件系統的卸載 os.chdir('/') # 子進程默認繼承父進程的umask(文件權限掩碼),重設為0(完全控制),以免影響程序讀寫文件 os.umask(0) # 讓子進程成為新的會話組長和進程組長 os.setsid() # 注意了,這里是第2次fork,也就是子進程的子進程,我們把它叫為孫子進程 _pid = os.fork() if _pid: # 退出子進程 sys.exit(0) # 此時,孫子進程已經是守護進程了,接下來重定向標準輸入、輸出、錯誤的描述符(是重定向而不是關閉, 這樣可以避免程序在 print 的時候出錯) # 刷新緩沖區先,小心使得萬年船 sys.stdout.flush() sys.stderr.flush() # dup2函數原子化地關閉和復制文件描述符,重定向到/dev/nul,即丟棄所有輸入輸出 with open('/dev/null') as read_null, open('/dev/null', 'w') as write_null: os.dup2(read_null.fileno(), sys.stdin.fileno()) os.dup2(write_null.fileno(), sys.stdout.fileno()) os.dup2(write_null.fileno(), sys.stderr.fileno()) # 寫入pid文件 if pid_file: with open(pid_file, 'w+') as f: f.write(str(os.getpid())) # 注冊退出函數,進程異常退出時移除pid文件 atexit.register(os.remove, pid_file)
概括一下守護進程的編寫步驟:
理解幾個要點
為什么要fork兩次
第一次fork,是為了脫離終端控制的魔爪。父進程之所以退出,是因為終端敲擊鍵盤、或者關閉時給它發送了信號;而fork出來的子進程,在父進程自殺后成為孤兒進程,進而被操作系統的init進程接管,因此脫離終端控制。
新聞熱點
疑難解答