如何在java中从grpc请求中获取请求标头？

Question

我正在构建一个Java的grpc服务器，我很难得到请求头。proto文件是使用Pro buf-maven-plugin编译的，基于生成的存根，我无法访问上下文或请求元数据。

我还尝试在消息请求中定义一个键/值列表，希望grpc能够处理头映射，但目前为止没有成功。

知道我怎么能拿到标题吗？

谢谢

我的原型文件内容：

...
// Version Request
message VersionRequest {
  // key/value pairs
  repeated Header headers = 1;
}

message Header {
  //key
  string key = 1;
  //value
  string constant = 2;
}
...



   ...
security_definitions: {
    security: {
      key: "clientIdAuth";
      value: {
        type: TYPE_API_KEY;
        in: IN_HEADER;
        name: "x-client-id";
      }
    }
    security: {
      key: "clientSecretAuth";
      value: {
        type: TYPE_API_KEY;
        in: IN_HEADER;
        name: "x-client-secret";
      }
    }
    security: {
      key: "bearerAuth";
      value: {
        type: TYPE_API_KEY;
        in: IN_HEADER;
        name: "Authorization";
      }
    }
  }
    ...
    // Retrieves system version
      //
      // Retrieves system version
      rpc GetVersion(VersionRequest) returns (Version) {
        option (google.api.http) = {
          get : "/api/v4/version"
        };
        option (grpc.gateway.protoc_gen_swagger.options.openapiv2_operation) = {
          security: {
            security_requirement: {
              key: "clientIdAuth";
              value: {}
            }
            security_requirement: {
              key: "clientSecretAuth";
              value: {}
            }
            security_requirement: {
              key: "bearerAuth";
              value: {}
            }
          }
        };
      }
    ...

服务器端实现如下所示：

...
@GrpcService
public class GrpcAPIService extends MixAPIGrpc.MixAPIImplBase {

    @Autowired
    private MwProxy mwProxy;

    @Override
    public void getVersion(VersionRequest versionRequest, StreamObserver<Version> streamObserver) {
        System.out.println("===============" + versionRequest.getHeadersCount());
        handleEGrpcCall("getVersion", mwProxy::getVersion, streamObserver);
    }
...

版本请求。getHeadersCount（）始终返回0，尽管在所有我的请求中都发送了3个标头。

Answer 1

标头，我想是您正在谈论的HTTP标头，属于HTTP协议。这就是为什么您未能在gRPC层中获取它。如果您使用MVC框架，请确保在Servlet或Controller中检索它（添加HttpServletRequest参数如下）。

  @GetMapping("hello")
  public CompletableFuture<GreetingResponse> greet(HttpServletRequest request, @RequestParams someParam) {
  // call gRPC Service here
}

然后您可以通过request.getHeader（"User-Agent"）访问标头。

Answer 2

除非您使用的是第三方提供的自定义grpc java codegen插件，否则grpc java库附带的标准grpc codegen插件https://search.maven.org/search?q=a:protoc-gen grpc java g：io。grpc不支持自定义方法选项，例如

option (grpc.gateway.protoc_gen_swagger.options.openapiv2_operation)

所以你不能利用这些选项。grpc-java可以做的是通过客户端和服务器拦截器发送和接收io.grpc.元数据。参见示例：https://github.com/grpc/grpc-java/tree/v1.33.1/examples/src/main/java/io/grpc/examples/header