项目上线后,难免因配置变更或功能迭代需要重启服务。直接使用 kill -9 pid 虽然简单粗暴,但正在处理的请求会因此中断。有没有更优雅、更体面的做法?

阅读本文需要了解 UNIX 系统中信号的基本概念,建议提前查阅相关资料进行预习。
优雅关机
所谓优雅关机,通俗来说,就是当收到关机指令后,不会立即终止进程,而是等待当前正在处理的请求全部完成后再关闭。相比之下,如果使用 Ctrl+C 强制结束,正在访问的请求会直接中断,对用户来说显然不够友好。
如何实现优雅关机?
自 Go 1.8 版本起,http.Server 内置的 Shutdown() 方法便支持优雅关机。下面是具体的实现代码:
// +build go1.8
package main
import (
"context"
"log"
"net/http"
"os"
"os/signal"
"syscall"
"time"
"github.com/gin-gonic/gin"
)
func main() {
router := gin.Default()
router.GET("/", func(c *gin.Context) {
time.Sleep(5 * time.Second)
c.String(http.StatusOK, "Welcome Gin Server")
})
srv := &http.Server{
Addr: ":8080",
Handler: router, // 将router注册到server里面
}
go func() {
// 开启一个goroutine启动服务
// 如果不开启ListenAndServe就会一直接收请求处理请求,代码就不会往下执行
if err := srv.ListenAndServe(); err != nil && err != http.ErrServerClosed {
log.Fatalf("listen: %sn", 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 // 阻塞在此,当接收到上述两种信号时才会往下执行
log.Println("Shutdown Server ...")
// 创建一个5秒超时的context
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
defer cancel()
// 5秒内优雅关闭服务(将未处理完的请求处理完再关闭服务),超过5秒就超时退出
if err := srv.Shutdown(ctx); err != nil {
log.Fatal("Server Shutdown: ", err)
}
log.Println("Server exiting")
}
如何验证其效果呢?将上述代码运行后,本地 8080 端口会启动一个 Web 服务,仅注册了 / 路由,后端会先休眠 5 秒再返回响应。操作步骤如下:
- 在终端编译并执行上述代码;
- 通过浏览器访问
127.0.0.1:8080/,此时页面保持白屏,等待服务端响应; - 在终端中迅速按下
Ctrl+C,发送syscall.SIGINT信号; - 程序不会立即退出,而是等待第 2 步的响应返回后才退出,从而实现了优雅关机。
优雅关机已经实现,那么优雅重启又该如何实现呢?
优雅重启
可以使用 fvbock/endless 库替换默认的 ListenAndServe 来启动服务。示例代码如下:
package main
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) {
time.Sleep(5 * time.Second)
c.String(http.StatusOK, "hello gin!")
})
// 默认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: %sn", err)
}
log.Println("Server exiting")
}
补充说明:
# 将main.go运行起来 go run main.go # 假设你当前在 graceful_restart_demo 目录下执行 go build # 生成的可执行文件叫 graceful_restart_demo # 指定程序名 go build -o graceful_restart # -o 是 output(输出)的缩写,用来指定生成的文件的名字。这里指定了生成的程序名叫 graceful_restart # 直接可以将这个可执行文件运行起来 ./graceful_restart
如何验证优雅重启?通过执行 kill -1 pid 发送 syscall.SIGHUP 信号,通知程序进行优雅重启。具体操作步骤如下:
- 在终端编译执行
go build -o graceful_restart并运行./graceful_restart,记下输出的 PID(假设为 43682); - 将代码中处理请求返回的
hello gin!改为hello q1mi!,再次编译go build -o graceful_restart; - 通过浏览器访问
127.0.0.1:8080/,页面保持白屏等待响应; - 在终端中迅速执行
kill -1 43682,发送syscall.SIGHUP信号; - 等待第 3 步的浏览器收到
hello gin!后,再次访问127.0.0.1:8080/,会收到hello q1mi!的响应。 - 代码替换完成,当前未处理完的请求不受影响,优雅重启成功。
需要注意的是,此时程序的 PID 发生了变化。这是因为 endless 通过 fork 子进程来处理新请求,待原进程处理完当前请求后再退出。因此,如果项目使用类似 supervisor 的进程管理工具,这种方式可能不适用。
无论是优雅关机还是优雅重启,其核心都是监听特定的系统信号,随后执行相应的逻辑,确保当前正在处理的请求正常完成后再关闭进程。具体采用哪种方式、如何实现,需要根据项目的实际需求来决定。
