opensource/rocketmq
3
0
Fork 0
mirror of https://github.com/apache/rocketmq.git synced 2026-09-21 13:49:50 +08:00
Code Issues Packages Projects Releases Wiki Activity

[ISSUE #3949] allow to config interceptor during building steps.

Browse Source
This commit is contained in:
Jixiang.jjx
2022-07-13 11:29:29 +08:00
committed by zhouxiang
parent 3ede21afbb
commit 2d594814b8
2 changed files with 12 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
proxy/src/main/java/org/apache/rocketmq/proxy/ProxyStartup.java
+1
View File
@@ -68,6 +68,7 @@ public class ProxyStartup {
// create grpcServer
GrpcServer grpcServer = GrpcServerBuilder.newBuilder(executor)
.addService(createServiceProcessor())
.configInterceptor()
.build();
PROXY_START_AND_SHUTDOWN.appendStartAndShutdown(grpcServer);
proxy/src/main/java/org/apache/rocketmq/proxy/grpc/GrpcServerBuilder.java
+11 -8
View File
@@ -27,9 +27,10 @@ import io.grpc.netty.shaded.io.netty.channel.socket.nio.NioServerSocketChannel;
import io.grpc.netty.shaded.io.netty.handler.ssl.ClientAuth;
import io.grpc.netty.shaded.io.netty.handler.ssl.util.InsecureTrustManagerFactory;
import io.grpc.netty.shaded.io.netty.handler.ssl.util.SelfSignedCertificate;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.nio.file.Files;
import java.nio.file.Paths;
import java.security.cert.CertificateException;
import java.util.List;
import java.util.concurrent.ThreadPoolExecutor;
@@ -83,8 +84,6 @@ public class GrpcServerBuilder {
.executor(executor);
}
configInterceptor(serverBuilder);
log.info(
"grpc server has built. port: {}, tlsKeyPath: {}, tlsCertPath: {}, threadPool: {}, queueCapacity: {}, "
+ "boosLoop: {}, workerLoop: {}, maxInboundMessageSize: {}",
@@ -122,8 +121,8 @@ public class GrpcServerBuilder {
String tlsKeyPath = ConfigurationManager.getProxyConfig().getGrpcTlsKeyPath();
String tlsCertPath = ConfigurationManager.getProxyConfig().getGrpcTlsCertPath();
try (InputStream serverKeyInputStream = new FileInputStream(tlsKeyPath);
InputStream serverCertificateStream = new FileInputStream(tlsCertPath)) {
try (InputStream serverKeyInputStream = Files.newInputStream(Paths.get(tlsKeyPath));
InputStream serverCertificateStream = Files.newInputStream(Paths.get(tlsCertPath))) {
serverBuilder.sslContext(GrpcSslContexts.forServer(serverCertificateStream, serverKeyInputStream)
.trustManager(InsecureTrustManagerFactory.INSTANCE)
.clientAuth(ClientAuth.NONE)
@@ -134,17 +133,21 @@ public class GrpcServerBuilder {
}
}
protected void configInterceptor(NettyServerBuilder serverBuilder) {
public GrpcServerBuilder configInterceptor() {
// grpc interceptors, including acl, logging etc.
if (ConfigurationManager.getProxyConfig().isEnableACL()) {
List<AccessValidator> accessValidators = ServiceProvider.load(ServiceProvider.ACL_VALIDATOR_ID, AccessValidator.class);
if (accessValidators.isEmpty()) {
throw new IllegalArgumentException("Load AccessValidator failed");
}
serverBuilder.intercept(new AuthenticationInterceptor(accessValidators));
this.serverBuilder.intercept(new AuthenticationInterceptor(accessValidators));
}
serverBuilder.intercept(new ContextInterceptor())
this.serverBuilder
.intercept(new ContextInterceptor())
.intercept(new HeaderInterceptor());
return this;
}
}