如何访问要定义的Java grpc服务的请求元数据？

Z时代
2024-01-10
分类：问答

在某些背景下，我尝试使用grpc auth来为我定义的某些服务提供安全性。

让我们看看我是否可以问这是一种有意义的方法。对于我的python代码，实现服务器端代码非常容易。

class TestServiceServer(service_pb2.TestServiceServer):
    def TestHello(self, request, context):
        ## credential metadata for the incoming request
        metadata = context.invocation_metadata()
        ## authenticate the user using the metadata

因此，您可以说，我很容易从“上下文”中获取元数据。对我而言，更难的是在Java中执行相同的操作。

public class TestImpl extends TestServiceGrpc.TestServiceImplBase {
    @Override
    public void testHello(TestRequest req, StreamObserver<TestResponse> responseObserver) {
        // How do I get access to similar request metadata here?
        // from the parameter positions, it looks like it should be
        // "responseObserver" but that doesn't seem similar to "context"
    }
}

我承认我的问题来自几个方向。

1）我不太精通Java

2）为了调试类并查看可用的方法，我大量使用了python的“ pdb”。我不知道/我不精通Java的类似工具。

3）此时的文档似乎还很少。它显示了如何在服务器端建立ssl连接，但是我找不到在服务器中查看请求元数据的示例，如python所示。

有人可以给我一个关于如何执行此操作的想法，或者向我展示一个与python pdb相同的针对Java的有用调试工具吗？

编辑/回答：

我需要首先编写一个实现接口ServerInterceptor的定义。

private class TestInterceptor implements ServerInterceptor {
    ....

然后，在实际绑定我的服务并构建服务器之前，我需要这样做。

TestImpl service = new TestImpl();
ServerServiceDefinition intercepted = ServerInterceptors.intercept(service, new TestInterceptor());

现在，我可以创建服务器了。

server = NettyServerBuilder.forPort(port)
    // enable tls
    .useTransportSecurity(
        new File(serverCert),
        new File(serverKey)
    )
    .addService(
        intercepted  // had been "new TestImpl()"
    )
    .build();
server.start();

这使我在触发客户端请求时实际上可以调用ServerInterceptor。

此链接对于解决此问题很有帮助。

回答：

使用ServerInterceptor，然后通过传播身份Context。这使您可以制定中央身份验证策略。

拦截器可以从中检索身份Metadata headers。 然后， 它 应该验证

身份。然后可以testHello通过io.grpc.Context以下方式将经过验证的身份传达给应用程序（即）：

/** Interceptor that validates user's identity. */
class MyAuthInterceptor implements ServerInterceptor {
  public static final Context.Key<Object> USER_IDENTITY
      = Context.key("identity"); // "identity" is just for debugging
  @Override
  public <ReqT, RespT> ServerCall.Listener<ReqT> interceptCall(
      ServerCall<ReqT, RespT> call,
      Metadata headers,
      ServerCallHandler<ReqT, RespT> next) {
    // You need to implement validateIdentity
    Object identity = validateIdentity(headers);
    if (identity == null) { // this is optional, depending on your needs
      // Assume user not authenticated
      call.close(Status.UNAUTENTICATED.withDescription("some more info"),
                 new Metadata());
      return new ServerCall.Listener() {};
    }
    Context context = Context.current().withValue(USER_IDENTITY, identity);
    return Contexts.interceptCall(context, call, headers, next);
  }
}
public class TestImpl extends TestServiceGrpc.TestServiceImplBase {
  @Override
  public void testHello(TestRequest req, StreamObserver<TestResponse> responseObserver) {
    // Access to identity.
    Object identity = MyAuthInterceptor.USER_IDENTITY.get();
    ...
  }
}
// Need to use ServerInterceptors to enable the interceptor
Server server = ServerBuilder.forPort(PORT)
    .addService(ServerInterceptors.intercept(new TestImpl(),
        new MyAuthInterceptor()))
    .build()
    .start();