fixes networknt#753 handle text/plain and content type missing in Bod…

…yHandler (networknt#755)
younggwon1 · Aug 25, 2020 · d1828e3 · d1828e3
1 parent f45b529
commit d1828e3
Showing 1 changed file with 13 additions and 5 deletions.
diff --git a/body/src/main/java/com/networknt/body/BodyHandler.java b/body/src/main/java/com/networknt/body/BodyHandler.java
@@ -87,12 +87,12 @@ public BodyHandler() {
     public void handleRequest(final HttpServerExchange exchange) throws Exception {
         // parse the body to map or list if content type is application/json
         String contentType = exchange.getRequestHeaders().getFirst(Headers.CONTENT_TYPE);
+        if (exchange.isInIoThread()) {
+            exchange.dispatch(this);
+            return;
+        }
+        exchange.startBlocking();
         if (contentType != null) {
-            if (exchange.isInIoThread()) {
-                exchange.dispatch(this);
-                return;
-            }
-            exchange.startBlocking();
             try {
                 if (contentType.startsWith("application/json")) {
                     InputStream inputStream = exchange.getInputStream();
@@ -103,6 +103,10 @@ public void handleRequest(final HttpServerExchange exchange) throws Exception {
                     }
                     // attach the parsed request body into exchange if the body parser is enabled
                     attachJsonBody(exchange, unparsedRequestBody);
+                } else if (contentType.startsWith("text/plain")) {
+                    InputStream inputStream = exchange.getInputStream();
+                    String unparsedRequestBody = StringUtils.inputStreamToString(inputStream, StandardCharsets.UTF_8);
+                    exchange.putAttachment(REQUEST_BODY, unparsedRequestBody);
                 } else if (contentType.startsWith("multipart/form-data") || contentType.startsWith("application/x-www-form-urlencoded")) {
                     // attach the parsed request body into exchange if the body parser is enabled
                     attachFormDataBody(exchange);
@@ -115,6 +119,10 @@ public void handleRequest(final HttpServerExchange exchange) throws Exception {
                 setExchangeStatus(exchange, CONTENT_TYPE_MISMATCH, contentType);
                 return;
             }
+        } else {
+            // attach the stream to the exchange if the content type is missing.
+            InputStream inputStream = exchange.getInputStream();
+            exchange.putAttachment(REQUEST_BODY, inputStream);
         }
         Handler.next(exchange, next);
     }