feat: add gRPC max message size configuration

Introduce a `--max-msg-size` flag to configure the maximum gRPC message size for receiving and sending. Update the gRPC server initialization to include these size limits, ensuring better control over large message handling.

Author: tzdybal

cmd/centralized-sequencer/main.go

@@ -13,6 +13,8 @@ import (
 	"syscall"
 	"time"
 
+	"google.golang.org/grpc"
+
 	"github.com/prometheus/client_golang/prometheus/promhttp"
 
 	"github.com/rollkit/centralized-sequencer/sequencing"
@@ -39,7 +41,10 @@ func main() {
 		db_path        string
 		metricsEnabled bool
 		metricsAddress string
+		maxMsgSize     int
 	)
+
+	// Define flags
 	flag.StringVar(&host, "host", defaultHost, "centralized sequencer host")
 	flag.StringVar(&port, "port", defaultPort, "centralized sequencer port")
 	flag.BoolVar(&listenAll, "listen-all", false, "listen on all network interfaces (0.0.0.0) instead of just localhost")
@@ -51,6 +56,7 @@ func main() {
 	flag.StringVar(&db_path, "db_path", "", "path to the database")
 	flag.BoolVar(&metricsEnabled, "metrics", false, "Enable Prometheus metrics")
 	flag.StringVar(&metricsAddress, "metrics-address", ":8080", "Address to expose Prometheus metrics")
+	flag.IntVar(&maxMsgSize, "max-msg-size", 4*1024*1024, "Maximum gRPC message size (in bytes)")
 
 	flag.Parse()
 
@@ -95,9 +101,17 @@ func main() {
 	if err != nil {
 		log.Fatalf("Failed to create centralized sequencer: %v", err)
 	}
-	grpcServer := sequencingGRPC.NewServer(centralizedSeq, centralizedSeq, centralizedSeq)
 
-	log.Println("Starting centralized sequencing gRPC server on port 50051...")
+	// Create gRPC server with max message size options
+	grpcOpts := []grpc.ServerOption{
+		grpc.MaxRecvMsgSize(maxMsgSize),
+		grpc.MaxSendMsgSize(maxMsgSize),
+	}
+
+	// Create and register the sequencing proxy
+	grpcServer := sequencingGRPC.NewServer(centralizedSeq, centralizedSeq, centralizedSeq, grpcOpts...)
+
+	log.Printf("Starting centralized sequencing gRPC server on %s...\n", address)
 	if err := grpcServer.Serve(lis); err != nil {
 		log.Fatalf("Failed to serve: %v", err)
 	}