文章详情

短信预约-IT技能 免费直播动态提醒

请输入下面的图形验证码

提交验证

短信预约提醒成功

go实现服务优雅关闭的示例

2023-02-09 12:01

关注

为什么需要优雅关闭

什么叫优雅关闭?先说不优雅关闭,就是什么都不管,强制关闭进程,这会导致有些正在处理中的请求被强行中断

这样做有什么问题?

优雅关闭的核心是以下功能:

监控服务退出信号

在go中使用下面的代码监听退出信号,如果c返回,说明监听到信号

不同的操作系统监听不同的退出信号

c := make(chan os.Signal, 1)
signals := []os.Signal{
   syscall.SIGINT,
   syscall.SIGTERM,
   syscall.SIGQUIT,
}

signal.Notify(c, signals...)
<-c

拒绝新请求

go在1.8后增加了shutdown方法来,我们看看它怎么实现优雅关闭:

srv.inShutdown.setTrue()
lnerr := srv.closeListenersLocked()
srv.closeDoneChanLocked()

这一段对应到http服务接收请求的流程:

for {
   rw, err := l.Accept()
   if err != nil {
      select {
      case <-srv.getDoneChan():
         return ErrServerClosed
   // ...       
}

一旦关闭listener,关闭doneChan后,http服务就不会再接收新的请求,直接返回

执行关闭之前的回调

for _, f := range srv.onShutdown {
   go f()
}

这里的回调实现得比较粗糙:

等待处理中的请求执行完毕

设置标识位可以拒绝新的请求,但依旧在执行的请求还在处理中,需要等这些请求执行完毕

等待处理中的请求执行完毕有两种思路:

go选择了两种方式结合的模式,通过ctx设置一个最大的等待时间,同时不断轮询正在请求中的计数

ctx超时或者计数变为0,都会返回

timer := time.NewTimer(nextPollInterval())
defer timer.Stop()
for {
   if srv.closeIdleConns() && srv.numListeners() == 0 {
      return lnerr
   }
   select {
   case <-ctx.Done():
      return ctx.Err()
   case <-timer.C:
      timer.Reset(nextPollInterval())
   }
}

这里每隔一定时间检查已有请求是否执行完毕,如果执行完毕,或者外部通过ctx设置的超时到期就会返回

检查间隔是多少?

怎么判断是否执行完毕?

服务收到监听信号返回之前,关闭连接和listener,会被这里检查到

在这里插入图片描述

实战

func main() {
   // 注册路由
   http.Handle("/aaa", http.HandlerFunc(func(writer http.ResponseWriter, req *http.Request) {
      time.Sleep(time.Second * 10)
      fmt.Println(111)
   }))
   server := http.Server{
      Addr:    "localhost:8080",
      Handler: http.DefaultServeMux,
   }

   close := make(chan int)
   go func() {
      quit := make(chan os.Signal)
      signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM, syscall.SIGQUIT)
      <-quit

      ctx, cancel := context.WithTimeout(context.Background(), time.Second*30)
      defer cancel()
      err := server.Shutdown(ctx)
      log.Print(err)
      
      // 控制外层退出
      close <- 1
   }()

   err := server.ListenAndServe()
   fmt.Println(err)

   <-close
}

该代码做了下面的事:

需要注意的是,一定要在子协程中优雅关闭结束后,主协程才能退出,这里用channel控制

因为主协程发现doneChan被关闭时会马上返回,但此时主协程开的业务处理协程还在进行中,如果主协程此时退出,无法达到优雅关闭的效果

按照以下流程测试:

支持强制退出

既然有优雅退出,那就有强制退出,我们假设如果按下两次ctrl+c,代表用户希望服务强制退出:

close := make(chan int, 2)
go func() {
   quit := make(chan os.Signal)
   signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM, syscall.SIGQUIT)
   <-quit
   go func() {
      <-quit
      os.Exit(1)
   }()

   // ...
}()

做法很简单,收到第一个退出信号后,再开一个子协程,如果再收到退出信号,就调用os.Exit退出进程

并且close channel的容量需要为2,避免当两次退出信号过短时丢失信号

到此这篇关于go实现服务优雅关闭的示例的文章就介绍到这了,更多相关go 服务关闭内容请搜索编程网以前的文章或继续浏览下面的相关文章希望大家以后多多支持编程网!

阅读原文内容投诉

免责声明:

① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。

② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341

软考中级精品资料免费领

  • 历年真题答案解析
  • 备考技巧名师总结
  • 高频考点精准押题
  • 2024年上半年信息系统项目管理师第二批次真题及答案解析(完整版)

    难度     813人已做
    查看
  • 【考后总结】2024年5月26日信息系统项目管理师第2批次考情分析

    难度     354人已做
    查看
  • 【考后总结】2024年5月25日信息系统项目管理师第1批次考情分析

    难度     318人已做
    查看
  • 2024年上半年软考高项第一、二批次真题考点汇总(完整版)

    难度     435人已做
    查看
  • 2024年上半年系统架构设计师考试综合知识真题

    难度     224人已做
    查看

相关文章

发现更多好内容

猜你喜欢

AI推送时光机
位置:首页-资讯-后端开发
咦!没有更多了?去看看其它编程学习网 内容吧
首页课程
资料下载
问答资讯