Go 优雅关停

2023-07-08 15:51:07 浏览数 (1)

什么是优雅关停

运行中的服务收到来自系统的停止信号,例如可能由于系统重启/关机等,需要在一小段时间内:

  • 停止对外提供服务
  • 完成已经接受/进行中的请求
  • 保存需要保存的数据
  • Flush I/O缓冲区
  • 其他清理

为什么需要优雅关停?

  • 能够保证请求完整性
  • 不会在服务正常重启的时候收到服务异常退出报警

在Go中怎么去做优雅关停

代码语言:javascript复制
ctx, stop := signal.NotifyContext(context.Background(), syscall.SIGINT, syscall.SIGTERM)
defer stop()

需要Go1.16版本 NotifyContext 代码实现

代码语言:javascript复制
func NotifyContext(parent context.Context, signals ...os.Signal) (ctx context.Context, stop context.CancelFunc) {
   ctx, cancel := context.WithCancel(parent)
   c := &signalCtx{
      Context: ctx,
      cancel:  cancel,
      signals: signals,
   }
   c.ch = make(chan os.Signal, 1)
   Notify(c.ch, c.signals...)
   if ctx.Err() == nil {
      go func() {
         select {
         case <-c.ch:
            c.cancel()
         case <-c.Done():
         }
      }()
   }
   return c, c.stop
}

0 人点赞