本文探討了如何通過Go代碼實現在后臺運行的程序。最近我用Go語言開發了一個WebSocket服務，我希望它能在后臺運行，并在異常退出時自動重新啟動。我的整體思路是將程序轉為后臺進程，也就是守護進程（daemon）。它不處理具體的業務邏輯，而是再次使用相同的參數調用自身，啟動一個子進程來處理業務邏輯。守護進程監視子進程的狀態，如果子進程退出，則再次啟動一個新的子進程。這樣就能保證在服務異常終止時及時重啟。
我在網上找到了一個開源庫，Github.com/sevlyar/go-daemon，它很方便地實現了在后臺啟動一個新的進程，但如果后臺進程再次嘗試作為另一個后臺進程啟動，會出現錯誤。
后來我閱讀了源代碼才發現：為了區分當前進程是父進程還是子進程，作者巧妙地設計了一個環境變量標識。正是因為這種識別策略，該庫只能啟動一次自身作為后臺進程，無法連續啟動自身為后臺進程。
不過，這種使用環境變量來區分進程身份的思路給我啟發很大。基于這個想法，我通過延伸和優化，最終實現了在保持參數不變的情況下連續啟動自身為后臺進程。我對作者表示敬意。
此外，我還找到了一些其他的庫，它們的思路有所不同，主要通過添加特殊參數來標記進程身份。但是，這些方法并沒有完美地解決讓進程啟動自身的問題，令我有些遺憾。
最終，我決定自己實現一個庫來解決我的項目需求，并希望它是一個通用的庫，可以快速方便地將用Go語言編寫的服務程序轉為后臺運行或守護進程模式運行。本文總結了我在這次探索中的經驗和收獲。
首先，讓我們區分一下兩個概念：后臺運行和守護進程。平常交流時，我們可能不太區分或區分不夠清晰。在本文中，我想明確如下定義：
后臺運行：指進程在操作系統中以非顯示方式運行，沒有與任何命令行終端或程序界面相關聯。這種方式下運行的進程稱為后臺進程，比如沒有與任何終端相關聯的命令行程序進程。
守護進程：也稱為守護進程，它首先以后臺運行方式啟動，然后還有額外的職責。在本文中，我的定義是守護進程可以監視Go服務程序進程的狀態，如果異常退出，可以自動重新啟動。這樣守護進程可以確保服務程序一直在后臺運行，即使它在某些情況下崩潰或意外終止。
接下來，我將介紹如何使用Go代碼來實現在后臺運行的程序，并將其轉化為一個守護進程。
后臺運行程序
要將Go程序在后臺運行，可以使用一些操作系統級別的方法。以下是一種簡單的方法：
package mAIn
import (
"fmt"
"os"
"os/exec"
"syscall"
)
func main() {
if os.Getppid() != 1 {
cmd := exec.Command(os.Args[0])
cmd.Start()
fmt.Println("Background process ID:", cmd.Process.Pid)
os.Exit(0)
}
// 在這里寫入具體的業務邏輯代碼
fmt.Println("Running in background...")
select {}
}
在上面的代碼中，我們首先使用os.Getppid()函數獲取當前進程的父進程ID。如果父進程不是1，說明當前進程不是守護進程，而是從終端啟動的。在這種情況下，我們創建一個新的命令，使用相同的參數再次啟動程序，并在后臺運行。我們打印出新進程的PID，并退出初始進程。
如果進程的父進程是1，那么說明當前進程已經是守護進程了，我們可以在此處寫入具體的業務邏輯代碼。
使用這種方法，我們可以確保程序在后臺運行，而且還可以檢查是否已經啟動了一個后臺進程。
守護進程
將程序轉化為守護進程需要額外的步驟，我們需要創建一個監聽子進程狀態的循環，并在子進程異常退出時重新啟動它。以下是一個簡單的守護進程實現：
package main
import (
"fmt"
"os"
"os/exec"
"syscall"
)
func main() {
if os.Getppid() != 1 {
cmd := exec.Command(os.Args[0])
cmd.Start()
fmt.Println("Background process ID:", cmd.Process.Pid)
os.Exit(0)
}
// 在這里寫入具體的業務邏輯代碼
fmt.Println("Running in background...")
for {
cmd := exec.Command(os.Args[0])
cmd.Start()
exitCh := make(chan error)
go func() {
exitCh <- cmd.Wait()
}()
err := <-exitCh
if err != nil {
fmt.Println("Process exited with error:", err)
} else {
fmt.Println("Process exited successfully")
}
select {
case <-exitCh:
default:
}
}
}
在上面的代碼中，我們添加了一個循環，用于監聽子進程的狀態。在每次子進程退出之后，我們使用相同的參數再次啟動守護進程，并重新開始監聽。這樣就可以確保服務程序在異常退出時能夠自動重新啟動。
這只是一個簡單的守護進程實現，你可以根據自己的需求進行擴展和優化。
