gRPC Context取消传播机制：Go客户端取消后服务器如何感知？

一、Go Context取消的核心底层逻辑 #

Go的context.Context取消本质是进程内信号广播机制：

• 可取消Context（比如context.WithCancel创建的）内部维护了一个触发取消的函数和一个信号通道。
• 调用cancel()时，会直接关闭内部信号通道，所有监听该通道的goroutine会立刻收到关闭通知（通道关闭后读操作会立即返回零值）。
• 这套逻辑完全是本地进程内的操作，不需要任何网络交互。

二、gRPC客户端取消Context时，是否会发起新HTTP请求？ #

不会。gRPC基于HTTP/2协议，它通过HTTP/2原生的控制帧而非新请求来通知服务端：

• 客户端本地调用cancel()取消Context后，gRPC底层的HTTP/2库会立刻向服务端发送一个RST_STREAM控制帧，标记当前gRPC请求对应的HTTP/2流为取消状态。
• 这个RST_STREAM帧属于当前请求流的一部分，不是独立的新HTTP请求。

三、服务端如何知晓客户端已取消Context？ #

服务端通过两层机制感知取消：

1. HTTP/2层的流关闭通知
   gRPC服务端的HTTP/2底层会监听RST_STREAM帧，一旦收到，会立即触发对应gRPC请求的Context取消逻辑。
2. Go Context的信号传递
   服务端处理gRPC请求时，会把请求关联的Context传入业务逻辑。底层收到取消信号后，会调用该Context的cancel()方法，关闭其内部信号通道。业务代码只要通过ctx.Done()监听这个通道，就能及时感知取消并终止执行中的逻辑。

举个服务端代码示例：

func (s *Server) SomeRPC(ctx context.Context, req *pb.Request) (*pb.Response, error) {
    select {
    case <-ctx.Done():
        // 感知到客户端取消，返回对应错误
        return nil, status.Errorf(codes.Canceled, "request canceled by client")
    case result := <-doBusinessWork():
        // 正常处理完成，返回结果
        return &pb.Response{Data: result}, nil
    }
}

另外，如果客户端进程意外崩溃导致HTTP/2连接断开，服务端也会通过连接关闭事件触发对应请求的Context取消。

内容的提问来源于stack exchange，提问作者Vishal