published on in Tech
tags: go graceful http

Golang 服务优雅重启

很多时候对于服务升级的做法简单粗暴, 就简单的杀进程启动新的进程.

还有的好一点就是多个相同的服务依次升级, 保证有服务可用. 但是公平的说这两种都会丢失请求中的连接.

鉴于这种情况, 在现实中我们可以使用优雅重启来搞定这个问题. Golang 实现优雅重启的原理也很简单:

  • 监听 USR2 信号;

  • 收到信号后将服务监听的文件描述符传递给新的子进程;

  • 此时新老进程同时接收请求;

  • 父进程停止接收新请求, 等待旧请求完成(或超时);

  • 父进程退出.

对于上面的原理看似简单, 其实是分成了两个大的要点:

  • 新老进程同时监听同一端口, 这个很简单, Go 很早旧支持;

  • 如何等待旧的请求完成, 这个在 Go 1.8 (新增了Server.Shutdown) 之前是需要费一番功夫的.

我们搞定了上面的原理之后, 加上 Go 1.8 的完美等待旧请求的实现,

我实现了一个简单的优雅重启库: https://github.com/douglarek/zerodown.

zerodown 完美兼容基于 Go 标准库 Server 监听服务.

对于标准库的使用, 我们可以象下面一样使用:

package main

import (
	"fmt"
	"log"
	"net/http"
	"time"

	"github.com/douglarek/zerodown"
)

func main() {
	http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
		time.Sleep(5 * time.Second)
		fmt.Fprintln(w, "Hello, World!")
	})
	log.Fatalln(zerodown.ListenAndServe(":8080", nil))

对于第三方库 Gin 我们可以:

package main

import (
	"log"
	"net/http"
	"time"

	"github.com/douglarek/zerodown"

	"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, World!")
	})
	log.Fatalln(zerodown.ListenAndServe(":8080", router))
}