Go教程

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

本文主要是介绍Golang 程序中实现优雅关闭 HTTP SERVER,对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!

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的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!