java - 如何将数据从gRPC拦截器传递到服务方法调用？

我需要从ServerAuthIntereptor中传递一些数据以进行调用。

ServerAuthInterceptor:

// used in context parameters map
private static final String AUTH_CONTEXT = "authContext";
private static final Context.Key<Object> AUTH_CONTEXT_KEY = Context.key(AUTH_CONTEXT);

// ...

@Override
public <ReqT, RespT> ServerCall.Listener<ReqT> interceptCall(
    ServerCall<ReqT, RespT> call, Metadata headers, ServerCallHandler<ReqT, RespT> next) {

    // check credentials from header

    // remember account and session token for usage later (eg. in logout)
    AuthContext authContext = new AuthContext();
    authContext.account = account;
    authContext.sessionToken = sessionToken;

    logger.info("Call {} allowed (account={}, authContext={})",
        new Object[] { call, account, authContext });

    Context context = Context
        .current()
        .withValue(AUTH_CONTEXT_KEY, authContext);

    return Contexts.interceptCall(context, call, headers, next);
}

// ...

/**
 * To be used from services
 * @return
 */
public static Account getCurrentContextAccount() {
    AuthContext authContext = (AuthContext) Context.current().key(AUTH_CONTEXT).get();
    return authContext.account;
}

在某些服务方法实现中，我尝试从当前上下文访问存储的“authContext”:

@Override
public void list(ListRequest request, StreamObserver<ListResponse> responseObserver) {
    logger.info("Starting list({}) ...", request);

     // ...

     account = ServerAuthInterceptor.getCurrentContextAccount();

但是authContext是null。
有时它在工作，所以我相信它与线程(本地)有关。

正确的做法是什么？

最佳答案

上下文键使用引用相等。因此，您必须使用相同的键实例进行设置和获取。

这段代码重新创建了密钥，因此它不匹配:

Context.current().key(AUTH_CONTEXT).get();

相反，它应该是:

AUTH_CONTEXT_KEY.get();

使用引用相等，因此您可以像对ThreadLocal一样，将常规Java可见性(公共，受保护，程序包私有，私有)规则应用于该值。