Go项目实现优雅关机与平滑重启功能_Golang

前言

优雅关机就是服务端关机命令发出后不是立即关机，而是等待当前还在处理的请求全部处理完毕后再退出程序，是一种对客户端友好的关机方式。而执行Ctrl+C关闭服务端时，会强制结束进程导致正在访问的请求出现问题。

什么是优雅关机？

优雅关机就是服务端关机命令发出后不是立即关机，而是等待当前还在处理的请求全部处理完毕后再退出程序，是一种对客户端友好的关机方式。而执行Ctrl+C关闭服务端时，会强制结束进程导致正在访问的请求出现问题。

实现原理

Go 1.8版本之后， http.Server 内置的 Shutdown() 方法就支持优雅地关机，说明一下Shutdown工作的机制：当程序检测到中断信号时，我们调用http.server种的shutdown方法，该方法将阻止新的请求进来，同时保持当前的连接，知道当前连接完成则终止程序！

实现优雅重启

				?

									package main

									import (

									    "context"

									    "fmt"

									    "github.com/spf13/viper"

									    "go.uber.org/zap"

									    "log"

									    "net/http"

									    "os"

									    "os/signal"

									    "syscall"

									    "time"

									)

									func main() {

									    //启动服务(优雅关机)

									    srv := &http.Server{

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

									        Handler: r,

									    }

									    go func() {

									        // 开启一个goroutine启动服务

									        if err := srv.ListenAndServe(); err != nil && err != http.ErrServerClosed {

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

									        }

									    }()

									    // 等待中断信号来优雅地关闭服务器，为关闭服务器操作设置一个5秒的超时

									    quit := make(chan os.Signal, 1) // 创建一个接收信号的通道

									    // kill 默认会发送 syscall.SIGTERM 信号

									    // kill -2 发送 syscall.SIGINT 信号，我们常用的Ctrl+C就是触发系统SIGINT信号

									    // kill -9 发送 syscall.SIGKILL 信号，但是不能被捕获，所以不需要添加它

									    // signal.Notify把收到的 syscall.SIGINT或syscall.SIGTERM 信号转发给quit

									    signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM) // 此处不会阻塞

									    <-quit                                               // 阻塞在此，当接收到上述两种信号时才会往下执行

									    zap.L().Info("Shutdown Server ...")

									    // 创建一个5秒超时的context

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

									    defer cancel()

									    // 5秒内优雅关闭服务（将未处理完的请求处理完再关闭服务），超过5秒就超时退出

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

									        zap.L().Fatal("Server Shutdown: ", zap.Error(err))

									    }

									    zap.L().Info("Server exiting")

									}

实现平滑重启

				?

									import (

									    "log"

									    "net/http"

									    "time"

									    "github.com/fvbock/endless"

									    "github.com/gin-gonic/gin"

									)

									func main() {

									    router := gin.Default()

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

									        c.String(http.StatusOK, "hello xiaosheng !")

									    })

									    // 默认endless服务器会监听下列信号：

									    // syscall.SIGHUP，syscall.SIGUSR1，syscall.SIGUSR2，syscall.SIGINT，syscall.SIGTERM和syscall.SIGTSTP

									    // 接收到 SIGHUP 信号将触发`fork/restart` 实现优雅重启（kill -1 pid会发送SIGHUP信号）

									    // 接收到 syscall.SIGINT或syscall.SIGTERM 信号将触发优雅关机

									    // 接收到 SIGUSR2 信号将触发HammerTime

									    // SIGUSR1 和 SIGTSTP 被用来触发一些用户自定义的hook函数

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

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

									    }

									    log.Println("Server exiting...")