Go 服务怎么实现优雅退出？

收到 SIGTERM → (K8s) 从 Service 端点移除，等几秒让流量切走
  → srv.Shutdown(ctx)：停止接收新请求，等进行中的请求完成
  → 取消后台任务的 context，wg.Wait() 等 goroutine 退出（MQ 消费者先停止拉取，处理完手上的消息再提交）
  → 关闭数据库、Redis、MQ 等连接，刷新日志和监控缓冲 → 进程退出

package main

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

func runWorker(ctx context.Context, wg *sync.WaitGroup) { // 后台定时任务，ctx 取消时退出
	defer wg.Done()
	ticker := time.NewTicker(10 * time.Second)
	defer ticker.Stop()
	for {
		select {
		case <-ctx.Done():
			return
		case <-ticker.C:
			slog.Info("worker tick") // 实际业务：清理过期数据、刷新缓存等
		}
	}
}

func run() error {
	ctx, stop := signal.NotifyContext(context.Background(), os.Interrupt, syscall.SIGTERM) // 收到信号时 ctx 被取消
	defer stop()

	workerCtx, cancelWorkers := context.WithCancel(context.Background())
	defer cancelWorkers()
	var wg sync.WaitGroup
	wg.Add(1)
	go runWorker(workerCtx, &wg)

	mux := http.NewServeMux()
	mux.HandleFunc("GET /hello", func(w http.ResponseWriter, _ *http.Request) { w.Write([]byte("hello\n")) })
	srv := &http.Server{Addr: ":8080", Handler: mux, ReadHeaderTimeout: 5 * time.Second}

	serveErr := make(chan error, 1)
	go func() { // 正常关闭时返回的 ErrServerClosed 不算错误
		if err := srv.ListenAndServe(); err != nil && !errors.Is(err, http.ErrServerClosed) {
			serveErr <- err
		}
	}()

	select {
	case err := <-serveErr: // 端口被占用等启动失败的情况
		return err
	case <-ctx.Done():
		slog.Info("shutting down", "cause", context.Cause(ctx))
	}
	stop() // 恢复默认信号行为：再按一次 Ctrl+C 可以直接退出

	// 用新的 context 计时，整个关闭过程最多 20 秒，要小于 terminationGracePeriodSeconds
	shutdownCtx, cancel := context.WithTimeout(context.Background(), 20*time.Second)
	defer cancel()
	if err := srv.Shutdown(shutdownCtx); err != nil {
		slog.Error("http shutdown", "err", err)
		srv.Close() // 超时了，强制断开剩下的连接
	}
	cancelWorkers() // 再通知后台任务退出，并等待它们结束
	wg.Wait()
	// 最后关闭依赖，如 db.Close()、redis.Close()
	return nil
}

func main() {
	if err := run(); err != nil { // 逻辑放在 run 里，defer 都能正常执行
		slog.Error("server exited", "err", err)
		os.Exit(1)
	}
}