go如何優(yōu)雅關(guān)閉Graceful?Shutdown服務(wù)
Shutdown方法
Go1.8之后有了Shutdown方法，用來優(yōu)雅的關(guān)閉（Graceful Shutdown）服務(wù)。
func (srv *Server) Shutdown(ctx context.Context) error
這個方法會在調(diào)用時進(jìn)行下述操作：
1.關(guān)閉所有open listeners
2.關(guān)閉所有idle connections
3.無限期等待connections回歸idle狀態(tài)
4.之后關(guān)閉服務(wù)
注：3的無限期等待可以用context的超時來解決。
RegisterOnShutdown方法
Go1.9之后有了RegisterOnShutdown方法，用來在優(yōu)雅的關(guān)閉服務(wù)的同時處理一些退出任務(wù)。
func (srv *Server) RegisterOnShutdown(f func())
關(guān)于RegisterOnShutdown如何生效，我們看代碼。
type Server struct {
...
onShutdown []func()
}
func (srv *Server) RegisterOnShutdown(f func()) {
srv.mu.Lock()
srv.onShutdown = append(srv.onShutdown, f)
srv.mu.Unlock()
}
func (srv *Server) Shutdown(ctx context.Context) error {
...
srv.mu.Lock()
...
for _, f := range srv.onShutdown {
go f()
}
srv.mu.Unlock()
...
}由代碼可知，在通過RegisterOnShutdown進(jìn)行注冊之后，這些函數(shù)被放入一個函數(shù)切片中，并在Shutdown被調(diào)用的時候，為每個函數(shù)啟動了一個goroutine進(jìn)行處理。
但由于沒有進(jìn)行后續(xù)處理，于是這里有個小問題，Shutdown是不等待這些處理函數(shù)結(jié)束的，就可能比這些處理函數(shù)先完成并進(jìn)而結(jié)束程序。
package main
import (
"context"
"log"
"net/http"
"os"
"os/signal"
"time"
)
func main() {
var srv http.Server
srv.RegisterOnShutdown(func() {
time.Sleep(3 * time.Second)
log.Println("Exit func")
})
idleConnsClosed := make(chan struct{})
go func() {
sigint := make(chan os.Signal, 1)
signal.Notify(sigint, os.Interrupt)
<-sigint
// We received an interrupt signal, shut down.
if err := srv.Shutdown(context.Background()); err != nil {
// Error from closing listeners, or context timeout:
log.Printf("HTTP server Shutdown: %v", err)
}
close(idleConnsClosed)
}()
if err := srv.ListenAndServe(); err != http.ErrServerClosed {
// Error starting or closing listener:
log.Fatalf("HTTP server ListenAndServe: %v", err)
}
<-idleConnsClosed
}運行時Ctrl+c之后，服務(wù)立刻結(jié)束了，并未等待log.Println("Exit func")的輸出。
sync.WaitGroup處理退出函數(shù)
為了解決這個問題我們使用sync.WaitGroup來處理這些退出函數(shù)，于是我們有了下面的代碼（解耦并增加了一些日志）：
package main
import (
"context"
"log"
"net/http"
"os"
"os/signal"
"sync"
"time"
)
func main() {
var srv http.Server
var wg sync.WaitGroup
register := func(f func()) {
wg.Add(1)
log.Println("Exit func register")
srv.RegisterOnShutdown(func() {
defer wg.Done()
f()
log.Println("Exit func done")
})
}
register(func() {
time.Sleep(3 * time.Second)
log.Println("Called on Shutdown")
})
idleConnsClosed := make(chan struct{})
go func() {
sigint := make(chan os.Signal, 1)
signal.Notify(sigint, os.Interrupt)
<-sigint
// We received an interrupt signal, shut down.
log.Println("Shutdown start")
if err := srv.Shutdown(context.Background()); err != nil {
// Error from closing listeners, or context timeout:
log.Printf("HTTP server Shutdown: %v", err)
}
close(idleConnsClosed)
}()
if err := srv.ListenAndServe(); err != http.ErrServerClosed {
// Error starting or closing listener:
log.Fatalf("HTTP server ListenAndServe: %v", err)
}
<-idleConnsClosed
log.Println("Shutdown finish")
wg.Wait()
}我們在RegisterOnShutdown時使用了time.Sleep(3 * time.Second)來模擬一個長時間的退出程序，并且在"Shutdown finish"之后進(jìn)行等待，等待退出程序完成任務(wù)。
執(zhí)行日志如下：
2023/05/12 16:48:53 Exit func register
^C2023/05/12 16:48:54 Shutdown start
2023/05/12 16:48:54 Shutdown finish
2023/05/12 16:48:57 Called on Shutdown
2023/05/12 16:48:57 Exit func done
可以看到這個服務(wù)本身很輕，Shutdown start之后立刻就finish并來到wg.Wait()，并等待register的函數(shù)完成之后再退出程序。
另外，如果在服務(wù)運行的過程中需要啟動一些其他的任務(wù)，也可以使用
wg.Add(1)
go func() {
defer wg.Done()
...
}來保證任務(wù)在服務(wù)Shutdown之后可以順利完成。
以上就是go如何優(yōu)雅關(guān)閉Graceful Shutdown服務(wù)的詳細(xì)內(nèi)容，更多關(guān)于go關(guān)閉Graceful Shutdown的資料請關(guān)注腳本之家其它相關(guān)文章！
相關(guān)文章
這篇文章主要來和大家探討一樣如何在Go語言中實現(xiàn)和使用堆棧，以及堆棧如何遵循先進(jìn)先出 (FIFO) 原則，文中的示例代碼簡潔易懂，需要的可以參考一下2023-07-07
Golang學(xué)習(xí)筆記之安裝Go1.15版本(win/linux/macos/docker安裝)
這篇文章主要介紹了Golang學(xué)習(xí)筆記之安裝Go1.15版本(win/linux/macos/docker安裝),本文給大家介紹的非常詳細(xì)，對大家的學(xué)習(xí)或工作具有一定的參考借鑒價值，需要的朋友可以參考下2020-12-12
本文主要介紹了一文掌握Golang模糊測試，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2023-06-06
這篇文章主要為大家介紹了go reflect要不要傳指針原理詳解，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進(jìn)步，早日升職加薪2023-01-01
這篇文章主要介紹了go日志庫中的logrus主要包括go日志庫logrus的安裝和使用，本文通過實例代碼給大家介紹的非常詳細(xì),需要的朋友可以參考下2022-08-08
golang實現(xiàn)整型和字節(jié)數(shù)組之間的轉(zhuǎn)換操作
這篇文章主要介紹了golang實現(xiàn)整型和字節(jié)數(shù)組之間的轉(zhuǎn)換操作，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2020-12-12
go使用SQLX操作MySQL數(shù)據(jù)庫的教程詳解
sqlx 是 Go 語言中一個流行的操作數(shù)據(jù)庫的第三方包,它提供了對 Go 標(biāo)準(zhǔn)庫 database/sql 的擴(kuò)展,簡化了操作數(shù)據(jù)庫的步驟,下面我們就來學(xué)習(xí)一下go如何使用SQLX實現(xiàn)MySQL數(shù)據(jù)庫的一些基本操作吧2023-11-11
在Go語言中,作用域是指變量的可見性范圍,它定義了變量在程序中的生命周期和可訪問性,本文主要介紹了Go作用域的理解,包括局部作用域、全局作用域、命名空間作用域等,感興趣的可以了解一下2023-11-11
go?mod?tidy報錯:zip:?not?a?valid?zip?file解決辦法
這篇文章主要給大家介紹了關(guān)于go?mod?tidy報錯:zip:?not?a?valid?zip?file的解決辦法,go mod是進(jìn)行代碼管理，這錯誤是因為本地分支和遠(yuǎn)程分支沖突,本文通過代碼介紹的非常詳細(xì),需要的朋友可以參考下2024-01-01
今天在刷leetcode的時候,第113題讓我遇到了一個Go語言中append函數(shù)的一個坑,所以復(fù)習(xí)下,這篇文章主要給大家介紹了關(guān)于Golang中Append()使用的相關(guān)資料,需要的朋友可以參考下2023-01-01