package main import ( "context" "github.com/gin-gonic/gin" "log" "net/http" "os" "os/signal" "syscall" "time" ) // 实现优雅关机和平滑重启 func main() { router := gin.Default() router.GET("/", func(c *gin.Context) { // 这个10秒的延时。是为了演示操作方便,实际上线一定注释掉 time.Sleep(time.Second * 10) c.String(http.StatusOK, "hello xiaosheng") }) srv := &http.Server{ Addr: ":8080", Handler: router, } // 必须开启一个go routine 因为如果不开起,下面会一直listen and serve,进入死循环 // err != http.ErrServerClosed这个很重要 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.SIGTREN信号 // kill -2发送syscall.SIGINT信号,我们常用的ctrl+c就是触发系统SIGINT信号 // kill -9发送syscall.SIGKILL信号,但是不能被捕获,所以不需要添加他 // signal.Notify把收到的syscall.SIGINT或syscall.SIGTREN信号传给quit signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM) // 此处不会阻塞 <-quit // 阻塞在此,当收到上述两种信号的时候才会往下执行 log.Println("ShutDown Server ...") // 创建一个5秒超时的context ctx, cancel := context.WithTimeout(context.Background(), time.Second*5) defer cancel() // 5秒内优雅关闭服务, (将未处理完的请求处理完再关闭服务), 超过5秒就退出 if err := srv.Shutdown(ctx); err != nil { log.Fatal("shut down:", err) } log.Println("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) { // 这个5秒的延时。是为了演示操作方便,实际上线一定注释掉 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: %s\n", err) } log.Println("Server exiting...")
服务端关机命令发出后不立即关机,而是等待当前还在处理的请求全部处理完毕后再退出程序,这样是一种对客户端友好的关机方式。
执行Ctrl+C时,会强制结束进程,这样导致正在访问的请求出现问题。
Go 1.8
版本之后, http.Server
内置的 Shutdown()
方法就支持优雅关机
8080端口开启了一个web服务,并且只注册了一条路由,“/”, 但客户端访问127.0.0.1:8080/
时,过10秒才会响应,如果这时我们按下ctrl+c,给程序发送syscall.SIGINT信号,他会等待10秒将当前请求处理完,他才会消亡,当然也取决于创建的5秒的context超时时间。
可以使用fvbock/endless
来替换默认的 ListenAndServe
启动服务来实现
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) { // 这个5秒的延时。是为了演示操作方便,实际上线一定注释掉 time.Sleep(5 * time.Second) 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...")
go build -o graceful_restart
编译,并执行./graceful_restart
,终端输出当前pid(假设为44444)hello xiaosheng!
修改为hello world!
,再次编译go build -o graceful_restart
127.0.0.1:8080/
,此时客户端浏览器等待服务端返回响应。kill -1 44444
命令给程序发送syscall.SIGHUP
重启信号hello xiaosheng!
后,再次访问127.0.0.1:8080/
会收到hello world!
的响应。这样做在不影响当前未处理完请求的同时完成了程序代码的替换,实现了平滑重启。但实际上用的不多,因为实际都是多台服务器,或者说有类似supervisor
的软件管理进程时就不适用这种方式,因为他进程pid变了,他自己重启和supervisor
的软件管理进程给他重启就冲突了。