我有一个Spring GraphQL应用程序,它使用gRPC协议调用其他内部微服务。我需要将承载令牌和其他信息设置到gRPC头,我相信我们可以使用gRPC拦截器(ClientInterceptor
实现)设置它。
我尝试了以下方法:
基于How to Access attributes from grpc Context.current()?,我创建了一个键,这样我们就可以在spring-service和interceptor中引用它
public class ContextKeyHolder {
public static Context.Key<String> USER_INFO = Context.key("USER");
public static Context.Key<String> BEARER = Context.key("BEARER");
}
// spring-service method
public Employee getEmployee() {
...
Context.current().withValue(ContextKeyHolder.USER_INFO, currentUser.getUsername());
Context.current().withValue(ContextKeyHolder.BEARER, currentUser.getBearerToken());
return grpcClient.getEmployee(...);
}
// interceptCall implementation
public <ReqT, RespT> ClientCall<ReqT, RespT> interceptCall(MethodDescriptor<ReqT, RespT> methodDescriptor,
CallOptions callOptions, Channel channel) {
return new ForwardingClientCall.SimpleForwardingClientCall<>(
channel.newCall(methodDescriptor, callOptions)) {
@Override
public void start(Listener<RespT> responseListener, Metadata headers) {
...
String userInfo = ContextKeyHolder.USER_INFO.get(Context.current());
System.out.println("user => " + userInfo);
...
super.start(responseListener, headers);
}
};
}
这里我在拦截器方法中获得了null
userInfo。我在这里错过了什么吗?
另一种选择是使用ThreadLocal来保存上下文,但我不确定这是否是正确的选择。
1条答案
按热度按时间h6my8fg21#
在调用时需要使用您创建的上下文。因此,您的代码应该是:
或者: