grpc-java拦截器实现鉴权转发时ServerInterceptor的Context未传递到ClientInterceptor

问题根源 #

gRPC的Context基于线程本地变量实现，默认仅当前线程可获取绑定的Context值，跨线程传递时如果没有主动处理就会出现上下文丢失，你遇到的是典型的异步调用上下文丢失场景：
异步存根发起外发gRPC调用时，CallCredentials的applyRequestMetadata方法会通过gRPC内置的appExecutor执行，该执行线程独立于服务端处理请求的线程，自然无法获取之前绑定的Context值。

解决方案 #

方案1：客户端拦截器提前捕获Context（推荐） #

在ClientInterceptor的interceptCall方法执行时（此时仍在服务端处理线程内，Context处于有效状态），直接捕获当前Context，无需等异步线程执行时再调用Context.current()获取，修改AuthContextClientInterceptor代码如下：

class AuthContextClientInterceptor implements ClientInterceptor {
  @Override
  public <ReqT, RespT> ClientCall<ReqT, RespT> interceptCall(
      MethodDescriptor<ReqT, RespT> method, CallOptions callOptions, Channel next) {
    // 捕获当前服务端处理线程的有效Context
    Context currentContext = Context.current();
    return next.newCall(
        method,
        callOptions.withCallCredentials(
            new CallCredentials() {
              @Override
              public void applyRequestMetadata(RequestInfo requestInfo, Executor appExecutor, MetadataApplier applier) {
                appExecutor.execute(
                    () -> {
                      try {
                        Metadata headers = new Metadata();
                        // 直接使用提前捕获的Context，不从当前线程取
                        for (GrpcMetadataPassthrough field : GrpcMetadataPassthrough.values()) {
                          field.copyContextToMetadata(currentContext, headers);
                        }
                        applier.apply(headers);
                      } catch (Throwable e) {
                        applier.fail(Status.UNAUTHENTICATED.withCause(e));
                      }
                    });

              }

              @Override
              public void thisUsesUnstableApi() {}
            }));
  }
}

该方案改造成本极低，逻辑清晰，无需修改其他现有代码。

方案2：包装执行器自动传递Context #

如果存在大量异步任务需要跨线程传递Context，可以使用Context.wrapExecutor()方法对执行器进行包装，提交到该执行器的所有任务都会自动携带当前Context，该方案适合全链路异步场景，对你当前的单一透传场景来说重量较高。

gRPC Context核心管理机制 #

1. Context是不可变对象，调用withValue方法会返回全新的Context实例，不会修改原有对象
2. Context通过attach()/detach()方法绑定/解绑到当前线程，只有绑定后Context.current()才能返回最新的Context值
3. 服务端拦截器中调用的Contexts.interceptCall已经自动完成了Context的绑定、解绑操作，因此你在服务端业务处理方法中可以拿到正确的Context值
4. 跨线程传递Context必须主动处理，要么提前捕获有效Context，要么包装执行器实现自动传递。

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