Go os/signal NotifyContext 如何优雅响应终止信号

Go 服务收到 Ctrl+C 或容器发送的 SIGTERM 时，最稳妥的处理方式不是单独开一个 goroutine 读取信号通道，而是用 signal.NotifyContext 把信号转换成上下文取消。这样 HTTP 服务、后台 worker 和数据库操作都可以监听同一个 ctx.Done()，主流程再统一执行清理。

官方文档：https://pkg.go.dev/os/signal

NotifyContext 解决的不是“捕获信号”，而是统一取消入口

传统写法是创建 chan os.Signal，调用 signal.Notify，再由某个 goroutine 读取通道。这能捕获信号，但取消信号还要手工转发给每个工作模块。NotifyContext 直接返回一个派生 Context，下面三件事任意一个先发生，ctx.Done() 都会关闭：指定的 SIGINT/SIGTERM 到达、父 Context 被取消、显式调用返回的 stop()。

这意味着业务函数只需要接受 context.Context，不必知道信号来自终端、systemd 还是容器编排平台。若信号导致 Context 取消，context.Cause(ctx) 还能返回描述该信号的错误；普通的 ctx.Err() 则适合做统一的取消判断。

把信号、Context 与清理动作接成一个闭环

服务入口通常只创建一次生命周期 Context，并同时关注 SIGINT 与 SIGTERM。defer stop() 用来覆盖初始化失败、提前返回等路径；当工作完成或确认取消后，也可以立刻再次调用 stop()，让后续同类信号恢复默认处理。

package main

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

func main() {
	// 将终端中断和服务停止信号统一映射到生命周期 Context。
	ctx, stop := signal.NotifyContext(context.Background(), os.Interrupt, syscall.SIGTERM)
	defer stop() // 兜底注销信号行为，覆盖提前返回路径。

	server := &http.Server{Addr: ":8080"}
	go func() {
		// ListenAndServe 的正常关闭会返回 http.ErrServerClosed。
		if err := server.ListenAndServe(); err != nil && !errors.Is(err, http.ErrServerClosed) {
			log.Printf("HTTP 服务异常: %v", err)
		}
	}()

	go worker(ctx)
	

示例里的 worker 只演示取消边界。真实项目应把同一个 ctx 传给数据库、HTTP 客户端和队列消费函数；如果下游重新使用 context.Background()，信号就到不了那里。HTTP server 的关闭、数据库连接的释放和队列消费的停止仍由各自组件负责，NotifyContext 只负责提供统一的取消入口。

为什么 stop 必须调用两次

这并不是要求所有代码机械地写两行 stop()。第一次放在 defer 中，是为了保证所有返回路径都能清理；第二次放在 之后，是为了在已经收到终止信号时立即注销信号行为，避免后续信号继续被转成同一个 Context 的取消。CancelFunc 可以安全地重复调用，所以这种“显式尽早清理 + defer 兜底”的写法适合服务入口。

不要把 stop() 当成“让 worker 停止”的函数：worker 停止依赖它监听 ctx.Done() 并自行返回；stop 的职责是撤销信号注册和释放关联资源。若还需要把取消原因记录到日志，先保存 cause := context.Cause(ctx)，再进入清理流程即可。

常见问题

NotifyContext 能捕获 SIGKILL 吗？

不能。官方文档明确说明 SIGKILL 和 SIGSTOP 无法被程序捕获，优雅清理只能针对可处理的异步信号。

Windows 上可以复用 SIGTERM 方案吗？

不能直接照搬。Windows 可用于 Notify 的主要是 os.Interrupt，跨平台程序应把具体信号注册放在平台适配层。

只写 defer stop() 够不够？

功能上通常能完成最终释放，但收到信号后继续保留通知行为没有必要。服务进入收尾阶段时主动调用一次，再用 defer 兜底更清晰。