go语言gin、net/http的优雅关机

辛集电子

已于 2022-10-25 20:19:52 修改

阅读量433

点赞数 1

CC 4.0 BY-SA版权

分类专栏： Go 文章标签： golang gin http

于 2022-10-15 16:53:31 首次发布

一年之内毫无半点成绩，想起来做人真是没趣。

本文链接：https://blog.youkuaiyun.com/qq_26039331/article/details/127337293

Go 专栏收录该内容

36 篇文章

订阅专栏

本文介绍HTTP服务器优雅关机的概念及其实现方法。优雅关机能够确保在服务器关闭前完成所有正在处理的请求，避免数据不一致的问题。文章详细解释了如何使用net/http库中的srv.Shutdown方法来实现这一目标。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

什么是优雅关机？

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

优雅关机是指：

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

net/http原生的shutdown函数支持优雅关机。

优雅关机的实现

首先，在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

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:
       }
    }
}