chore: update max msg size

takenet · Feb 7, 2025 · 70f58f7 · 70f58f7
1 parent 2bb3b3e
commit 70f58f7
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/internal/config/service.go b/internal/config/service.go
@@ -30,6 +30,11 @@ var GrpcServerMaxConnectionAgeGrace = Create(&ViperConfigKey{
 	Key: "grpc.server.max_connection_age_grace",
 })
 
+var GrpcServerMaxMsgSize = Create(&ViperConfigKey{
+	Key:     "grpc.server.max_msg_size",
+	Default: 1024 * 1024 * 11,
+})
+
 var TlsServerCertFilePaths = Create(&ViperConfigKey{
 	Key: "tls.server.cert_file_paths",
 })

diff --git a/internal/service/deckard_service.go b/internal/service/deckard_service.go
@@ -68,6 +68,7 @@ func NewDeckardService(qpool queue.DeckardQueue, queueConfigurationService queue
 
 func (d *Deckard) ServeGRPCServer(ctx context.Context) (*grpc.Server, error) {
 	port := config.GrpcPort.GetInt()
+	maxMsgSize := config.GrpcServerMaxMsgSize.GetInt()
 
 	listen, err := net.Listen("tcp", fmt.Sprint(":", port))
 	if err != nil {
@@ -78,6 +79,8 @@ func (d *Deckard) ServeGRPCServer(ctx context.Context) (*grpc.Server, error) {
 		grpc.UnaryInterceptor(otelgrpc.UnaryServerInterceptor()),
 		grpc.StreamInterceptor(otelgrpc.StreamServerInterceptor()),
 		getGrpcKeepaliveParams(),
+		grpc.MaxRecvMsgSize(maxMsgSize),
+		grpc.MaxSendMsgSize(maxMsgSize),
 	}
 
 	credentials, err := loadTLSCredentials()