Golang 程序中实现优雅关闭 HTTP SERVER

2022-03-13 03:42:36

Golang 中实现一个 HTTP SERVER 异常的简单，利用标准库 net/http 的实现仅需数行代码即可，但是一个生产环境可用的 HTTP SERVER 还必须考虑更多的问题，其中如何实现优雅关闭 HTTP SERVER 是一个必须要处理的问题。这里所说的优雅即是指在 HTTP SERVER 里监听特定的信号，并在接收到信号时做出相应的处理。中文世界里也常用平滑一词代替，说的都是同一个意思。

如果你对 Golang 中监听并处理信号不了解，建议先花三分钟浏览一下本专栏的另一篇关于信号处理的文章：

Golang 程序中如何监听并处理信号？

Talk is cheap. Show me the code.

package main

import (
    "context"
    "fmt"
    "net/http"
    "os"
    "os/signal"
    "sync"
    "syscall"
    "time"
)

const addr = ":9527"

func main() {
    http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
        fmt.Fprint(w, "just another http server...")
    })
    //使用默认路由创建 http server
    srv := http.Server{
        Addr:    addr,
        Handler: http.DefaultServeMux,
    }
    //使用WaitGroup同步Goroutine
    var wg sync.WaitGroup
    exit := make(chan os.Signal)
    //监听 Ctrl+C 信号
    signal.Notify(exit, syscall.SIGINT, syscall.SIGTERM)
    go func() {
        <-exit
        wg.Add(1)
        //使用context控制srv.Shutdown的超时时间
        ctx, cancel := context.WithTimeout(context.Background(), 15*time.Second)
        defer cancel()
        err := srv.Shutdown(ctx)
        if err != nil {
            fmt.Println(err)
        }
        wg.Done()
    }()

    fmt.Println("listening at " + addr)
    err := srv.ListenAndServe()

    fmt.Println("waiting for the remaining connections to finish...")
    wg.Wait()
    if err != nil && err != http.ErrServerClosed {
        panic(err)
    }
    fmt.Println("gracefully shutdown the http server...")
}

Golang 程序中实现优雅关闭 HTTP SERVER

相关文章