如何用 Go 实现热重启
![如何用 Go 实现热重启](https://mojotv.cn/assets/image/ginbro_coverage.jpg)
Referred From https://segmentfault.com/a/1190000019790072
热重启
热重启(Zero Downtime),指新老进程无缝切换,在替换过程中可保持对 client 的服务.
原理
- 父进程监听重启信号
- 在收到重启信号后,父进程调用
fork
,同时传递socket
描述符给子进程 - 子进程接收并监听父进程传递的
socket
描述符 - 在子进程启动成功之后,父进程停止接收新连接,同时等待旧连接处理完成(或超时)
- 父进程退出,热重启完成
实现
package main
import (
"context"
"errors"
"flag"
"log"
"net"
"net/http"
"os"
"os/exec"
"os/signal"
"syscall"
"time"
)
var (
server *http.Server
listener net.Listener = nil
graceful = flag.Bool("graceful", false, "listen on fd open 3 (internal use only)")
message = flag.String("message", "Hello World", "message to send")
)
func handler(w http.ResponseWriter, r *http.Request) {
time.Sleep(5 * time.Second)
w.Write([]byte(*message))
}
func main() {
var err error
// 解析参数
flag.Parse()
http.HandleFunc("/test", handler)
server = &http.Server{Addr: ":3000"}
// 设置监听器的监听对象(新建的或已存在的 socket 描述符)
if *graceful {
// 子进程监听父进程传递的 socket 描述符
log.Println("listening on the existing file descriptor 3")
// 子进程的 0, 1, 2 是预留给标准输入,标准输出,错误输出,故传递的 socket 描述符
// 应放在子进程的 3
f := os.NewFile(3, "")
listener, err = net.FileListener(f)
} else {
// 父进程监听新建的 socket 描述符
log.Println("listening on a new file descriptor")
listener, err = net.Listen("tcp", server.Addr)
}
if err != nil {
log.Fatalf("listener error: %v", err)
}
go func() {
err = server.Serve(listener)
log.Printf("server.Serve err: %v\n", err)
}()
// 监听信号
handleSignal()
log.Println("signal end")
}
func handleSignal() {
ch := make(chan os.Signal, 1)
// 监听信号