go如何热重启服务端进程

作者: adm 分类: go 发布时间: 2022-05-27

热重启方式
使用负载均衡等技术分批次灰度机器
使用fork系统调用来实现平滑重启
go的平滑重启实现
具体原理就是子进程与父进程共享文件描述符,当开启子进程时文件描述符引用计数增加,父进程结束了引用计数减一

父进程调用close()并不会引发TCP连接断开从而使得子进程依旧可以处理来自客户端的请求

package main

import (
    "context"
    "errors"
    "flag"
    "log"
    "net"
    "net/http"
    "os"
    "os/exec"
    "os/signal"
    "syscall"
    "time"
)

var (
    server   *http.Server
    listener net.Listener
    graceful = flag.Bool("graceful", false, "listen on fd open 3 (internal use only)")
)

func handler(w http.ResponseWriter, r *http.Request) {
    time.Sleep(20 * time.Second)
    w.Write([]byte("hello world233333!!!!"))
}

func main() {
    flag.Parse()

    http.HandleFunc("/hello", handler)
    server = &http.Server{Addr: ":9999"}

    var err error
    if *graceful {
        log.Print("main: Listening to existing file descriptor 3.")
        // cmd.ExtraFiles: If non-nil, entry i becomes file descriptor 3+i.
        // when we put socket FD at the first entry, it will always be 3(0+3)
        f := os.NewFile(3, "")
        listener, err = net.FileListener(f)
    } else {
        log.Print("main: Listening on a new file descriptor.")
        listener, err = net.Listen("tcp", server.Addr)
    }

    if err != nil {
        log.Fatalf("listener error: %v", err)
    }

    go func() {
        // server.Shutdown() stops Serve() immediately, thus server.Serve() should not be in main goroutine
        err = server.Serve(listener)
        log.Printf("server.Serve err: %v\n", err)
    }()
    signalHandler()
    log.Printf("signal end")
}

func reload() error {
    tl, ok := listener.(*net.TCPListener)
    if !ok {
        return errors.New("listener is not tcp listener")
    }

    f, err := tl.File()
    if err != nil {
        return err
    }

    args := []string{"-graceful"}
//这里利用exec启动一个新的进程
    cmd := exec.Command(os.Args[0], args...)
    cmd.Stdout = os.Stdout
    cmd.Stderr = os.Stderr
    // put socket FD at the first entry
    //这里启动的子进程会继承父进程的文件描述符
    cmd.ExtraFiles = []*os.File{f}
    return cmd.Start()
}

func signalHandler() {
    ch := make(chan os.Signal, 1)
    signal.Notify(ch, syscall.SIGINT, syscall.SIGTERM, syscall.SIGUSR2)
    for {
        sig := <-ch
        log.Printf("signal: %v", sig)

        // timeout context for shutdown
        ctx, _ := context.WithTimeout(context.Background(), 20*time.Second)
        switch sig {
        case syscall.SIGINT, syscall.SIGTERM:
            // stop
            log.Printf("stop")
            signal.Stop(ch)
            server.Shutdown(ctx)
            log.Printf("graceful shutdown")
            return
        case syscall.SIGUSR2:
            // reload
            log.Printf("reload")
            err := reload()
            if err != nil {
                log.Fatalf("graceful restart error: %v", err)
            }
            server.Shutdown(ctx)
            log.Printf("graceful reload")
            return
        }
    }
}


➜  go curl http://localhost:9999/hello
hello world233333!!!!#

2021/03/23 16:00:39 main: Listening on a new file descriptor.
2021/03/23 16:01:16 signal: user defined signal 2
2021/03/23 16:01:16 reload
2021/03/23 16:01:16 server.Serve err: http: Server closed
2021/03/23 16:01:16 main: Listening to existing file descriptor 3.
2021/03/23 16:01:17 graceful reload
2021/03/23 16:01:17 signal end
➜  go 2021/03/23 16:02:30 signal: user defined signal 2
2021/03/23 16:02:30 reload
2021/03/23 16:02:30 server.Serve err: http: Server closed
2021/03/23 16:02:30 main: Listening to existing file descriptor 3.
2021/03/23 16:02:43 graceful reload
2021/03/23 16:02:43 signal end

root     31631     1  0 16:01 pts/0    00:00:00 ./reboot -graceful
root     31901 31631  0 16:02 pts/0    00:00:00 ./reboot -graceful

如果觉得我的文章对您有用,请随意赞赏。您的支持将鼓励我继续创作!