Add MAX_ENTITY_SIZE undertow option in http server config (#574)

Author: Squiry

http/http-server-common/src/main/java/ru/tinkoff/kora/http/server/common/HttpServerConfig.java

@@ -1,5 +1,6 @@
 package ru.tinkoff.kora.http.server.common;
 
+import ru.tinkoff.kora.common.util.Size;
 import ru.tinkoff.kora.config.common.annotation.ConfigValueExtractor;
 import ru.tinkoff.kora.http.server.common.telemetry.HttpServerTelemetryConfig;
 
@@ -33,4 +34,8 @@ default Duration shutdownWait() {
     }
 
     HttpServerTelemetryConfig telemetry();
+
+    default Size maxRequestBodySize() {
+        return Size.of(256, Size.Type.MiB);
+    }
 }

http/http-server-common/src/test/java/ru/tinkoff/kora/http/server/common/router/HttpServerHandlerProcessTests.java

@@ -6,6 +6,7 @@
 import org.junit.jupiter.params.provider.MethodSource;
 import org.mockito.Mockito;
 import ru.tinkoff.kora.application.graph.All;
+import ru.tinkoff.kora.common.util.Size;
 import ru.tinkoff.kora.http.common.body.HttpBody;
 import ru.tinkoff.kora.http.common.body.HttpBodyInput;
 import ru.tinkoff.kora.http.common.header.HttpHeaders;
@@ -156,7 +157,8 @@ private HttpServerConfig config(boolean ignoreTrailingSlash) {
                 new $HttpServerTelemetryConfig_HttpServerLoggingConfig_ConfigValueExtractor.HttpServerLoggingConfig_Defaults(),
                 new $HttpServerTelemetryConfig_HttpServerMetricsConfig_ConfigValueExtractor.HttpServerMetricsConfig_Defaults(),
                 new $HttpServerTelemetryConfig_HttpServerTracingConfig_ConfigValueExtractor.HttpServerTracingConfig_Defaults()
-            )
+            ),
+            Size.ofBytesBinary(1024)
         );
     }
 

http/http-server-common/src/test/java/ru/tinkoff/kora/http/server/common/router/HttpServerHandlerTest.java

@@ -3,6 +3,7 @@
 import org.assertj.core.api.Assertions;
 import org.junit.jupiter.api.Test;
 import org.mockito.Mockito;
+import ru.tinkoff.kora.common.util.Size;
 import ru.tinkoff.kora.http.server.common.$HttpServerConfig_ConfigValueExtractor;
 import ru.tinkoff.kora.http.server.common.HttpServerConfig;
 import ru.tinkoff.kora.http.server.common.HttpServerResponse;
@@ -151,7 +152,9 @@ private HttpServerConfig config(boolean ignoreTrailingSlash) {
                 new $HttpServerTelemetryConfig_HttpServerLoggingConfig_ConfigValueExtractor.HttpServerLoggingConfig_Defaults(),
                 new $HttpServerTelemetryConfig_HttpServerMetricsConfig_ConfigValueExtractor.HttpServerMetricsConfig_Defaults(),
                 new $HttpServerTelemetryConfig_HttpServerTracingConfig_ConfigValueExtractor.HttpServerTracingConfig_Defaults()
-            ));
+            ),
+            Size.ofBytesBinary(1024)
+        );
     }
 
     private HttpServerRequestHandler handler(String method, String route) {

http/http-server-common/src/testFixtures/java/ru/tinkoff/kora/http/server/common/HttpServerTestKit.java

@@ -20,6 +20,7 @@
 import ru.tinkoff.kora.common.liveness.LivenessProbeFailure;
 import ru.tinkoff.kora.common.readiness.ReadinessProbe;
 import ru.tinkoff.kora.common.readiness.ReadinessProbeFailure;
+import ru.tinkoff.kora.common.util.Size;
 import ru.tinkoff.kora.http.common.body.HttpBody;
 import ru.tinkoff.kora.http.common.body.HttpBodyOutput;
 import ru.tinkoff.kora.http.common.header.HttpHeaders;
@@ -968,7 +969,8 @@ protected void startServer(boolean ignoreTrailingSlash, List<HttpServerIntercept
                 new $HttpServerTelemetryConfig_HttpServerLoggingConfig_ConfigValueExtractor.HttpServerLoggingConfig_Defaults(),
                 new $HttpServerTelemetryConfig_HttpServerMetricsConfig_ConfigValueExtractor.HttpServerMetricsConfig_Defaults(),
                 new $HttpServerTelemetryConfig_HttpServerTracingConfig_ConfigValueExtractor.HttpServerTracingConfig_Defaults()
-            )
+            ),
+            Size.of(1, Size.Type.GiB)
         );
         var publicApiHandler = new HttpServerHandler(List.of(handlers), interceptors, config);
         this.httpServer = this.httpServer(valueOf(config), publicApiHandler, this.telemetry);

http/http-server-undertow/src/main/java/ru/tinkoff/kora/http/server/undertow/UndertowHttpServer.java

@@ -1,6 +1,7 @@
 package ru.tinkoff.kora.http.server.undertow;
 
 import io.undertow.Undertow;
+import io.undertow.UndertowOptions;
 import io.undertow.server.HttpHandler;
 import io.undertow.server.HttpServerExchange;
 import io.undertow.server.handlers.GracefulShutdownHandler;
@@ -113,7 +114,9 @@ private Undertow createServer() {
             .setWorker(this.xnioWorker)
             .setServerOption(Options.READ_TIMEOUT, ((int) config.socketReadTimeout().toMillis()))
             .setServerOption(Options.WRITE_TIMEOUT, ((int) config.socketWriteTimeout().toMillis()))
-            .setServerOption(Options.KEEP_ALIVE, config.socketKeepAliveEnabled());
+            .setServerOption(Options.KEEP_ALIVE, config.socketKeepAliveEnabled())
+            .setServerOption(UndertowOptions.MAX_ENTITY_SIZE, config.maxRequestBodySize().toBytes())
+            ;
 
         if (this.configurer != null) {
             undertow = this.configurer.configure(undertow);