From 843d44519bd5212db8dcbf7f12178e33b668b13b Mon Sep 17 00:00:00 2001
From: aminst <aminst79@gmail.com>
Date: Mon, 26 Feb 2024 22:22:52 -0500
Subject: [PATCH] Increase shardnode message size

---
 pkg/shardnode/client.go | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/pkg/shardnode/client.go b/pkg/shardnode/client.go
index 3262a70..c427925 100644
--- 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