Go 语言中如何优雅地关闭程序 (graceful shutdown)？

推荐答案

在 Go 语言中，可以通过监听操作系统信号（如 SIGINT 或 SIGTERM）来实现优雅关闭程序。以下是一个示例代码：

// www.javascriptcn.com code example
package main

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

func main() {
	// 创建一个 HTTP 服务器
	server := &http.Server{
		Addr: ":8080",
	}

	// 使用一个 goroutine 启动服务器
	go func() {
		if err := server.ListenAndServe(); err != nil && err != http.ErrServerClosed {
			log.Fatalf("Server error: %v\n", err)
		}
	}()

	// 创建一个 channel 来接收信号
	stop := make(chan os.Signal, 1)
	signal.Notify(stop, syscall.SIGINT, syscall.SIGTERM)

	// 阻塞直到接收到信号
	<-stop

	// 创建一个 context 用于优雅关闭
	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
	defer cancel()

	// 优雅关闭服务器
	if err := server.Shutdown(ctx); err != nil {
		log.Fatalf("Server shutdown error: %v\n", err)
	}

	fmt.Println("Server gracefully stopped")
}

本题详细解读

1. 监听操作系统信号

在 Go 中，os/signal 包提供了监听操作系统信号的功能。通过 signal.Notify 函数，我们可以捕获 SIGINT（通常由 Ctrl+C 触发）和 SIGTERM（通常由系统关闭或 kill 命令触发）等信号。

2. 使用 http.Server 的 Shutdown 方法

http.Server 结构体提供了一个 Shutdown 方法，用于优雅地关闭 HTTP 服务器。该方法会停止接受新的请求，并等待所有正在处理的请求完成后再关闭服务器。

3. 使用 context.Context 控制超时

在调用 Shutdown 方法时，通常会传入一个 context.Context 对象，用于控制关闭的超时时间。如果服务器在指定的时间内未能完成关闭，Shutdown 方法会返回一个错误。

4. 阻塞主 goroutine

通过 <-stop 语句，主 goroutine 会阻塞，直到接收到操作系统信号。这样可以确保程序不会在接收到信号之前提前退出。

5. 处理关闭错误

在调用 Shutdown 方法后，应该检查是否有错误返回。如果有错误，可以根据实际情况进行处理，比如记录日志或强制退出程序。

通过这种方式，Go 程序可以在接收到关闭信号时，优雅地完成所有正在处理的任务，并安全地退出。