快捷導航

Go Web服務(wù)優(yōu)雅平滑重啟

更新時間：2025年09月21日 11:15:20 作者：樹上有只程序猿

在生產(chǎn)環(huán)境中,當我們需要對正在運行的服務(wù)進行升級時,如何確保不影響當前未處理完的請求,同時又能應(yīng)用新的代碼,下面就來介紹一下Go Web服務(wù)優(yōu)雅平滑重啟,感興趣的可以了解一下

什么是優(yōu)雅重啟？

優(yōu)雅重啟的核心思想是：在服務(wù)啟動新的進程處理新請求的同時，允許舊的進程繼續(xù)完成其手頭未完成的工作，然后再優(yōu)雅地退出。這種方式可以確保服務(wù)在升級的過程中不會出現(xiàn)中斷，提升用戶體驗的同時，也降低了在服務(wù)切換過程中的風險。

實現(xiàn)優(yōu)雅重啟的代碼示例

下面的代碼演示了如何使用 endless 包來實現(xiàn) Gin 服務(wù)的優(yōu)雅重啟。

package main

import (
    "log"
    "net/http"
    "time"

    "github.com/fvbock/endless"
    "github.com/gin-gonic/gin"
)

func main() {
    router := gin.Default()
    router.GET("/ping", func(c *gin.Context) {
        // 模擬程序處理請求需要 5 秒
        time.Sleep(5 * time.Second)
        c.JSON(http.StatusOK, gin.H{
            "message": "pong",
        })
    })

    // 默認 endless 服務(wù)器會監(jiān)聽下列信號：
    // syscall.SIGHUP，syscall.SIGUSR1，syscall.SIGUSR2，syscall.SIGINT，syscall.SIGTERM 和 syscall.SIGTSTP
    // 接收到 SIGHUP 信號將觸發(fā) `fork/restart` 實現(xiàn)優(yōu)雅重啟（kill -1 pid 會發(fā)送 SIGHUP 信號）
    // 接收到 syscall.SIGINT 或 syscall.SIGTERM 信號將觸發(fā)優(yōu)雅關(guān)機
    // 接收到 SIGUSR2 信號將觸發(fā) HammerTime
    // SIGUSR1 和 SIGTSTP 被用來觸發(fā)一些用戶自定義的 hook 函數(shù)
    if err := endless.ListenAndServe(":8080", router); err != nil {
        log.Printf("Server err: %v\n", err)
    }

    log.Println("Server exiting")
}

如何驗證優(yōu)雅重啟？

為了驗證這個實現(xiàn)，我們可以通過以下步驟進行測試：

打開終端，執(zhí)行 go build -o gin_graceful_restart gin_graceful_restart.go 編譯程序，然后執(zhí)行 ./gin_graceful_restart 啟動服務(wù)。終端會輸出當前服務(wù)的 PID（例如：[pid] 12345）。
修改代碼中的 /ping 接口的響應(yīng)內(nèi)容，比如將 pong 修改為 pong1。
再次執(zhí)行 go build -o gin_graceful_restart gin_graceful_restart.go 編譯程序。
在瀏覽器中訪問 http://127.0.0.1:8080/ping，此時瀏覽器會等待服務(wù)返回響應(yīng)（由于接口模擬了 5 秒的延遲）。
在另一個終端中執(zhí)行 kill -1 12345 命令，向服務(wù)發(fā)送 syscall.SIGHUP 信號，12345 為第一步中的 PID。
依舊在第 4 步的瀏覽器中等待，等響應(yīng)到 pong 信息之后，再次刷新頁面，你會發(fā)現(xiàn)響應(yīng)內(nèi)容變成了 pong1，這意味著服務(wù)已經(jīng)應(yīng)用了新的代碼，同時之前的請求也得到了正確的處理。

Endless 的工作原理

endless 包實現(xiàn)優(yōu)雅重啟的原理非常簡單：它會 fork 一個新的子進程來處理新的請求，而舊的進程則繼續(xù)處理已經(jīng)接收的請求。當舊的進程處理完所有的請求后才會退出。因此，雖然 PID 發(fā)生了變化，但服務(wù)依舊保持了無縫銜接。

值得注意的是，由于 endless 的這種機制，當你的項目是通過類似 supervisor 的軟件來管理進程時，這種方式就不再適用了。因為 supervisor 會根據(jù) PID 來管理進程，而在優(yōu)雅重啟過程中 PID 是會變化的，這會導致 supervisor 認為服務(wù)已經(jīng)崩潰。