快捷導(dǎo)航

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

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

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

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

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

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

下面的代碼演示了如何使用 endless 包來(lái)實(shí)現(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) {
        // 模擬程序處理請(qǐng)求需要 5 秒
        time.Sleep(5 * time.Second)
        c.JSON(http.StatusOK, gin.H{
            "message": "pong",
        })
    })

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

    log.Println("Server exiting")
}

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

為了驗(yàn)證這個(gè)實(shí)現(xiàn)，我們可以通過(guò)以下步驟進(jìn)行測(cè)試：

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

Endless 的工作原理

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

值得注意的是，由于 endless 的這種機(jī)制，當(dāng)你的項(xiàng)目是通過(guò)類似 supervisor 的軟件來(lái)管理進(jìn)程時(shí)，這種方式就不再適用了。因?yàn)?supervisor 會(huì)根據(jù) PID 來(lái)管理進(jìn)程，而在優(yōu)雅重啟過(guò)程中 PID 是會(huì)變化的，這會(huì)導(dǎo)致 supervisor 認(rèn)為服務(wù)已經(jīng)崩潰。