第Go項(xiàng)目實(shí)現(xiàn)優(yōu)雅關(guān)機(jī)與平滑重啟功能目錄前言什么是優(yōu)雅關(guān)機(jī)？實(shí)現(xiàn)原理實(shí)現(xiàn)優(yōu)雅重啟實(shí)現(xiàn)平滑重啟測(cè)試總結(jié)

前言

優(yōu)雅關(guān)機(jī)就是服務(wù)端關(guān)機(jī)命令發(fā)出后不是立即關(guān)機(jī)，而是等待當(dāng)前還在處理的請(qǐng)求全部處理完畢后再退出程序，是一種對(duì)客戶(hù)端友好的關(guān)機(jī)方式。而執(zhí)行Ctrl+C關(guān)閉服務(wù)端時(shí)，會(huì)強(qiáng)制結(jié)束進(jìn)程導(dǎo)致正在訪(fǎng)問(wèn)的請(qǐng)求出現(xiàn)問(wèn)題。

什么是優(yōu)雅關(guān)機(jī)？

優(yōu)雅關(guān)機(jī)就是服務(wù)端關(guān)機(jī)命令發(fā)出后不是立即關(guān)機(jī)，而是等待當(dāng)前還在處理的請(qǐng)求全部處理完畢后再退出程序，是一種對(duì)客戶(hù)端友好的關(guān)機(jī)方式。而執(zhí)行Ctrl+C關(guān)閉服務(wù)端時(shí)，會(huì)強(qiáng)制結(jié)束進(jìn)程導(dǎo)致正在訪(fǎng)問(wèn)的請(qǐng)求出現(xiàn)問(wèn)題。

實(shí)現(xiàn)原理

Go1.8版本之后，http.Server內(nèi)置的Shutdown()方法就支持優(yōu)雅地關(guān)機(jī)，說(shuō)明一下Shutdown工作的機(jī)制：當(dāng)程序檢測(cè)到中斷信號(hào)時(shí)，我們調(diào)用http.server種的shutdown方法，該方法將阻止新的請(qǐng)求進(jìn)來(lái)，同時(shí)保持當(dāng)前的連接，知道當(dāng)前連接完成則終止程序！

實(shí)現(xiàn)優(yōu)雅重啟

packagemain

import(

"context"

"fmt"

"/spf13/viper"

"/zap"

"log"

"net/http"

"os"

"os/signal"

"syscall"

"time"

funcmain(){

//啟動(dòng)服務(wù)(優(yōu)雅關(guān)機(jī))

srv:=http.Server{

Addr:fmt.Sprintf(":%d",viper.GetInt("app.port")),

Handler:r,

gofunc(){

//開(kāi)啟一個(gè)goroutine啟動(dòng)服務(wù)

iferr:=srv.ListenAndServe();err!=nilerr!=http.ErrServerClosed{

log.Fatalf("listen:%s\n",err)

//等待中斷信號(hào)來(lái)優(yōu)雅地關(guān)閉服務(wù)器，為關(guān)閉服務(wù)器操作設(shè)置一個(gè)5秒的超時(shí)

quit:=make(chanos.Signal,1)//創(chuàng)建一個(gè)接收信號(hào)的通道

//kill默認(rèn)會(huì)發(fā)送syscall.SIGTERM信號(hào)

//kill-2發(fā)送syscall.SIGINT信號(hào)，我們常用的Ctrl+C就是觸發(fā)系統(tǒng)SIGINT信號(hào)

//kill-9發(fā)送syscall.SIGKILL信號(hào)，但是不能被捕獲，所以不需要添加它

//signal.Notify把收到的syscall.SIGINT或syscall.SIGTERM信號(hào)轉(zhuǎn)發(fā)給quit

signal.Notify(quit,syscall.SIGINT,syscall.SIGTERM)//此處不會(huì)阻塞

-quit//阻塞在此，當(dāng)接收到上述兩種信號(hào)時(shí)才會(huì)往下執(zhí)行

zap.L().Info("ShutdownServer...")

//創(chuàng)建一個(gè)5秒超時(shí)的context

ctx,cancel:=context.WithTimeout(context.Background(),5*time.Second)

defercancel()

//5秒內(nèi)優(yōu)雅關(guān)閉服務(wù)（將未處理完的請(qǐng)求處理完再關(guān)閉服務(wù)），超過(guò)5秒就超時(shí)退出

iferr:=srv.Shutdown(ctx);err!=nil{

zap.L().Fatal("ServerShutdown:",zap.Error(err))

zap.L().Info("Serverexiting")

}

實(shí)現(xiàn)平滑重啟

import(

"log"

"net/http"

"time"

"/fvbock/endless"

"/gin-gonic/gin"

funcmain(){

router:=gin.Default()

router.GET("/",func(c*gin.Context){

c.String(http.StatusOK,"helloxiaosheng!")

//默認(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-1pid會(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ā)一些用戶(hù)自定義的hook函數(shù)

iferr:=endless.ListenAndServe(":8080",router);err!=nil{

log.Fatalf("listen:%s\n",err)

log.Println("Serverexiting...")

測(cè)試

我們通過(guò)執(zhí)行kill-1pid命令發(fā)送syscall.SIGINT來(lái)通知程序優(yōu)雅重啟，具體做法如下：

打開(kāi)終端，gobuild-ograceful_restart編譯并執(zhí)行./graceful_restart,終端輸出當(dāng)前pid(假設(shè)為43682)將代碼中處理請(qǐng)求函數(shù)返回的hellogin!修改為helloq1mi!，再次編譯gobuild-ograceful_restart打開(kāi)一個(gè)瀏覽器，訪(fǎng)問(wèn):8080/，此時(shí)瀏覽器白屏等待服務(wù)端返回響應(yīng)。在終端迅速執(zhí)行kill-143682命令給程序發(fā)送syscall.SIGHUP信號(hào)等第3步瀏覽器收到響應(yīng)信息hellogin!后再次訪(fǎng)問(wèn):8080/會(huì)收到helloq1mi!的響應(yīng)。在不影響當(dāng)前未處理完請(qǐng)求的同時(shí)完成了程序代碼的替換，實(shí)現(xiàn)了優(yōu)雅重啟。

但是需要注意的是，此時(shí)程序的PID變化了，因?yàn)閑ndless是通過(guò)fork子進(jìn)程處理新請(qǐng)求，待原進(jìn)程處理完當(dāng)前請(qǐng)求后再退出的方式實(shí)現(xiàn)優(yōu)雅重啟的。所以當(dāng)你的項(xiàng)目是使用類(lèi)似su