Go語言實現(xiàn)優(yōu)雅關(guān)機和重啟的示例詳解

更新時間：2025年04月21日 09:35:30 作者：大學生亨亨

優(yōu)雅的關(guān)機是指在關(guān)閉服務之前,先讓服務處理完當前正在處理的請求,然后再關(guān)閉服務,本文主要為大家詳細介紹了如何使用Go語言實現(xiàn)優(yōu)雅關(guān)機和重啟,感興趣的小伙伴可以參考一下

一、優(yōu)雅關(guān)機

生活化例子

餐館關(guān)門：你去餐館吃火鍋，剛坐下點完菜（客戶端發(fā)請求），餐館老板突然接到通知要停電（收到關(guān)機指令）。老板很貼心，先停止接待新客人（停止接收新請求），等你這桌和其他正在吃飯的客人（正在處理的請求）都吃完了，再關(guān)門走人。這樣你的火鍋就能安安全全吃完，不會有任何損失。

代碼示例

package main
 
import (
	"context"
	"log"
	"net/http"
	"os"
	"os/signal"
	"syscall"
	"time"
 
	"github.com/gin-gonic/gin"
)
 
func main() {
	// 創(chuàng)建路由器
	router := gin.Default()
	router.GET("/cook-hotpot", func(c *gin.Context) {
		time.Sleep(5 * time.Second) // 模擬煮火鍋需要時間
		c.String(http.StatusOK, "火鍋煮好了，可以吃啦！")
	})
 
	// 創(chuàng)建服務器
	srv := &http.Server{
		Addr:    ":8080",
		Handler: router,
	}
 
	// 在單獨的goroutine中啟動服務器
	go func() {
		if err := srv.ListenAndServe(); err != nil && err != http.ErrServerClosed {
			log.Fatalf("服務器監(jiān)聽錯誤: %s", err)
		}
	}()
 
	// 創(chuàng)建一個通道來接收系統(tǒng)信號
	quit := make(chan os.Signal, 1)
	signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM) // 監(jiān)聽Ctrl+C和系統(tǒng)終止信號
 
	// 阻塞等待信號
	<-quit
	log.Println("開始優(yōu)雅關(guān)機...")
 
	// 設(shè)置一個5秒超時的上下文
	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
	defer cancel()
 
	// 調(diào)用Shutdown方法優(yōu)雅關(guān)閉服務器
	if err := srv.Shutdown(ctx); err != nil {
		log.Fatalf("服務器優(yōu)雅關(guān)機失敗: %s", err)
	}
 
	log.Println("服務器已安全退出")
}

驗證效果

運行上面的代碼，啟動服務。
打開瀏覽器訪問 http://127.0.0.1:8080/cook-hotpot，模擬開始煮火鍋。
在終端迅速執(zhí)行 Ctrl+C 命令，向程序發(fā)送關(guān)機信號。
觀察程序不會立即退出，而是等待火鍋煮好（請求處理完）后才退出，實現(xiàn)優(yōu)雅關(guān)機。

二、優(yōu)雅重啟

生活化例子

保安換班：小區(qū)門口有兩名保安，保安A（老進程）正在值班，保安B（新進程）來接班。保安A不會直接走人，而是等手頭的事（比如處理一輛正在進入的車）忙完，再讓保安B接管工作。這樣小區(qū)門口的秩序不會因為換班而受到影響。

代碼示例

package main
 
import (
	"log"
	"net/http"
	"time"
 
	"github.com/fvbock/endless"
	"github.com/gin-gonic/gin"
)
 
func main() {
	// 創(chuàng)建路由器
	router := gin.Default()
	router.GET("/greet", func(c *gin.Context) {
		time.Sleep(5 * time.Second) // 模擬處理請求需要時間
		c.String(http.StatusOK, "你好呀，歡迎來到小區(qū)！")
	})
 
	// 使用endless啟動服務器，支持優(yōu)雅重啟
	if err := endless.ListenAndServe(":8080", router); err != nil {
		log.Fatalf("服務器監(jiān)聽錯誤: %s", err)
	}
 
	log.Println("服務器已安全退出")
}

驗證效果

編譯并運行上述代碼，終端會輸出當前進程的 PID。
修改代碼中處理請求函數(shù)的返回值，如將 "你好呀，歡迎來到小區(qū)！" 修改為 "歡迎光臨，小區(qū)新保安在此！"，然后重新編譯。
打開瀏覽器訪問 http://127.0.0.1:8080/greet，模擬車輛進入小區(qū)。
在終端迅速執(zhí)行 kill -1 <PID> 命令，向程序發(fā)送優(yōu)雅重啟信號。
等當前請求處理完（保安A處理完手頭的事），再次訪問時會收到新的歡迎語，說明在不影響當前請求的情況下完成了優(yōu)雅重啟，同時進程號也變成了新的（保安B接班）。

三、總結(jié)

優(yōu)雅關(guān)機和優(yōu)雅重啟的核心就是“有始有終”，不甩手不管正在做的事。

優(yōu)雅關(guān)機就像餐館老板等你吃完火鍋再關(guān)門，通過監(jiān)聽系統(tǒng)信號（如Ctrl+C），調(diào)用Shutdown()方法，停止接收新請求并等待現(xiàn)有請求處理完。

優(yōu)雅重啟就像保安換班，使用endless庫監(jiān)聽特定信號（如SIGHUP），啟動新進程處理新請求，同時讓老進程處理完現(xiàn)有請求再退出。

在實際項目中，根據(jù)需求選擇：需要安全關(guān)閉服務就用優(yōu)雅關(guān)機；需要更新代碼且不影響現(xiàn)網(wǎng)運行就用優(yōu)雅重啟。

到此這篇關(guān)于Go語言實現(xiàn)優(yōu)雅關(guān)機和重啟的示例詳解的文章就介紹到這了,更多相關(guān)Go語言優(yōu)雅關(guān)機和重啟內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: