Go语言如何在Web服务中实现优雅关机_Golang

在构建 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.sigint 和 syscall.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.server 的 shutdown 方法来实现优雅关机。shutdown 方法接受一个 context 参数，这个 context 设置了一个超时时间。在这里，我们设置了一个 5 秒的超时时间，意味着服务器将在 5 秒内等待未完成的请求处理完毕，然后关闭。如果超过了设定的超时时间，服务器将退出，程序也会正常结束。