grpc-gateway拦截器终止请求时WithForwardResponseOption未生效问题

是的，runtime.WithForwardResponseOption 默认不会处理 gRPC 拦截器提前返回的错误响应。

原因 #

这个选项的作用范围是RPC方法正常执行完成后的响应阶段，只有当请求通过拦截器校验、进入实际的RPC方法逻辑并返回结果/错误时，对应的响应处理钩子才会触发。而拦截器直接返回错误终止请求时，请求生命周期还没走到RPC方法执行的环节，所以该选项的逻辑不会被调用。

解决方案 #

针对拦截器阶段的错误，有两种常用处理方式：

1. 在拦截器内手动写入响应

在令牌校验拦截器中，当验证失败时，直接从上下文获取http.ResponseWriter，按照预期格式写入响应：

func authInterceptor(ctx context.Context, req interface{}, info *grpc.UnaryServerInfo, handler grpc.UnaryHandler) (interface{}, error) {
    // 执行令牌校验逻辑
    tokenValid := checkToken(ctx)
    if !tokenValid {
        // 从grpc-gateway上下文获取HTTP响应写入器
        w, ok := runtime.ServerHTTPResponseWriterFromContext(ctx)
        if ok {
            w.WriteHeader(http.StatusUnauthorized)
            w.Write([]byte(`{"code":401,"message":"无效的Bearer令牌"}`))
        }
        // 返回gRPC错误，终止请求流程
        return nil, status.Error(codes.Unauthenticated, "令牌校验失败")
    }
    // 校验通过，继续执行RPC方法
    return handler(ctx, req)
}

2. 配置全局错误处理器

通过runtime.WithErrorHandler设置全局错误处理逻辑，统一捕获所有gRPC错误（包括拦截器返回的）：

func customErrorHandler(ctx context.Context, mux *runtime.ServeMux, marshaler runtime.Marshaler, w http.ResponseWriter, r *http.Request, err error) {
    // 解析gRPC错误状态
    s, ok := status.FromError(err)
    if ok {
        switch s.Code() {
        case codes.Unauthenticated:
            w.WriteHeader(http.StatusUnauthorized)
            marshaler.NewEncoder(w).Encode(map[string]string{
                "error": "未授权：无效或缺失Bearer令牌",
            })
            return
        // 可添加其他错误码的自定义处理逻辑
        }
    }
    // 未匹配到自定义处理时，使用默认错误处理
    runtime.DefaultHTTPErrorHandler(ctx, mux, marshaler, w, r, err)
}

// 初始化grpc-gateway时配置
mux := runtime.NewServeMux(
    runtime.WithErrorHandler(customErrorHandler),
    // 其他配置选项...
)

总结 #

WithForwardResponseOption 仅适用于RPC方法执行后的响应自定义，拦截器阶段的错误需要通过拦截器内手动处理或全局错误处理器来控制返回格式。

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