Copy value of Content-Type from Netty response headers to AWS respons…

…e builder
quarkusio · Mar 26, 2021 · 5ebe4d3 · 5ebe4d3
1 parent 2fbb0a6
commit 5ebe4d3
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/...on-lambda-http/runtime/src/main/java/io/quarkus/amazon/lambda/http/LambdaHttpHandler.java b/...on-lambda-http/runtime/src/main/java/io/quarkus/amazon/lambda/http/LambdaHttpHandler.java
@@ -6,6 +6,7 @@
 import java.nio.channels.WritableByteChannel;
 import java.nio.charset.StandardCharsets;
 import java.util.Base64;
+import java.util.HashMap;
 import java.util.HashSet;
 import java.util.Map;
 import java.util.Set;
@@ -25,6 +26,7 @@
 import io.netty.handler.codec.http.DefaultLastHttpContent;
 import io.netty.handler.codec.http.HttpContent;
 import io.netty.handler.codec.http.HttpHeaderNames;
+import io.netty.handler.codec.http.HttpHeaders;
 import io.netty.handler.codec.http.HttpMethod;
 import io.netty.handler.codec.http.HttpResponse;
 import io.netty.handler.codec.http.HttpVersion;
@@ -92,6 +94,16 @@ public void handleMessage(Object msg) {
                     HttpResponse res = (HttpResponse) msg;
                     responseBuilder.setStatusCode(res.status().code());
 
+                    final HttpHeaders responseHeaders = res.headers();
+                    if (responseHeaders != null) {
+                        final String contentType = responseHeaders.getAsString(HttpHeaderNames.CONTENT_TYPE);
+                        if (contentType != null && contentType.trim().length() > 0) {
+                            final Map<String, String> headers = new HashMap<>();
+                            headers.put("Content-Type", contentType);
+                            responseBuilder.setHeaders(headers);
+                        }
+                    }
+
                     Headers multiValueHeaders = new Headers();
                     responseBuilder.setMultiValueHeaders(multiValueHeaders);
                     for (String name : res.headers().names()) {