Increase shardnode message size

dsg-uwaterloo · Feb 27, 2024 · 843d445 · 843d445
1 parent f9a9c26
commit 843d445
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/pkg/shardnode/client.go b/pkg/shardnode/client.go
@@ -3,6 +3,7 @@ package shardnode
 import (
 	"context"
 	"fmt"
+	"math"
 
 	oramnodepb "github.com/dsg-uwaterloo/oblishard/api/oramnode"
 	"github.com/dsg-uwaterloo/oblishard/pkg/config"
@@ -65,6 +66,7 @@ func StartOramNodeRPCClients(endpoints []config.OramNodeEndpoint) (map[int]Repli
 		conn, err := grpc.Dial(serverAddr,
 			grpc.WithTransportCredentials(insecure.NewCredentials()),
 			grpc.WithUnaryInterceptor(rpc.ContextPropagationUnaryClientInterceptor()),
+			grpc.WithDefaultCallOptions(grpc.MaxCallRecvMsgSize(math.MaxInt64), grpc.MaxCallSendMsgSize(math.MaxInt64)),
 		)
 		if err != nil {
 			return nil, err