1. 背景

　　项目开发过程中，随着需求的迭代，代码的发布会频繁进行，在发布过程中，如何让程序做到优雅的退出？

 

为什么需要优雅的退出？

  • 你的 http 服务，监听端口没有关闭，客户的请求发过来了，但处理了一半，可能造成脏数据。
  • 你的协程 worker 的一个任务运行了一半，程序退出了，结果不符合预期。

 

如下我们以 http 服务，gRPC 服务，单独的 woker 协程为例子，一步步说明平滑关闭的写法。

 

2. 常见的几种平滑关闭

为了解决退出可能出现的潜在问题，平滑关闭一般做如下一些事情

  • 关闭对外的监听端口，拒绝新的连接
  • 关闭异步运行的协程
  • 关闭依赖的资源
  • 等待如上资源关闭
  • 然后平滑关闭

2.1 http server 平滑关闭

原来的写法

　　

带平滑关闭的写法

把平滑关闭注册到http.Server的关闭函数中

　　

2.2 gRPC server 平滑关闭

原来的写法

带平滑关闭的写法　

　　

2.3 worker 协程平滑关闭

单独的协程启停，可以通过计数的方式注册到退出事件处理器中。

  • 启动协程 增加计数
    •  quit.GetQuitEvent().AddGoroutine()
  • 停止协程 减计数 
    •  quit.GetQuitEvent().DoneGoroutine()
  • 常驻后台运行的协程退出的条件改成退出事件是否结束的条件 
    • !quit.GetQuitEvent().HasFired()
  • 常驻后台运行的协程若通过 select 处理 chan，同时增加退出事件的chan
    •  case <-quit.GetQuitEvent().Done()

　　

2.4 实现 io.Closer 接口的自定义服务平滑关闭

实现 io.Closer 接口的结构体，增加到退出事件处理器中 

　　

2.5 集成其他框架怎么做

退出信号处理由某一框架接管，寻找框架如何注册退出函数，优秀的框架一般都会实现安全实现退出的机制。

如下将退出事件注册到某一框架的平滑关闭函数中

 

参考：

https://github.com/mygityf/go-library/blob/main/quit/quit.go

 

完。

祝玩的开心~