从信号捕获到流量摘除：Go 服务优雅停机与无损下线实战

package main

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

type AppServer struct {
	httpServer *http.Server
	workerWg   sync.WaitGroup
	quitChan   chan struct{}
}

func NewAppServer() *AppServer {
	mux := http.NewServeMux()
	mux.HandleFunc("/api/v1/order/create", func(w http.ResponseWriter, r *http.Request) {
		// 模拟耗时业务逻辑
		time.Sleep(2 * time.Second)
		w.WriteHeader(http.StatusOK)
		w.Write([]byte(`{"status":"success"}`))
	})

	return &AppServer{
		httpServer: &http.Server{
			Addr:    ":8080",
			Handler: mux,
		},
		quitChan: make(chan struct{}),
	}
}

// 模拟后台异步任务（如从 Kafka 消费订单事件）
func (s *AppServer) StartBackgroundWorker() {
	s.workerWg.Add(1)
	go func() {
		defer s.workerWg.Done()
		log.Println("[Worker] 异步任务处理中心启动...")
		for {
			select {
			case <-s.quitChan:
				log.Println("[Worker] 收到退出信号，停止消费新消息，等待在途任务完成...")
				return
			default:
				// 模拟处理业务
				time.Sleep(500 * time.Millisecond)
			}
		}
	}()
}

func main() {
	app := NewAppServer()
	app.StartBackgroundWorker()

	// 1. 异步启动 HTTP 服务
	go func() {
		log.Printf("[HTTP] 服务启动，监听端口: %s\n", app.httpServer.Addr)
		if err := app.httpServer.ListenAndServe(); err != nil && !errors.Is(err, http.ErrServerClosed) {
			log.Fatalf("[HTTP] 服务启动异常: %v\n", err)
		}
	}()

	// 2. 监听系统停机信号
	// 必须同时监听 SIGINT (Ctrl+C) 和 SIGTERM (K8s/Docker 默认停机信号)
	sigChan := make(chan os.Signal, 1)
	signal.Notify(sigChan, syscall.SIGINT, syscall.SIGTERM)

	sig := <-sigChan
	log.Printf("[System] 接收到停机信号: %s，开始执行优雅下线流程...\n", sig.String())

	// 3. 设定整个下线的最大超时时间（兜底，防止任务挂起导致无法退出）
	shutdownCtx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
	defer cancel()

	// 4. 阶段一：停止 HTTP 请求入口
	// http.Server.Shutdown 会先关闭 Listener 拒绝新请求，再阻塞等待所有活跃连接处理完毕
	if err := app.httpServer.Shutdown(shutdownCtx); err != nil {
		log.Printf("[HTTP] 服务强制关闭: %v\n", err)
	} else {
		log.Println("[HTTP] 所有在途 HTTP 请求已处理完毕")
	}

	// 5. 阶段二：停止后台 Worker
	close(app.quitChan)
	workerDone := make(chan struct{})
	go func() {
		app.workerWg.Wait()
		close(workerDone)
	}()

	select {
	case <-workerDone:
		log.Println("[Worker] 异步任务已全部安全退出")
	case <-shutdownCtx.Done():
		log.Println("[Worker] 异步任务退出超时，强制跳过")
	}

	// 6. 阶段三：清理基础资源（DB 连接池、Redis 客户端、注册中心反注册等）
	cleanupResources()

	log.Println("[System] 服务已完全无损退出")
}

func cleanupResources() {
	log.Println("[Resource] 正在关闭 MySQL/Redis 连接池...")
	// db.Close()
	// redisClient.Close()
	log.Println("[Resource] 资源清理完成")
}

apiVersion: apps/v1
kind: Deployment
metadata:
  name: order-service
spec:
  replicas: 3
  template:
    spec:
      containers:
      - name: order-app
        image: order-service:v1.0.0
        lifecycle:
          preStop:
            exec:
              # 因果链：先 sleep 10s，给 kube-proxy / Ingress 留出更新路由表的时间
              # 期间 Pod 依然正常处理请求，10s 后再向 Go 进程发送 SIGTERM
              command: ["/bin/sh", "-c", "sleep 10"]
        # 必须大于 preStop sleep 时间 + Go Shutdown 超时时间
        # 否则时间一到，K8s 会直接发 SIGKILL (kill -9) 强杀进程
      terminationGracePeriodSeconds: 45