Grpc max message size support config by user. (#564)

* grpc max message size support config by user. * code format. * code format. * code clean. * reuse GrpcRaftRpcFactory.RPC_MAX_INBOUND_MESSAGE_SIZE in RpcClient. * add prefix 'jraft'.
sofastack · Apr 4, 2021 · 1b9f6d2 · 1b9f6d2
1 parent 5c12c2a
commit 1b9f6d2
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 3 deletions.
diff --git a/jraft-extension/rpc-grpc-impl/src/main/java/com/alipay/sofa/jraft/rpc/impl/GrpcClient.java b/jraft-extension/rpc-grpc-impl/src/main/java/com/alipay/sofa/jraft/rpc/impl/GrpcClient.java
@@ -183,6 +183,7 @@ private ManagedChannel getChannel(final Endpoint endpoint) {
             final ManagedChannel ch = ManagedChannelBuilder.forAddress(ep.getIp(), ep.getPort()) //
                 .usePlaintext() //
                 .directExecutor() //
+                .maxInboundMessageSize(GrpcRaftRpcFactory.RPC_MAX_INBOUND_MESSAGE_SIZE) //
                 .build();
             // channel connection event
             ch.notifyWhenStateChanged(ConnectivityState.READY, () -> {

diff --git a/...ension/rpc-grpc-impl/src/main/java/com/alipay/sofa/jraft/rpc/impl/GrpcRaftRpcFactory.java b/...ension/rpc-grpc-impl/src/main/java/com/alipay/sofa/jraft/rpc/impl/GrpcRaftRpcFactory.java
@@ -41,9 +41,14 @@
 public class GrpcRaftRpcFactory implements RaftRpcFactory {
 
     static final String             FIXED_METHOD_NAME              = "_call";
-    static final int                RPC_SERVER_PROCESSOR_POOL_SIZE = SystemPropertyUtil.getInt(
-                                                                       "grpc.default_rpc_server_processor_pool_size",
-                                                                       100);
+    static final int                RPC_SERVER_PROCESSOR_POOL_SIZE = SystemPropertyUtil
+                                                                       .getInt(
+                                                                           "jraft.grpc.default_rpc_server_processor_pool_size",
+                                                                           100);
+
+    static final int                RPC_MAX_INBOUND_MESSAGE_SIZE   = SystemPropertyUtil.getInt(
+                                                                       "jraft.grpc.max_inbound_message_size.bytes",
+                                                                       4 * 1024 * 1024);
 
     static final RpcResponseFactory RESPONSE_FACTORY               = new GrpcResponseFactory();
 
@@ -86,6 +91,7 @@ public RpcServer createRpcServer(final Endpoint endpoint, final ConfigHelper<Rpc
         final Server server = ServerBuilder.forPort(port) //
             .fallbackHandlerRegistry(handlerRegistry) //
             .directExecutor() //
+            .maxInboundMessageSize(RPC_MAX_INBOUND_MESSAGE_SIZE) //
             .build();
         final RpcServer rpcServer = new GrpcServer(server, handlerRegistry, this.parserClasses, getMarshallerRegistry());
         if (helper != null) {