From d08dfce4fc4f344fc4331f859097f180abe1417b Mon Sep 17 00:00:00 2001 From: "Jixiang.jjx" Date: Fri, 29 Apr 2022 15:11:35 +0800 Subject: [PATCH] [ISSUE #3949] allow to add user-defined gRPC Server interceptors. --- .../org/apache/rocketmq/proxy/grpc/GrpcServerBuilder.java | 6 ++++++ .../proxy/grpc/interceptor/InterceptorConstants.java | 3 --- 2 files changed, 6 insertions(+), 3 deletions(-) diff --git a/proxy/src/main/java/org/apache/rocketmq/proxy/grpc/GrpcServerBuilder.java b/proxy/src/main/java/org/apache/rocketmq/proxy/grpc/GrpcServerBuilder.java index 0c4be76384..02e6d3a9df 100644 --- a/proxy/src/main/java/org/apache/rocketmq/proxy/grpc/GrpcServerBuilder.java +++ b/proxy/src/main/java/org/apache/rocketmq/proxy/grpc/GrpcServerBuilder.java @@ -17,6 +17,7 @@ package org.apache.rocketmq.proxy.grpc; import io.grpc.BindableService; +import io.grpc.ServerInterceptor; import io.grpc.ServerServiceDefinition; import io.grpc.netty.shaded.io.grpc.netty.GrpcSslContexts; import io.grpc.netty.shaded.io.grpc.netty.NettyServerBuilder; @@ -100,6 +101,11 @@ public class GrpcServerBuilder { return this; } + public GrpcServerBuilder appendInterceptor(ServerInterceptor interceptor) { + this.serverBuilder.intercept(interceptor); + return this; + } + public GrpcServer build() { return new GrpcServer(this.serverBuilder.build()); } diff --git a/proxy/src/main/java/org/apache/rocketmq/proxy/grpc/interceptor/InterceptorConstants.java b/proxy/src/main/java/org/apache/rocketmq/proxy/grpc/interceptor/InterceptorConstants.java index 04d5137e14..0fe16789dd 100644 --- a/proxy/src/main/java/org/apache/rocketmq/proxy/grpc/interceptor/InterceptorConstants.java +++ b/proxy/src/main/java/org/apache/rocketmq/proxy/grpc/interceptor/InterceptorConstants.java @@ -21,9 +21,6 @@ import io.grpc.Context; import io.grpc.Metadata; public class InterceptorConstants { - private InterceptorConstants() { - } - public static final Context.Key METADATA = Context.key("rpc-metadata"); /**