context 在 gRPC / Gin / K8s 中的实战

下面从 实战角度 出发，系统讲解 context 在 gRPC / Gin / Kubernetes 中的真实用法、设计模式和踩坑经验。

这是 Go 工程中最接近「生产级」的一部分。

一、context 在 gRPC 中的实战

gRPC 是 context 最重要的使用场景之一

1️⃣ gRPC 中 context 的核心地位

每一个 RPC 调用都必须传 context.Context
context 决定：
- 请求生命周期
- 超时 & 取消
- 认证信息
- Trace / Span ID
  
  resp, err := client.GetUser(ctx, req)

2️⃣ 超时控制（最常见）

复制代码

ctx, cancel := context.WithTimeout(context.Background(), 500*time.Millisecond)
defer cancel()

resp, err := client.GetUser(ctx, req)

📌 服务端感知方式

复制代码

func (s *Server) GetUser(ctx context.Context, req *pb.UserRequest) (*pb.UserReply, error) {
    select {
    case <-time.After(1 * time.Second):
        return &pb.UserReply{}, nil
    case <-ctx.Done():
        return nil, status.Error(codes.DeadlineExceeded, ctx.Err().Error())
    }
}

3️⃣ gRPC Interceptor 中使用 context

客户端拦截器

复制代码

func unaryClientInterceptor(ctx context.Context, method string, req, resp any, cc *grpc.ClientConn, invoker grpc.UnaryInvoker, opts ...grpc.CallOption) error {
    ctx = metadata.AppendToOutgoingContext(ctx, "x-request-id", "123")
    return invoker(ctx, method, req, resp, cc, opts...)
}

服务端拦截器

复制代码

func unaryServerInterceptor(ctx context.Context, req any, info *grpc.UnaryServerInfo, handler grpc.UnaryHandler) (any, error) {
    md, _ := metadata.FromIncomingContext(ctx)
    userID := md["user-id"]
    return handler(ctx, req)
}

✅ 这是 gRPC 中传递鉴权信息、TraceID 的标准方式

4️⃣ gRPC Streaming + context

复制代码

for {
    select {
    case <-stream.Context().Done():
        log.Println("client disconnected")
        return
    default:
        stream.Send(&msg)
    }
}

二、context 在 Gin 中的实战

Gin 的 gin.Context≠ context.Context，但二者必须配合使用

1️⃣ Gin 中正确的 context 用法

复制代码

func handler(c *gin.Context) {
    ctx := c.Request.Context()

    select {
    case <-ctx.Done():
        c.AbortWithStatus(499)
        return
    case res := <-doWork(ctx):
        c.JSON(200, res)
    }
}

📌 499 Client Closed Request 是 Gin 中非常常见的状态码

2️⃣ 超时控制（Gin + context）

复制代码

func handler(c *gin.Context) {
    ctx, cancel := context.WithTimeout(c.Request.Context(), 2*time.Second)
    defer cancel()

    result, err := callService(ctx)
    if err != nil {
        c.JSON(500, gin.H{"error": err.Error()})
        return
    }
    c.JSON(200, result)
}

3️⃣ 中间件中注入 context 值

复制代码

func AuthMiddleware() gin.HandlerFunc {
    return func(c *gin.Context) {
        ctx := context.WithValue(c.Request.Context(), "userID", 123)
        c.Request = c.Request.WithContext(ctx)
        c.Next()
    }
}

4️⃣ Gin 中 context 使用的 ❌ 错误示范

❌ 错误：

复制代码

ctx := context.Background() // 丢失请求生命周期

✅ 正确：

复制代码

ctx := c.Request.Context()

三、context 在 Kubernetes 中的实战

K8s 是 context 的"教科书级使用者"

1️⃣ Controller / Operator 中的 context

K8s controller 中 几乎所有函数都接收 context

复制代码

func (c *Controller) Run(ctx context.Context) error {
    for {
        select {
        case <-ctx.Done():
            return nil
        case <-time.After(time.Second):
            c.sync()
        }
    }
}

2️⃣ client-go 中的 context 使用

复制代码

pods, err := clientset.CoreV1().
    Pods("default").
    List(ctx, metav1.ListOptions{})

ctx 控制：
- APIServer 请求超时
- 控制器退出
- Leader Election 释放

3️⃣ Leader Election 中的 context

复制代码

leaderelection.RunOrDie(ctx, leaderelection.LeaderElectionConfig{
    Lock: lock,
    Callbacks: leaderelection.LeaderCallbacks{
        OnStartedLeading: func(ctx context.Context) {
            runController(ctx)
        },
    },
})

📌 Leader 切换时，旧 Leader 通过 ctx 优雅退出

4️⃣ Informer 中的 context

复制代码

factory := informers.NewSharedInformerFactory(clientset, 0)
informer := factory.Core().V1().Pods().Informer()

informer.Run(ctx.Done())

四、三者对比总结

场景	context 的核心作用
gRPC	RPC 生命周期、认证、追踪
Gin	HTTP 请求取消、超时、透传
K8s	控制器生命周期、优雅退出

五、生产级最佳实践（强烈建议）

✅ 统一规范

所有对外 API 第一参数必须是 ctx
所有 goroutine 必须监听 ctx.Done()
所有 context 必须 defer cancel()

✅ context 传递链

复制代码

HTTP / gRPC
   ↓
Service
   ↓
Repository
   ↓
DB / Redis / RPC

✅ 日志 & Trace

复制代码

traceID := ctx.Value("trace_id")

六、一句话总结

**context 是 Go 分布式系统的"神经系统"：

gRPC 用它控制 RPC，Gin 用它感知请求，Kubernetes 用它管理生命。**