golang net/http的优雅关机

什么是优雅关机

http-server运行过程中，若进程被关闭，那么正在处理的请求可能只被处理了一半就停止了，可能会产生数据的不一致。
优雅关机是指：

• 首先，停止接收新请求；
• 然后，等待队列中的请求被处理完毕；
• 最后，应用程序退出；

net/http如何实现优雅关机

net/http原生支持优雅关机。

首先，在goroutine中启动http-server:

srv := &http.Server{
  Addr:              ":8090",
  Handler:           r,
}
go func() {
  if err := srv.ListenAndServe(); err != nil && err != http.ErrServerClosed {
     log.Fatal("listen: ", err)
  } else {
       log.Println("ListenAndServe break")
  }
}()

然后，在main中监听关机信号：

• SIGTERM: kill的默认信号；
• SIGINT：kill -2，一般是Ctrl+C的退出；
• SIGKILL：kill -9，捕获不到；

quit := make(chan os.Signal, 1)
signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM)
<- quit

最后，监听到关机信号后，执行优雅关机：

• 调用srv.Shutdown()执行优雅关机；
• 默认等待所有队列中的请求处理完毕后，才返回；
• 传入timeoutContext，增加超时时间，超时时间到后返回；

ctx, cancel := context.WithTimeout(context.TODO(), 20*time.Second)
defer cancel()
if err := srv.Shutdown(ctx); err != nil {
   log.Fatal("Server shutdown: ", err)
}

net/http优雅关机的实现原理

优雅关机是调用net/http的srv.Shutdown(ctx)实现的，该方法会：

• 先拒绝后面的connection请求；
• 然后再慢慢的处理未处理完毕的请求；
• 当未处理的请求一旦被处理完毕，其connection变成Idle，然后被Close()掉；

func (srv *Server) Shutdown(ctx context.Context) error {
    ...
    for _, f := range srv.onShutdown {
       go f()
    }
    ...

    ticker := time.NewTicker(shutdownPollInterval)
    defer ticker.Stop()
    for {
       if srv.closeIdleConns() {    //关闭idle连接
          return lnerr
       }
       select {
           case <-ctx.Done():       //ctx到期，如cancel()被调用
              return ctx.Err()
           case <-ticker.C:
       }
    }
}

另外，当调用Shutdown()后，srv.ListenAndServer方法将退出，并返回ErrServerClose错误。