Merge pull request #102 from anti-duhring/AUEML-2778

feat: adding param MaxReceiveMessageSize on NewServer
gothunder · Dec 30, 2024 · c65c64f · c65c64f
2 parents 4dd30d4 + 5c03e77
commit c65c64f
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/pkg/grpc/server.go b/pkg/grpc/server.go
@@ -18,8 +18,9 @@ type BareServer struct {
 type NewServerParams struct {
 	fx.In
 
-	Logger       *zerolog.Logger
-	Interceptors []grpc.UnaryServerInterceptor `group:"interceptors"`
+	Logger                *zerolog.Logger
+	Interceptors          []grpc.UnaryServerInterceptor `group:"interceptors"`
+	MaxReceiveMessageSize *int
 }
 
 func NewServer(params NewServerParams) *BareServer {
@@ -32,13 +33,20 @@ func NewServer(params NewServerParams) *BareServer {
 		append(params.Interceptors, grpcLoggerInterceptor(params.Logger))...,
 	)
 
+	// default max message size is 4MB
+	maxReceiveMessageSize := 4 * 1024 * 1024
+	if params.MaxReceiveMessageSize != nil {
+		maxReceiveMessageSize = *params.MaxReceiveMessageSize
+	}
+
 	sv := grpc.NewServer(
 		grpc.StatsHandler(
 			otelgrpc.NewServerHandler(),
 		),
 		grpc.ChainUnaryInterceptor(
 			params.Interceptors...,
 		),
+		grpc.MaxRecvMsgSize(maxReceiveMessageSize),
 	)
 	grpcServer.Server = sv