Go项目中关于优雅关闭的那些事

简介: 如何实现简单的优雅关闭

优雅关闭,是使Go Web项目变得可靠的重要一环。

目录

为什么需要优雅关闭

小demo

信号:

官方源码:


为什么需要优雅关闭

想象一下这个场景:你的服务器正在处理一些用户的请求,这时你需要重启或关闭服务(例如发布新版本)。你直接用 Ctrl+C 或者 kill -9 很粗暴的终止进程,会发生什么?

  • 正在处理的请求会被突然中断,用户可能看到错误。
  • 数据库操作可能只完成一半,导致数据不一致
  • 一些重要的清理工作(如关闭文件、释放连接)无法进行。

而优雅关闭,就是为了解决这个问题。

如:停止接收新的请求、完成所有正在进行的请求、释放资源、最后安全退出什么的....

小demo

package main
import (
    "context"
    "log"
    "net/http"
    "os"
    "os/signal"
    "syscall"
    "time"
    "github.com/gin-gonic/gin"
)
func main() {
    // 1. 创建Gin路由引擎
    router := gin.Default()
    // 2. 定义一个测试路由,我们让它延迟5秒响应,模拟一个长时间运行的请求
    router.GET("/", func(c *gin.Context) {
        time.Sleep(5 * time.Second) // 模拟耗时操作
        c.JSON(http.StatusOK, gin.H{
            "message": "请求处理完成!",
        })
    })
    // 3. 创建HTTP Server,并注入Gin引擎
    srv := &http.Server{
        Addr:    ":8080",
        Handler: router,
    }
    // 4. 在一个新的goroutine里启动Server
    go func() {
        log.Println("服务器正在启动,监听地址: http://localhost:8080")
        if err := srv.ListenAndServe(); err != nil && err != http.ErrServerClosed {
            log.Fatalf("服务器启动失败:%s\n", err)
        }
    }()
    // 5. 等待中断信号以优雅地关闭服务器(设置 10 秒的超时时间)
    // 创建一个接收信号的通道
    quit := make(chan os.Signal, 1)
    // signal.Notify会监听指定的系统信号,并发送到quit通道
    // kill 默认会发送 syscall.SIGTERM 信号
    // kill -2 发送 syscall.SIGINT 信号(我们常用的Ctrl+C)
    // kill -9 发送 syscall.SIGKILL 信号,但无法捕获,所以不需要添加
    signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM)
    <-quit // 阻塞在这里,直到有信号写入quit通道
    log.Println("收到关闭信号,正在优雅地关闭服务器...")
    // 6. 创建一个带有超时的context,用于通知服务器还有10秒时间完成当前请求
    ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
    defer cancel() // 在函数返回前执行,释放资源
    // 7. 调用Shutdown方法,传入超时上下文
    if err := srv.Shutdown(ctx); err != nil {
        log.Fatal("服务器强制关闭:", err)
    }
    log.Println("服务器已安全退出")
}

image.gif

信号:

本demo中,需要掌握信号(signal)。

信号的本质就是操作系统于进程之间的 “通知机制” 。

用来告诉进程进行特定的操作(如:退出、暂停、重启...啥的)

每个信号,都有特定的编号和含义,就如下:

常用信号表:

信号名 信号编号 kill 命令写法 含义与默认行为 是否可被捕获? 常见场景
SIGINT 2 kill -2 PIDCtrl+C 中断信号(“优雅请求退出”) 用户按 Ctrl+C 终止前台进程
SIGTERM 15 kill PID(默认)或 kill -15 PID 终止信号(“友好退出请求”) 系统 / 工具默认的 “停止进程” 信号
SIGKILL 9 kill -9 PID 强制杀死信号(“不可抗拒的死亡命令”) 进程卡死时强制清理

注意,不是所有信号都可以被捕获的,如kill -9 ,根本捕获不到,因为进程会直接被杀死。

平时咱们项目正常关闭,都会发送SIGTERM(kill -15)。只有当进程对-15没响应时,才会-9强制杀死。

当然

   signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM)

   <-quit // 阻塞在这里,直到有信号写入quit通道

这行代码告诉Go,当收到SIGINT(Ctrl+C),或SIGTERM(kill命令默认信号)时,

