Go实现优雅关机与平滑重启

简介: Go实现优雅关机与平滑重启

前言

优雅关机就是服务端关机命令发出后不是立即关机,而是等待当前还在处理的请求全部处理完毕后再退出程序,是一种对客户端友好的关机方式。而执行Ctrl+C关闭服务端时,会强制结束进程导致正在访问的请求出现问题。


实现原理

Go 1.8版本之后, http.Server 内置的 Shutdown() 方法就支持优雅地关机,说明一下Shutdown工作的机制:当程序检测到中断信号时,我们调用http.server种的shutdown方法,该方法将阻止新的请求进来,同时保持当前的连接,知道当前连接完成则终止程序!


实现优雅重启

package main
import (
  "context"
  "fmt"
  "github.com/spf13/viper"
  "go.uber.org/zap"
  "log"
  "net/http"
  "os"
  "os/signal"
  "syscall"
  "time"
)
func main() {
  //启动服务(优雅关机)
  srv := &http.Server{
    Addr:    fmt.Sprintf(":%d", viper.GetInt("app.port")),
    Handler: r,
  }
  go func() {
    // 开启一个goroutine启动服务
    if err := srv.ListenAndServe(); err != nil && err != http.ErrServerClosed {
      log.Fatalf("listen: %s\n", err)
    }
  }()
  // 等待中断信号来优雅地关闭服务器,为关闭服务器操作设置一个5秒的超时
  quit := make(chan os.Signal, 1) // 创建一个接收信号的通道
  // kill 默认会发送 syscall.SIGTERM 信号
  // kill -2 发送 syscall.SIGINT 信号,我们常用的Ctrl+C就是触发系统SIGINT信号
  // kill -9 发送 syscall.SIGKILL 信号,但是不能被捕获,所以不需要添加它
  // signal.Notify把收到的 syscall.SIGINT或syscall.SIGTERM 信号转发给quit
  signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM) // 此处不会阻塞
  <-quit                                               // 阻塞在此,当接收到上述两种信号时才会往下执行
  zap.L().Info("Shutdown Server ...")
  // 创建一个5秒超时的context
  ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
  defer cancel()
  // 5秒内优雅关闭服务(将未处理完的请求处理完再关闭服务),超过5秒就超时退出
  if err := srv.Shutdown(ctx); err != nil {
    zap.L().Fatal("Server Shutdown: ", zap.Error(err))
  }
  zap.L().Info("Server exiting")
}

实现平滑重启

