Golang 中实现一个 HTTP SERVER 异常的简单,利用标准库 net/http 的实现仅需数行代码即可,但是一个生产环境可用的 HTTP SERVER 还必须考虑更多的问题,其中如何实现优雅关闭 HTTP SERVER 是一个必须要处理的问题。这里所说的 优雅 即是指在 HTTP SERVER 里监听特定的信号,并在接收到信号时做出相应的处理。中文世界里也常用平滑 一词代替,说的都是同一个意思。
如果你对 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...")
}
有疑问加站长微信联系(非本文作者)