ServerInterceptor中的grpc-java异步调用

Question

我有一个grpc服务器，在处理请求之前，我需要对auth服务进行异步调用。我认为这应该在拦截器中完成，但是它需要从interceptCall()同步返回一个侦听器。

class AuthInterceptor implements ServerInterceptor {

    @Override
    public <ReqT, RespT> ServerCall.Listener<ReqT> interceptCall(
        ServerCall<ReqT, RespT> call,
        Metadata headers,
        ServerCallHandler<ReqT, RespT> next
    ) {
        String token = ""; //get token from headers
        authService.authorize(token).subscribe(
            ok -> // process the request
            error -> call.close(Status.UNAUTHENTICATED, headers)
        );
        // Here we need to return a Listener, but we haven't started a call yet
    }
}

因此，问题是:如何从ServerInterceptor进行异步调用，如果不能执行，异步验证请求grpc的正确方法是什么？我知道可以通过StreamObservers直接在grpc服务中实现这一点，但是请求授权是一个横切的关注点，拦截器似乎是一个非常适合它的地方。

Answer 1

您确实需要返回一个ServerCall.Listener。但是，由于您不知道要委托的Listener，所以可以覆盖Listener中的每个方法来将回调添加到队列中。完成身份验证后，释放队列。

class DelayedListener<ReqT> extends Listener<ReqT> {
  private Listener<ReqT> delegate;
  private List<Runnable> events = new ArrayList<Runnable>();

  @Override public synchronized void onMessage(ReqT message) {
    if (delegate == null) {
      events.add(() -> delegate.onMessage(message));
    } else {
      delegate.onMessage(message);
    }
  }
  ...
  public synchronized void setDelegate(Listener<ReqT> delegate) {
    this.delegate = delegate;
    for (Runnable runnable : events) {
      runnable.run();
    }
    events = null;
  }
}