Go Web服务中如何优雅关机?

简介: 在构建 Web 服务时,优雅关机是一个关键的技术点,它确保服务关闭时所有正在处理的请求都能顺利完成。本文通过一个简单的 Go 语言示例,展示了如何使用 Gin 框架实现优雅关机。通过捕获系统信号和使用 `http.Server` 的 `Shutdown` 方法,我们可以在服务关闭前等待所有请求处理完毕,从而提升用户体验,避免数据丢失或不一致。

在构建 Web 服务时,我们往往会遇到一个棘手的问题:当我们想要停止服务时,如何确保正在处理的请求能够顺利完成,而不是突然中断? 这种技术被称为“优雅关机”,它可以确保在服务关闭时,所有的请求都被妥善处理。

在这篇文章中,我们将通过一个简单的例子来演示如何在 Go 语言中使用 Gin 框架实现优雅关机。

什么是优雅关机?

优雅的关机是指在关闭服务之前,先让服务处理完当前正在处理的请求,然后再关闭服务。这样可以保证服务不会丢失请求,也不会影响到正在处理的请求。这种方式可以提高用户体验,防止服务中断造成的数据丢失或不一致。
而执行 Ctrl + C 或者 kill -2 pid 命令关闭服务,是不会等待服务处理完请求的,这样就会导致服务丢失请求。

如何实现优雅的关机?

Go 1.8 版本之后,http.Server 内置的 Shutdown() 方法就支持优雅地关机。

代码实现

我们来看一个具体的代码示例,通过这个例子我们将展示如何实现优雅关机。

package main

import (
    "context"
    "log"
    "net/http"
    "os"
    "os/signal"
    "syscall"
    "time"

    "github.com/gin-gonic/gin"
)

