ServerInterceptor 中的 grpc-java 异步调用

Question

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

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 请求的正确方法是什么？我知道它可以直接在 grpc 服务中使用 StreamObservers 完成，但请求授权是一个交叉问题，拦截器似乎是一个完美的地方。

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;
  }
}