fix: max size for grpc calls in go

shifty11 · Feb 27, 2024 · 9f6da1c · 9f6da1c
1 parent c83274f
commit 9f6da1c
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 3 deletions.
diff --git a/tools/kystrap/grpcall/grpcall.go b/tools/kystrap/grpcall/grpcall.go
@@ -102,7 +102,14 @@ func (gc *GrpcCaller) dial() (*grpc.ClientConn, error) {
 	var creds credentials.TransportCredentials
 
 	grpcurlUA := "kystrap"
-	opts = append(opts, grpc.WithUserAgent(grpcurlUA))
+	maxMessageSize := 2 * 1024 * 1024 * 1024 // 2 GB
+	opts = append(
+		opts,
+		grpc.WithUserAgent(grpcurlUA),
+		grpc.WithDefaultCallOptions(
+			grpc.MaxCallRecvMsgSize(maxMessageSize),
+			grpc.MaxCallSendMsgSize(maxMessageSize)),
+	)
 
 	network := "tcp"
 

diff --git a/tools/kystrap/templates/go/scripts/test.sh b/tools/kystrap/templates/go/scripts/test.sh
@@ -4,7 +4,7 @@
 #               Variables               #
 #########################################
 # Key, config, and expected values (CHANGE THESE TO MATCH YOUR RUNTIME!!!)
-KEY=4943400
+KEY="4943400"
 API="https://rpc.kyve.network"
 CONFIG='{"network":"kyve-1","rpc":"'$API'"}'
 EXPECTED_SUMMARY="\"$KEY\""

diff --git a/tools/kystrap/templates/go/server/start.go b/tools/kystrap/templates/go/server/start.go
@@ -11,6 +11,7 @@ import (
 const (
 	host = "0.0.0.0"
 	port = "50051"
+	maxMessageSize = 2 * 1024 * 1024 * 1024 	// 2 GB
 )
 
 func StartServer() {
@@ -21,7 +22,7 @@ func StartServer() {
 	}
 
 	// Create a new gRPC server instance
-	server := grpc.NewServer()
+	server := grpc.NewServer(grpc.MaxRecvMsgSize(maxMessageSize), grpc.MaxSendMsgSize(maxMessageSize))
 
 	// Register the Tendermint service with the gRPC server
 	pb.RegisterRuntimeServiceServer(server, &{{ .name | ToPascal }}Server{})