func main() {
   
    router := gin.Default()

    // 定义一个简单的路由
    router.GET("/ping", func(c *gin.Context) {
   
        // 模拟一个耗时操作,比如数据库查询或外部API调用
        time.Sleep(5 * time.Second)
        c.JSON(http.StatusOK, gin.H{
   
            "message": "pong",
        })
    })

    // 配置 HTTP 服务器
    srv := &http.Server{
   
        Addr:    ":8080",
        Handler: router,
    }

    // 启动服务器
    go func() {
   
        if err := srv.ListenAndServe(); err != nil && err != http.ErrServerClosed {
   
            log.Fatalf("Failed to start server: %v", 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                                               // 阻塞在此,当接收到上述两种信号时才会往下执行
    log.Println("Shutdown Server ...")

    // 创建一个 5 秒超时的 context
    ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
    defer cancel()
    // 5 秒内优雅关闭服务(将未处理完的请求处理完再关闭服务),超过 5 秒就超时退出
    if err := srv.Shutdown(ctx); err != nil {
   
        log.Fatal("Server Shutdown: ", err)
    }

    log.Println("Server exiting")

}

代码解析

1. 路由定义和服务启动

router := gin.Default()
router.GET("/ping", func(c *gin.Context) {
   
    time.Sleep(5 * time.Second)
    c.JSON(http.StatusOK, gin.H{
   
        "message": "pong",
    })
})

首先,我们创建了一个简单的 Gin 路由,并定义了一个 /ping 接口。当访问这个接口时,服务器会模拟一个耗时 5 秒的操作,然后返回一个 JSON 响应。这段代码展示了一个可能需要优雅关机的典型场景:服务器可能正在处理耗时的请求,如果此时直接关机,请求会被中断。

2. HTTP 服务器配置和启动

srv := &http.Server{
   
    Addr:    ":8080",
    Handler: router,
}

go func() {
   
    if err := srv.ListenAndServe(); err != nil && err != http.ErrServerClosed {
   
        log.Fatalf("Failed to start server: %v", err)
    }
}()

我们使用 http.Server 结构体配置并启动了一个 HTTP 服务器。服务器在一个单独的 goroutine 中运行,这样主程序可以继续执行,而不必等待服务器启动完成。

3. 捕获系统信号

quit := make(chan os.Signal, 1)
signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM)

<-quit

为了实现优雅关机,我们需要捕获系统信号。这里使用了 os/signal 包来监听 syscall.SIGINTsyscall.SIGTERM 信号。当用户按下 Ctrl+C 或者通过 kill 命令发送信号时,这些信号会被捕获并发送到 quit 通道,程序会随即从阻塞状态中恢复,继续执行后续代码。

4. 实现优雅关机

ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
defer cancel()

if err := srv.Shutdown(ctx); err != nil {
   
    log.Fatal("Server Shutdown: ", err)
}

在捕获到关机信号后,我们使用 http.ServerShutdown 方法来实现优雅关机。Shutdown 方法接受一个 context 参数,这个 context 设置了一个超时时间。在这里,我们设置了一个 5 秒的超时时间,意味着服务器将在 5 秒内等待未完成的请求处理完毕,然后关闭。如果超过了设定的超时时间,服务器将退出,程序也会正常结束。

如何验证优雅关机的效果?

要验证优雅关机的效果,可以按照以下步骤操作:

  1. 打开终端,运行 go run gin_shutdown.go
  2. 打开浏览器,并访问 http://127.0.0.1:8080/ping 此时浏览器应该会白屏等待服务端返回响应
  3. 在刚刚打开的终端上迅速按下 Ctrl+C 命令,此时会自动给程序发送 syscall.SIGINT 信号
  4. 此时程序并不会立即退出,而是会等上面的第 2 步的响应返回之后再退出,从而实现优雅关机的效果

总结

优雅关机是构建健壮 Web 服务的一个重要技术点,它确保了在服务关闭时所有正在处理的请求都能被妥善完成。在本文中,我们通过 Gin 框架演示了如何在 Go 中实现优雅关机。通过这种方式,我们可以提升用户体验,减少由于服务中断导致的各种潜在问题。

希望这篇文章能够帮助你更好地理解和实现 Go 服务中的优雅关机。如果你有任何问题或建议,欢迎在评论区与我讨论!

相关文章
|
3天前
|
存储 人工智能 弹性计算
阿里云弹性计算_加速计算专场精华概览 | 2024云栖大会回顾
2024年9月19-21日,2024云栖大会在杭州云栖小镇举行,阿里云智能集团资深技术专家、异构计算产品技术负责人王超等多位产品、技术专家,共同带来了题为《AI Infra的前沿技术与应用实践》的专场session。本次专场重点介绍了阿里云AI Infra 产品架构与技术能力,及用户如何使用阿里云灵骏产品进行AI大模型开发、训练和应用。围绕当下大模型训练和推理的技术难点,专家们分享了如何在阿里云上实现稳定、高效、经济的大模型训练,并通过多个客户案例展示了云上大模型训练的显著优势。
|
7天前
|
存储 人工智能 调度
阿里云吴结生:高性能计算持续创新,响应数据+AI时代的多元化负载需求
在数字化转型的大潮中,每家公司都在积极探索如何利用数据驱动业务增长,而AI技术的快速发展更是加速了这一进程。
|
4天前
|
人工智能 运维 双11
2024阿里云双十一云资源购买指南(纯客观,无广)
2024年双十一,阿里云推出多项重磅优惠,特别针对新迁入云的企业和初创公司提供丰厚补贴。其中,36元一年的轻量应用服务器、1.95元/小时的16核60GB A10卡以及1元购域名等产品尤为值得关注。这些产品不仅价格亲民,还提供了丰富的功能和服务,非常适合个人开发者、学生及中小企业快速上手和部署应用。
|
12天前
|
人工智能 弹性计算 文字识别
基于阿里云文档智能和RAG快速构建企业"第二大脑"
在数字化转型的背景下,企业面临海量文档管理的挑战。传统的文档管理方式效率低下,难以满足业务需求。阿里云推出的文档智能(Document Mind)与检索增强生成(RAG)技术,通过自动化解析和智能检索,极大地提升了文档管理的效率和信息利用的价值。本文介绍了如何利用阿里云的解决方案,快速构建企业专属的“第二大脑”,助力企业在竞争中占据优势。
|
14天前
|
自然语言处理 数据可视化 前端开发
从数据提取到管理:合合信息的智能文档处理全方位解析【合合信息智能文档处理百宝箱】
合合信息的智能文档处理“百宝箱”涵盖文档解析、向量化模型、测评工具等,解决了复杂文档解析、大模型问答幻觉、文档解析效果评估、知识库搭建、多语言文档翻译等问题。通过可视化解析工具 TextIn ParseX、向量化模型 acge-embedding 和文档解析测评工具 markdown_tester,百宝箱提升了文档处理的效率和精确度,适用于多种文档格式和语言环境,助力企业实现高效的信息管理和业务支持。
3935 2
从数据提取到管理:合合信息的智能文档处理全方位解析【合合信息智能文档处理百宝箱】
|
3天前
|
算法 安全 网络安全
阿里云SSL证书双11精选,WoSign SSL国产证书优惠
2024阿里云11.11金秋云创季活动火热进行中,活动月期间(2024年11月01日至11月30日)通过折扣、叠加优惠券等多种方式,阿里云WoSign SSL证书实现优惠价格新低,DV SSL证书220元/年起,助力中小企业轻松实现HTTPS加密,保障数据传输安全。
494 3
阿里云SSL证书双11精选,WoSign SSL国产证书优惠
|
10天前
|
安全 数据建模 网络安全
2024阿里云双11,WoSign SSL证书优惠券使用攻略
2024阿里云“11.11金秋云创季”活动主会场,阿里云用户通过完成个人或企业实名认证,可以领取不同额度的满减优惠券,叠加折扣优惠。用户购买WoSign SSL证书,如何叠加才能更加优惠呢?
985 3
|
7天前
|
机器学习/深度学习 存储 人工智能
白话文讲解大模型| Attention is all you need
本文档旨在详细阐述当前主流的大模型技术架构如Transformer架构。我们将从技术概述、架构介绍到具体模型实现等多个角度进行讲解。通过本文档,我们期望为读者提供一个全面的理解,帮助大家掌握大模型的工作原理,增强与客户沟通的技术基础。本文档适合对大模型感兴趣的人员阅读。
391 15
白话文讲解大模型| Attention is all you need
|
7天前
|
算法 数据建模 网络安全
阿里云SSL证书2024双11优惠,WoSign DV证书220元/年起
2024阿里云11.11金秋云创季火热进行中,活动月期间(2024年11月01日至11月30日),阿里云SSL证书限时优惠,部分证书产品新老同享75折起;通过优惠折扣、叠加满减优惠券等多种方式,阿里云WoSign SSL证书将实现优惠价格新低,DV SSL证书220元/年起。
559 5
|
3天前
|
安全 网络安全
您有一份网络安全攻略待领取!!!
深入了解如何保护自己的云上资产,领取超酷的安全海报和定制鼠标垫,随时随地提醒你保持警惕!
689 1
您有一份网络安全攻略待领取!!!