import (
  "log"
  "net/http"
  "time"
  "github.com/fvbock/endless"
  "github.com/gin-gonic/gin"
)
func main() {
  router := gin.Default()
  router.GET("/", func(c *gin.Context) {
    c.String(http.StatusOK, "hello xiaosheng !")
  })
  // 默认endless服务器会监听下列信号:
  // syscall.SIGHUP,syscall.SIGUSR1,syscall.SIGUSR2,syscall.SIGINT,syscall.SIGTERM和syscall.SIGTSTP
  // 接收到 SIGHUP 信号将触发`fork/restart` 实现优雅重启(kill -1 pid会发送SIGHUP信号)
  // 接收到 syscall.SIGINT或syscall.SIGTERM 信号将触发优雅关机
  // 接收到 SIGUSR2 信号将触发HammerTime
  // SIGUSR1 和 SIGTSTP 被用来触发一些用户自定义的hook函数
  if err := endless.ListenAndServe(":8080", router); err!=nil{
    log.Fatalf("listen: %s\n", err)
  }
  log.Println("Server exiting...")

测试

我们通过执行kill -1 pid命令发送syscall.SIGINT来通知程序优雅重启,具体做法如下:


打开终端,go build -o graceful_restart编译并执行./graceful_restart,终端输出当前pid(假设为43682)

将代码中处理请求函数返回的hello gin!修改为hello q1mi!,再次编译go build -o graceful_restart

打开一个浏览器,访问127.0.0.1:8080/,此时浏览器白屏等待服务端返回响应。

在终端迅速执行kill -1 43682命令给程序发送syscall.SIGHUP信号

等第3步浏览器收到响应信息hello gin!后再次访问127.0.0.1:8080/会收到hello q1mi!的响应。

在不影响当前未处理完请求的同时完成了程序代码的替换,实现了优雅重启。

但是需要注意的是,此时程序的PID变化了,因为endless 是通过fork子进程处理新请求,待原进程处理完当前请求后再退出的方式实现优雅重启的。所以当你的项目是使用类似supervisor的软件管理进程时就不适用这种方式了。


总结

无论是优雅关机还是优雅重启归根结底都是通过监听特定系统信号,然后执行一定的逻辑处理保障当前系统正在处理的请求被正常处理后再关闭当前进程。使用优雅关机还是使用优雅重启以及怎么实现,这就需要根据项目实际情况来决定了。


以上就是使用go如何优雅关机和平滑重启 的详细内容,更多关于go关机重启 的内容请关注博主的其它相关文章!


相关文章
|
10月前
|
Go UED
Go Web服务中如何优雅平滑重启?
在生产环境中,服务升级时如何确保不中断当前请求并应用新代码是一个挑战。本文介绍了如何使用 Go 语言的 `endless` 包实现服务的优雅重启,确保在不停止服务的情况下完成无缝升级。通过示例代码和测试步骤,详细展示了 `endless` 包的工作原理和实际应用。
193 3
|
10月前
|
JSON Go UED
Go Web服务中如何优雅关机?
在构建 Web 服务时,优雅关机是一个关键的技术点,它确保服务关闭时所有正在处理的请求都能顺利完成。本文通过一个简单的 Go 语言示例,展示了如何使用 Gin 框架实现优雅关机。通过捕获系统信号和使用 `http.Server` 的 `Shutdown` 方法,我们可以在服务关闭前等待所有请求处理完毕,从而提升用户体验,避免数据丢失或不一致。
136 1
|
安全 Go Docker
Go服务Docker Pod不断重启排查和解决
该文章分享了Go服务在Docker Pod中不断重启的问题排查过程和解决方案,识别出并发写map导致fatal error的问题,并提供了使用sync.Map或concurrent-map库作为并发安全的替代方案。
180 4
|
Kubernetes Cloud Native 网络协议
k8s 上 go 服务实战: 扩容 发版更新 回滚 平滑重启
我是 dayday, 读书写作敲代码, 永远在路上 感谢 [网管叨bi叨 - #Kubernetes学习笔记](https://mp.weixin.qq.com/mp/appmsgalbum?__biz=MzUzNTY5MzU2MA==&action=getalbum&album_id=1394839706508148737&scene=173#wechat_redirect "网管叨bi叨 - #Kubernetes学习笔记")
455 0
|
7月前
|
编译器 Go
揭秘 Go 语言中空结构体的强大用法
Go 语言中的空结构体 `struct{}` 不包含任何字段,不占用内存空间。它在实际编程中有多种典型用法:1) 结合 map 实现集合(set)类型;2) 与 channel 搭配用于信号通知;3) 申请超大容量的 Slice 和 Array 以节省内存;4) 作为接口实现时明确表示不关注值。此外,需要注意的是,空结构体作为字段时可能会因内存对齐原因占用额外空间。建议将空结构体放在外层结构体的第一个字段以优化内存使用。
|
7月前
|
运维 监控 算法
监控局域网其他电脑:Go 语言迪杰斯特拉算法的高效应用
在信息化时代,监控局域网成为网络管理与安全防护的关键需求。本文探讨了迪杰斯特拉(Dijkstra)算法在监控局域网中的应用,通过计算最短路径优化数据传输和故障检测。文中提供了使用Go语言实现的代码例程,展示了如何高效地进行网络监控,确保局域网的稳定运行和数据安全。迪杰斯特拉算法能减少传输延迟和带宽消耗,及时发现并处理网络故障,适用于复杂网络环境下的管理和维护。
|
29天前
|
数据采集 Go API
Go语言实战案例:多协程并发下载网页内容
本文是《Go语言100个实战案例 · 网络与并发篇》第6篇,讲解如何使用 Goroutine 和 Channel 实现多协程并发抓取网页内容,提升网络请求效率。通过实战掌握高并发编程技巧,构建爬虫、内容聚合器等工具,涵盖 WaitGroup、超时控制、错误处理等核心知识点。
|
1月前
|
数据采集 JSON Go
Go语言实战案例:实现HTTP客户端请求并解析响应
本文是 Go 网络与并发实战系列的第 2 篇,详细介绍如何使用 Go 构建 HTTP 客户端,涵盖请求发送、响应解析、错误处理、Header 与 Body 提取等流程,并通过实战代码演示如何并发请求多个 URL,适合希望掌握 Go 网络编程基础的开发者。
|
2月前
|
JSON 前端开发 Go
Go语言实战:创建一个简单的 HTTP 服务器
本篇是《Go语言101实战》系列之一,讲解如何使用Go构建基础HTTP服务器。涵盖Go语言并发优势、HTTP服务搭建、路由处理、日志记录及测试方法,助你掌握高性能Web服务开发核心技能。
|
2月前
|
Go
如何在Go语言的HTTP请求中设置使用代理服务器
当使用特定的代理时,在某些情况下可能需要认证信息,认证信息可以在代理URL中提供,格式通常是:
164 0