add parameter for MaxInBoundMessageSize

gumreal · Mar 12, 2022 · 54d16a2 · 54d16a2
1 parent cb10e9b
commit 54d16a2
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 2 deletions.
diff --git a/build.gradle b/build.gradle
@@ -41,7 +41,7 @@ apply plugin: 'signing'
 
 group = 'io.dgraph'
 archivesBaseName = 'dgraph4j'
-version = '21.03.2.p15.20220310b'
+version = '21.03.2.p16.20220312a'
 sourceCompatibility = 1.8
 targetCompatibility = 1.8
 

diff --git a/src/main/java/io/dgraph/wrapper/ClientBuilder.java b/src/main/java/io/dgraph/wrapper/ClientBuilder.java
@@ -16,6 +16,7 @@ public class ClientBuilder {
   private Logger logger = LoggerFactory.getLogger(ClientBuilder.class.getName());
   private Set<String> uriSet = new HashSet<>();
   private List<AlphaUri> alphaList = new ArrayList<>();
+  private int maxInBoundBytes = 4 * 1024 * 1024;
 
   private ClientBuilder() {}
 
@@ -30,6 +31,10 @@ public static ClientBuilder newInstance() {
    * @throws WrapperException
    */
   public ClientBuilder withAlpha(String host, int port) throws WrapperException {
+    return withAlpha(host, port, 0);
+  }
+
+  public ClientBuilder withAlpha(String host, int port, int maxInBytes) throws WrapperException {
     if (GeneralHelper.isEmpty(host) || port <= 0) {
       throw new WrapperException("invalid alpha uri");
     }
@@ -41,6 +46,10 @@ public ClientBuilder withAlpha(String host, int port) throws WrapperException {
       alphaList.add(new AlphaUri(host, port));
     }
 
+    if (maxInBytes > this.maxInBoundBytes) {
+      this.maxInBoundBytes = maxInBytes;
+    }
+
     return this;
   }
 
@@ -67,7 +76,11 @@ public DgraphClient build() throws WrapperException {
    * @return
    */
   private DgraphGrpc.DgraphStub makeStub(String addr, int port) {
-    ManagedChannel channel = ManagedChannelBuilder.forAddress(addr, port).usePlaintext().build();
+    ManagedChannel channel =
+        ManagedChannelBuilder.forAddress(addr, port)
+            .usePlaintext()
+            .maxInboundMessageSize(maxInBoundBytes)
+            .build();
     return DgraphGrpc.newStub(channel);
   }