Adjusted max rpc message size

M1chlCZ · Aug 1, 2023 · 2d54a50 · 2d54a50
1 parent 7c0ffce
commit 2d54a50
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/grpcServer/server.go b/grpcServer/server.go
@@ -10,6 +10,8 @@ import (
 	"nsfw_sherlock/utils"
 )
 
+const maxMessageSize = 100 * 1024 * 1024
+
 func StartGrpcServer() {
 	// Dry run for TF
 	test, err := common.TestPictureNSFW("./pic.jpg")
@@ -36,7 +38,12 @@ func StartGrpcServer() {
 
 	s := Server{}
 	//, grpc.UnaryInterceptor(serverInterceptor)
-	grpcServer := grpc.NewServer(grpc.Creds(insecure.NewCredentials()))
+	opts := []grpc.ServerOption{
+		grpc.Creds(insecure.NewCredentials()),
+		grpc.MaxRecvMsgSize(maxMessageSize),
+		grpc.MaxSendMsgSize(maxMessageSize),
+	}
+	grpcServer := grpc.NewServer(opts...)
 	grpcModels.RegisterNSFWServer(grpcServer, &s)
 	if err := grpcServer.Serve(lis); err != nil {
 		utils.WrapErrorLog(err.Error())