Go?Web開發(fā)之Gin多服務配置及優(yōu)雅關閉平滑重啟實現(xiàn)方法
更新時間：2024年01月31日 10:15:39 作者：WeiyiGeek?全棧工程師
這篇文章主要為大家介紹了Go?Web開發(fā)之Gin多服務配置及優(yōu)雅關閉平滑重啟實現(xiàn)方法詳解,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進步,早日升職加薪
如何自定義Gin服務配置及其啟動多個服務?
描述: 在Gin的生產(chǎn)環(huán)境中通常會自定義HTTP配置以達到最優(yōu)性能，此處我們簡單一下 Server 結構體中可配置的參數(shù)項。
// A Server defines parameters for running an HTTP server.
// The zero value for Server is a valid configuration.
type Server struct {
// 配置監(jiān)聽地址:端口，默認是:8080
Addr string
// 要調用的處理程序，http.DefaultServeMux如果為nil
Handler Handler
// 如果為true，則將“OPTIONS*”請求傳遞給Handler
DisableGeneralOptionsHandler bool
// 提供TLS配置
TLSConfig *tls.Config
//讀取整個請求（包括正文）的最長持續(xù)時間。
ReadTimeout time.Duration
// 讀取整請求（Header）的最長持續(xù)時間。
ReadHeaderTimeout time.Duration
// 超時寫入響應之前的最長持續(xù)時間
WriteTimeout time.Duration
// 啟用保持活動時等待下一個請求的最長時間
IdleTimeout time.Duration
// 控制服務器解析請求標頭的鍵和值（包括請求行）時讀取的最大字節(jié)數(shù) (通常情況下不進行設置)
MaxHeaderBytes int
// 在發(fā)生ALPN協(xié)議升級時接管所提供TLS連接的所有權。
TLSNextProto map[string]func(*Server, *tls.Conn, Handler)
// 指定了一個可選的回調函數(shù)，當客戶端連接更改狀態(tài)時調用該函數(shù)
ConnState func(net.Conn, ConnState)
// 為接受連接的錯誤、處理程序的意外行為以及潛在的FileSystem錯誤指定了一個可選的記錄器
ErrorLog *log.Logger
// 返回/此服務器上傳入請求的基本上下文
BaseContext func(net.Listener) context.Context
// 指定一個函數(shù)來修改用于新連接c的上下
ConnContext func(ctx context.Context, c net.Conn) context.Context
// 當服務器處于關閉狀態(tài)時為true
inShutdown atomic.Bool
disableKeepAlives atomic.Bool
nextProtoOnce sync.Once // guards setupHTTP2_* init
nextProtoErr error // result of http2.ConfigureServer if used
mu sync.Mutex
listeners map[*net.Listener]struct{}
activeConn map[*conn]struct{}
onShutdown []func()
listenerGroup sync.WaitGroup
}模塊更新
go get -u golang.org/x/sync/errgroup go mod tidy
示例代碼:
package main
import (
"log"
"net/http"
"time"
"github.com/gin-gonic/gin"
"golang.org/x/sync/errgroup"
)
// 處理屬于同一總體任務的子任務的goroutine的集合
var (
g errgroup.Group
)
// s2 Gin 服務的 Handler
func router02() http.Handler {
e := gin.New()
e.Use(gin.Recovery())
e.GET("/", func(c *gin.Context) {
c.JSON(
http.StatusOK,
gin.H{
"code": http.StatusOK,
"msg": "Welcome server 02 blog.weiyigeek.top",
},
)
})
return e
}
func main() {
// Default返回一個Engine實例，該實例已連接Logger和Recovery中間件。
router := gin.Default()
// Gin 服務s1.用于運行HTTP服務器的參數(shù) （常規(guī)參數(shù)）
s1 := &http.Server{
// Gin運行的監(jiān)聽端口
Addr: ":8080",
// 要調用的處理程序，http.DefaultServeMux如果為nil
Handler: router,
// ReadTimeout是讀取整個請求（包括正文）的最長持續(xù)時間。
ReadTimeout: 5 * time.Second,
// WriteTimeout是超時寫入響應之前的最長持續(xù)時間
WriteTimeout: 10 * time.Second,
// MaxHeaderBytes控制服務器解析請求標頭的鍵和值（包括請求行）時讀取的最大字節(jié)數(shù) (通常情況下不進行設置)
MaxHeaderBytes: 1 << 20,
}
// Go在一個新的goroutine中調用給定的函數(shù),此處將Go語言的并發(fā)體現(xiàn)的淋漓盡致。
g.Go(func() error {
return s1.ListenAndServe()
})
// 配置Gin中間件
// Recovery返回一個中間件，該中間件可以從任何exception中恢復，并在出現(xiàn)exception時寫入500。
router.Use(gin.Recovery())
// 服務s1的路由
router.GET("/", func(c *gin.Context) {
c.JSON(
http.StatusOK,
gin.H{
"code": http.StatusOK,
"msg": "Welcome server 01 www.weiyigeek.top",
},
)
})
// Gin 服務s1.定義了不同的監(jiān)聽端口以及Handler
s2 := &http.Server{
Addr: ":8081",
Handler: router02(),
ReadTimeout: 5 * time.Second,
WriteTimeout: 10 * time.Second,
}
g.Go(func() error {
return s2.ListenAndServe()
})
if err := g.Wait(); err != nil {
log.Fatal(err)
}
}執(zhí)行結果:
如何優(yōu)雅的關閉或者重啟Gin應用程序?
1.使用 chan 通道監(jiān)聽中斷信號（SIGINT和SIGTERM）
描述: 在Go Gin中，可以使用以下代碼實現(xiàn)優(yōu)雅地重啟或停止, 確保所有連接都被正確關閉，避免數(shù)據(jù)丟失或損壞。
代碼示例:
package main
import (
"context"
"log"
"net/http"
"os"
"os/signal"
"syscall"
"time"
"github.com/gin-gonic/gin"
)
func main() {
// 創(chuàng)建 Gin 實例
router := gin.Default()
// 添加路由
router.GET("/", func(c *gin.Context) {
c.String(http.StatusOK, "Hello, World! weiyigeek.top")
})
// 創(chuàng)建 HTTP Server
srv := &http.Server{
Addr: ":8080",
Handler: router,
}
// 開啟一個goroutine啟動服務 啟動 HTTP Server
go func() {
if err := srv.ListenAndServe(); err != nil && err != http.ErrServerClosed {
log.Fatalf("listen: %s\n", err)
}
}()
// 等待中斷信號
quit := make(chan os.Signal)
// kill 默認會發(fā)送 syscall.SIGTERM 信號
// kill -2 發(fā)送 syscall.SIGINT 信號，我們常用的Ctrl+C就是觸發(fā)系統(tǒng)SIGINT信號
// kill -9 發(fā)送 syscall.SIGKILL 信號，但是不能被捕獲，所以不需要添加它
// signal.Notify把收到的 syscall.SIGINT或syscall.SIGTERM 信號轉發(fā)給quit
signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM) // 此處不會阻塞
<-quit // 阻塞在此，當接收到上述兩種信號時才會往下執(zhí)行
log.Println("Shutdown Server ...")
// 創(chuàng)建一個 5 秒的超時上下文
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
defer cancel()
// 關閉 HTTP Server
// // 5秒內優(yōu)雅關閉服務（將未處理完的請求處理完再關閉服務），超過5秒就超時退出
if err := srv.Shutdown(ctx); err != nil {
log.Fatal("Server Shutdown:", err)
}
log.Println("Server exiting")
}代碼解析:
首先創(chuàng)建了一個Gin實例和一個HTTP Server，然后啟動HTTP Server。接下來，使用signal.Notify()函數(shù)監(jiān)聽中斷信號（SIGINT和SIGTERM）,當接收到中斷信號時，服務器會進入優(yōu)雅關閉流程，即先關閉HTTP Server，然后等待5秒鐘，最后退出程序。
在關閉HTTP Server時，我們使用了srv.Shutdown()函數(shù)，它會優(yōu)雅地關閉HTTP Server并等待所有連接關閉。如果在5秒鐘內沒有關閉完所有連接，函數(shù)會返回錯誤。
知識補充:
使用os/signal包實現(xiàn)對信號的處理, 最常見的信號列表。
2.使用 os/exec 包來執(zhí)行Gin平滑重啟
描述: 在Linux的Go-gin環(huán)境中我們可以使用 os/exec 包來執(zhí)行重啟命令，然后在 Gin 中定義一個路由，使得訪問該路由時會執(zhí)行重啟命令。
代碼示例:
package main
import (
"fmt"
"net/http"
"os"
"os/exec"
"github.com/gin-gonic/gin"
)
func main() {
r := gin.Default()
// 重啟的路由 /restart
r.GET("/restart", func(c *gin.Context) {
cmd := exec.Command("killall", "-HUP", "appweiyigeek")
err := cmd.Run()
if err != nil {
fmt.Println("Error executing restart command:", err)
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to restart Gin server."})
return
}
c.JSON(http.StatusOK, gin.H{"message": "Gin server restarted successfully."})
})
r.Run(":8080")
}編譯執(zhí)行:
go build ./main.go -o appweiyigeek ./appweiyigeek
在上面的例子中，我們定義了一個路由 /restart，當訪問該路由時，它會執(zhí)行 killall -HUP appweiyigeek 命令來重啟 Gin 服務, 這里的appweiyigeek應該替換為你實際的 Gin 應用程序的名稱。
溫馨提示: 此種重啟方式可能會導致請求失敗或者超時，因為它會強制關閉正在處理的連接, 如果你需要更加優(yōu)雅的重啟方式，可以考慮使用優(yōu)雅重啟的方式。
3.使用 fvbock/endless 包實現(xiàn)訪問指定路由平滑重啟Gin服務
描述: 由于endless在windows環(huán)境是不支持，所以博主針對下述代碼在Linux環(huán)境下載并編譯成二進制文件打包到Linux環(huán)境運行進行驗證。
依賴下載:
go get -u github.com/fvbock/endless go mod tidy
代碼示例:
package main
import (
"fmt"
"log"
"net/http"
"os/exec"
"strconv"
"syscall"
"github.com/fvbock/endless"
"github.com/gin-gonic/gin"
)
func main() {
pid := syscall.Getpid()
// 1.默認的Gin引擎
router := gin.Default()
// 傳統(tǒng)方式
// server := &http.Server{
// Addr: ":8080",
// Handler: router,
// ReadTimeout: 5 * time.Second,
// WriteTimeout: 10 * time.Second,
// }
// 2.獲取 Pid
router.GET("/pid", func(c *gin.Context) {
pid = syscall.Getpid()
fmt.Println("Pid:", pid)
c.JSON(http.StatusOK,
gin.H{
"code": http.StatusOK,
"msg": fmt.Sprintf("Gin Server Pid -> %d.", pid),
})
})
// 3.重啟 Gin 服務
router.POST("/restart", func(c *gin.Context) {
pid = syscall.Getpid()
fmt.Println("Restarting Gin Server.......", pid)
err := exec.Command("kill", "-1", strconv.Itoa(pid)).Run()
if err != nil {
fmt.Println("Error executing restart command:", err)
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to restart Gin server."})
return
}
c.JSON(http.StatusOK, gin.H{"message": "Gin server restarted successfully.", "pid": pid})
})
// 4.使用endless偵聽TCP網(wǎng)絡地址addr，然后使用處理程序調用Serve來處理傳入連接上的請求
err := endless.ListenAndServe(":8080", router)
if err != nil || err != http.ErrServerClosed {
log.Println("err:", err)
}
// 5.引入了endless擴展，將原本的Run方式啟動項目改成了ListenAndServe方式所有此處主席掉
// router.Run(":8080")
}編譯構建:
# 切換編譯在Linux平臺的64位可執(zhí)行程序環(huán)境 go env -w CGO_ENABLED=0 GOOS=linux GOARCH=amd64 # 編譯 go build -o endless-test-1 .\main.go # 執(zhí)行驗證 chmod +x endless-test-1 nohup ./endless-test-1 & [1] 1147978
執(zhí)行效果:
# GET 請求 10.20.176.101:8080/pid
# POST 請求 10.20.176.101:8080/restart
請求restart后可以看見go-gin已經(jīng)平滑重啟了是不是很方便，效果如下。
以上就是Go Web開發(fā)之Gin多服務配置及優(yōu)雅關閉平滑重啟實現(xiàn)方法的詳細內容，更多關于Go Web Gin多服務配置的資料請關注腳本之家其它相關文章！
相關文章
golang 檢查網(wǎng)絡狀態(tài)是否正常的方法
今天小編就為大家分享一篇golang 檢查網(wǎng)絡狀態(tài)是否正常的方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2019-07-07
Go?語言?net/http?包使用之HTTP?服務器、客戶端與中間件詳解
Go 語言標準庫中的net/http包十分的優(yōu)秀,提供了非常完善的 HTTP 客戶端與服務端的實現(xiàn),僅通過幾行代碼就可以搭建一個非常簡單的 HTTP 服務器,本文給大家介紹Go語言net/http包使用之HTTP服務器、客戶端與中間件的操作,感興趣的朋友一起看看吧2025-05-05
這篇文章主要為大家介紹了如何優(yōu)雅的管理Go?Project生命周期，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進步，早日升職加薪2023-03-03
今天給大家推薦一款使用Go語言編寫的流量回放工具?--?goreplay；工作中你一定遇到過需要在服務器上抓包的場景，有了這個工具就可以助你一臂之力，廢話不多，我們接下來來看一看這個工具2022-09-09
這篇文章主要給大家介紹了關于VSCode安裝go相關插件失敗的簡單解決方案,VSCode是我們開發(fā)go程序的常用工具,最近安裝的時候遇到了些問題,需要的朋友可以參考下2023-07-07
這篇文章主要給大家介紹了關于golang如何實現(xiàn)proxy代理簡單方法的相關資料,Proxy是golang實現(xiàn)的高性能http,https,websocket,tcp,udp,socks5,ss代理服務器,文中通過代碼介紹的非常詳細,需要的朋友可以參考下2023-10-10
DuckDB是一個嵌入式SQL數(shù)據(jù)庫引擎,它與眾所周知的SQLite非常相似,但它是為olap風格的工作負載設計的,DuckDB支持各種數(shù)據(jù)類型和SQL特性,憑借其在以內存為中心的環(huán)境中處理高速分析的能力,它迅速受到數(shù)據(jù)科學家和分析師的歡迎,在這篇博文中,我們將探索在Go中使用DuckDB2025-01-01
Go prometheus metrics條目自動回收與清理方法
這篇文章主要為大家介紹了Go prometheus metrics條目自動回收與清理方法詳解，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進步，早日升職加薪2022-11-11
Token 認證機制的核心思想是,服務端在用戶登錄時生成一個 Token,客戶端在后續(xù)的請求中攜帶這個 Token,服務端通過驗證 Token 的有效性來確認用戶的身份,本文將帶你深入探索基于 Token 的登錄流程,這是一種更為靈活且適用于現(xiàn)代應用架構的認證方式2024-03-03
這篇文章主要為大家詳細介紹了Go語言中的監(jiān)視器模式與配置熱更新的相關知識,文中的示例代碼講解詳細,感興趣的小伙伴可以跟隨小編一起學習一下2024-03-03