Increase message size

dsg-uwaterloo · Feb 27, 2024 · f9a9c26 · f9a9c26
1 parent 8000560
commit f9a9c26
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 1 deletion.
diff --git a/pkg/client/client.go b/pkg/client/client.go
@@ -195,6 +195,7 @@ func StartRouterRPCClients(endpoints []config.RouterEndpoint) (RouterClients, er
 		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

diff --git a/pkg/oramnode/client.go b/pkg/oramnode/client.go
@@ -3,6 +3,7 @@ package oramnode
 import (
 	"context"
 	"fmt"
+	"math"
 	"math/rand"
 
 	shardnodepb "github.com/dsg-uwaterloo/oblishard/api/shardnode"
@@ -100,7 +101,7 @@ func StartShardNodeRPCClients(endpoints []config.ShardNodeEndpoint) (map[int]Rep
 	for _, endpoint := range endpoints {
 		serverAddr := fmt.Sprintf("%s:%d", endpoint.IP, endpoint.Port)
 		log.Debug().Msgf("Starting ShardNode RPC client for endpoint: %s", serverAddr)
-		conn, err := grpc.Dial(serverAddr, grpc.WithTransportCredentials(insecure.NewCredentials()))
+		conn, err := grpc.Dial(serverAddr, grpc.WithTransportCredentials(insecure.NewCredentials()), grpc.WithDefaultCallOptions(grpc.MaxCallRecvMsgSize(math.MaxInt64), grpc.MaxCallSendMsgSize(math.MaxInt64)))
 		if err != nil {
 			return nil, err
 		}

diff --git a/pkg/router/client.go b/pkg/router/client.go
@@ -2,6 +2,7 @@ package router
 
 import (
 	"fmt"
+	"math"
 
 	shardnodepb "github.com/dsg-uwaterloo/oblishard/api/shardnode"
 	"github.com/dsg-uwaterloo/oblishard/pkg/config"
@@ -27,6 +28,7 @@ func StartShardNodeRPCClients(endpoints []config.ShardNodeEndpoint) (map[int]Rep
 		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