golang 監聽服務的信號,實現平滑啟動,linux信號說明詳解

package main import ('context''fmt''golang.org/x/sync/errgroup''net/http''os''os/signal''syscall') func main() { g, ctx := errgroup.WithContext(context.Background())fmt.Println('服務啟動start！')addr := ':9091's :=&http.Server{Addr: addr,Handler:http.DefaultServeMux,}g.Go(func() error {http.HandleFunc('/test1', func(writer http.ResponseWriter, request *http.Request) {fmt.Println('tes1')writer.Write([]byte('tes1'))})return s.ListenAndServe()})g.Go(func() error {exit := make(chan os.Signal)//監聽 Ctrl+C 信號signal.Notify(exit, syscall.SIGINT, syscall.SIGTERM)select {case <-exit:fmt.Println('進程已被取消~')return s.Shutdown(ctx)}})err := g.Wait()if err != nil {fmt.Println(err)}fmt.Println('服務啟動成功！')if ctx.Err() !=nil {fmt.Println(ctx.Err())fmt.Println('服務關閉成功！')os.Exit(0)} }

補充：golang http服務實現平滑重啟

package main import ( 'context' 'encoding/json' 'fmt' 'math/rand' 'net/http' 'os' 'os/signal' 'time') var logChan = make(chan map[string]interface{}) var requestStatusMap = map[int]bool{} var done = make(chan bool, 1)var quit = make(chan os.Signal, 1) //為什么這樣可以平滑重啟？// 正常情況下是server.ListenAndServe() 這個位置hang住整個進程的// 可以把這個程序看成兩部分，1個是web服務的監聽部分，一個是處理部分， 如果web服務器不開啟了，那么就不能處理新進來的請求了（可以理解為一個帶路的）// 真正讓這個請求斷掉 是因為主進程（main）被kill// 所以平滑重啟的原理就是，先kill掉web服務器，不讓新的請求進來，等現有的全部請求完了，然后結束當前進程func main() { server := newServer() signal.Notify(quit, os.Interrupt) go monitorKill(server, quit) server.ListenAndServe() <-done} func newServer() *http.Server { router := http.NewServeMux() router.HandleFunc('/hello', sayHello) return &http.Server{Addr: ':8262',Handler: router, }} func monitorKill(server *http.Server, quit <-chan os.Signal) { <-quit go shutDown(server) for {if len(requestStatusMap) != 0 { fmt.Println('目前還有進行中的請求，請稍等') time.Sleep(time.Second * 1) continue} else { close(done) break} }} func shutDown(server *http.Server) { if err := server.Shutdown(context.Background()); err != nil {fmt.Println(err) }} func sayHello(w http.ResponseWriter, r *http.Request) { go WriteInfo()//請求寫日志 var uniqueId = GenerateRangeNum(1, 1000) requestStatusMap[uniqueId] = false url := r.URL.Path query := r.URL.RawQuery method := r.Method a := map[string] interface{}{'url' : url,'method' : method,'query' : query,'response': 'hello world!', } logChan<-a w.Write([]byte('hello world!')) time.Sleep(time.Second * 10) delete(requestStatusMap, uniqueId)} func WriteInfo() { info := <-logChan fileName := '/tmp/weekhomework.log' _, err := os.Stat(fileName) if err != nil || os.IsNotExist(err) {_, _ = os.Create(fileName) } f,err := os.OpenFile(fileName, os.O_WRONLY, 0644) defer f.Close() if err !=nil {fmt.Println(err.Error()) } else {//追加寫入 為什么O_APPEND 模式無法寫入？ todon, _ := f.Seek(0, 2)infostr, _ := json.Marshal(info)_,err=f.WriteAt([]byte(string(infostr) +'n'), n) }} func GenerateRangeNum(min int, max int) int { if min == max {return min } rand.Seed(time.Now().Unix()) randNum := rand.Intn(max-min) + min return randNum}主要思路：

對于每個請求都做記錄，處理完成之后做刪除。 用一個協程去監控中斷信號，有中斷信號先把http服務關閉。

如果這個時候還有請求沒有處理完，那么就輪訓等待，等全部處理完那么就 發出終止信號結束main進程的執行

以上為個人經驗，希望能給大家一個參考，也希望大家多多支持好吧啦網。如有錯誤或未考慮完全的地方，望不吝賜教。