会把信号发送到quit。

代码不在阻塞,可以正常进行下去,最终来到Context这里。

此后可以通过上下文设置context:

1、可以限制优雅关闭的最大等待时间

2、支持主动取消优雅关闭

官方源码:

package main
import (
  "context"
  "log"
  "net/http"
  "os"
  "os/signal"
)
func main() {
  var srv http.Server
  idleConnsClosed := make(chan struct{})
  go func() {
    sigint := make(chan os.Signal, 1)
    signal.Notify(sigint, os.Interrupt)
    <-sigint
    // We received an interrupt signal, shut down.
    if err := srv.Shutdown(context.Background()); err != nil {
      // Error from closing listeners, or context timeout:
      log.Printf("HTTP server Shutdown: %v", err)
    }
    close(idleConnsClosed)
  }()
  if err := srv.ListenAndServe(); err != http.ErrServerClosed {
    // Error starting or closing listener:
    log.Fatalf("HTTP server ListenAndServe: %v", err)
  }
  <-idleConnsClosed
}

image.gif

func (s *Server) Shutdown(ctx context.Context) error

官方文档翻译:

关机将在不中断任何活动连接的情况下优雅地关闭服务器。Shutdown的工作原理是首先关闭所有打开的侦听器,然后关闭所有空闲连接,然后无限期地等待连接返回空闲状态,然后关闭。如果提供的上下文在关闭完成之前过期,则shutdown返回上下文的错误,否则返回关闭服务器的底层侦听器所返回的任何错误。

当Shutdown被调用时,Serve, ServeTLS, ListenAndServe和ListenAndServeTLS会立即返回ErrServerClosed。确保程序没有退出,而是等待Shutdown返回。

关闭不试图关闭或等待被劫持的连接,如WebSockets。如果需要的话,Shutdown的调用者应该单独通知这些长寿命连接关闭,并等待它们关闭。看到服务器。注册关机通知函数的方法。

一旦在服务器上调用Shutdown,它可能不会被重用;以后对Serve等方法的调用将返回ErrServerClosed。


借鉴:

https://pkg.go.dev/net/http#Server.Shutdown


感谢支持,后续会继续深入讲解

目录
相关文章
|
1天前
|
人工智能 JSON 机器人
让龙虾成为你的“公众号分身” | 阿里云服务器玩Openclaw
本文带你零成本玩转OpenClaw:学生认证白嫖6个月阿里云服务器,手把手配置飞书机器人、接入免费/高性价比AI模型(NVIDIA/通义),并打造微信公众号“全自动分身”——实时抓热榜、AI选题拆解、一键发布草稿,5分钟完成热点→文章全流程!
10100 24
让龙虾成为你的“公众号分身” | 阿里云服务器玩Openclaw
|
13天前
|
人工智能 安全 Linux
【OpenClaw保姆级图文教程】阿里云/本地部署集成模型Ollama/Qwen3.5/百炼 API 步骤流程及避坑指南
2026年,AI代理工具的部署逻辑已从“单一云端依赖”转向“云端+本地双轨模式”。OpenClaw(曾用名Clawdbot)作为开源AI代理框架,既支持对接阿里云百炼等云端免费API,也能通过Ollama部署本地大模型,完美解决两类核心需求:一是担心云端API泄露核心数据的隐私安全诉求;二是频繁调用导致token消耗过高的成本控制需求。
5831 14
|
21天前
|
人工智能 JavaScript Ubuntu
5分钟上手龙虾AI!OpenClaw部署(阿里云+本地)+ 免费多模型配置保姆级教程(MiniMax、Claude、阿里云百炼)
OpenClaw(昵称“龙虾AI”)作为2026年热门的开源个人AI助手,由PSPDFKit创始人Peter Steinberger开发,核心优势在于“真正执行任务”——不仅能聊天互动,还能自动处理邮件、管理日程、订机票、写代码等,且所有数据本地处理,隐私完全可控。它支持接入MiniMax、Claude、GPT等多类大模型,兼容微信、Telegram、飞书等主流聊天工具,搭配100+可扩展技能,成为兼顾实用性与隐私性的AI工具首选。
22771 119