暴力的重启服务方案
一般服务器重启可以直接通过 kill 命令杀死进程,然后重新启动一个新的进程即可。但这种方法比较粗暴,有可能导致某些正在处理中的客户端请求失败,如果请求正在写数据,那么还有可能导致数据丢失或者数据不一致等。
那么有什么方式可以优雅的重启服务呢?
优雅的重启服务方案
优雅的重启方式流程如下:
从上面的流程可以看出,旧进程必须等待所有的请求连接完成后才会退出,请求不会被强制关闭,所以是个优雅的重启方式。
使用Go实现优雅重启
下面我们使用Go语言来演示怎么实现优雅启动功能,我们先来看看原理图:
从原理图可以知道,重启时首先通过发送 SIGHUP信号 给服务进程,服务进程收到 SIGHUP信号 后会 fork 一个新进程来处理新的请求,然后新进程会发送 SIGTERM信号 给旧服务进程(父进程),旧服务进程接收到 SIGTERM信号 后会关闭监听的 socket句柄 (停止接收新请求),并且等待未处理完成的请求完成后再退出进程。
下面通过代码来说明这个流程,代码主要参考 endless 这个库,有兴趣可以查看其源码。
首先我们定义一个名为 endlessServer 的结构并且继承 http.Server 结构:
type endlessServer struct {
http.Server
EndlessListener net.Listener
wg sync.WaitGroup
sigChan chan os.Signal
isChild bool
state uint8
lock *sync.RWMutex
}
Go的继承很简单,就是在定义结构时把要继承的结构嵌入到里面就可以了。
这里说明一下 endlessServer 各个成员的作用吧:
Server:用于继承 http.Server 结构 EndlessListener:监听客户端请求的 Listener wg:用于记录还有多少客户端请求没有完成 sigChan:用于接收信号的管道 isChild:用于重启时标志本进程是否是为一个新进程 state:当前进程的状态 lock:用于锁定一些资源定义一个创建 endlessServer 结构的函数:
func NewServer(addr string, handler http.Handler) (srv *endlessServer) {
isChild := os.Getenv("ENDLESS_CONTINUE") != ""
srv = &endlessServer{
wg: sync.WaitGroup{},
sigChan: make(chan os.Signal),
isChild: isChild,
state: STATE_INIT,
lock: &sync.RWMutex{},
}
srv.Server.Addr = addr
srv.Server.ReadTimeout = 0
srv.Server.WriteTimeout = 0
srv.Server.MaxHeaderBytes = 0
srv.Server.Handler = handler
return
}
NewServer() 函数的实现比较简单,就是创建一个 endlessServer 结构,然后初始化其各个成员。要注意的是,是否为新进程是通过读取环境变量 ENDLESS_CONTINUE 来判断的,如果定义了 ENDLESS_CONTINUE 环境变量,就是说当前进程是新的服务进程。









