Go 优雅关停

王小明_HIT

发布于 2023-07-08 15:51:07

2370

发布于 2023-07-08 15:51:07

文章被收录于专栏：程序员奇点程序员奇点

什么是优雅关停

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

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

为什么需要优雅关停?

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

在Go中怎么去做优雅关停

ctx, stop := signal.NotifyContext(context.Background(), syscall.SIGINT, syscall.SIGTERM)
defer stop()

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

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
}

本文参与腾讯云自媒体分享计划，分享自微信公众号。

原始发表：2023-04-28，如有侵权请联系 cloudcommunity@tencent.com 删除