Set the maximum length of the content at the session layer

naver · Sep 17, 2023 · 2b2a212 · 2b2a212
1 parent c7ab774
commit 2b2a212
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/scavenger-collector/src/main/kotlin/com/navercorp/scavenger/config/ArmeriaConfiguration.kt b/scavenger-collector/src/main/kotlin/com/navercorp/scavenger/config/ArmeriaConfiguration.kt
@@ -43,14 +43,16 @@ class ArmeriaConfiguration(
     @Bean
     fun armeriaServiceInitializer(tomcatService: TomcatService): ArmeriaServerConfigurator {
         return ArmeriaServerConfigurator { serviceBuilder: ServerBuilder ->
+            val maxMessageSize = maxMessageSize.toBytes().toInt()
             serviceBuilder.service("prefix:/", tomcatService)
                 .service(GrpcService.builder()
                     .addService(grpcAgentController)
                     .addExceptionMapping(IllegalArgumentException::class.java, Status.INVALID_ARGUMENT)
                     .addExceptionMapping(LicenseKeyNotFoundException::class.java, Status.UNAUTHENTICATED)
-                    .maxRequestMessageLength(maxMessageSize.toBytes().toInt())
+                    .maxRequestMessageLength(maxMessageSize)
                     .build()
                 )
+                .maxRequestLength(maxMessageSize.toLong())
         }
     }
 }