use stream api

micronaut-projects · Apr 5, 2023 · 52b852e · 52b852e
1 parent 0f624cb
commit 52b852e
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/...-aws-api-proxy/src/main/java/io/micronaut/function/aws/proxy/MicronautResponseWriter.java b/...-aws-api-proxy/src/main/java/io/micronaut/function/aws/proxy/MicronautResponseWriter.java
@@ -123,8 +123,9 @@ private static Map<String, String> getHeaders(AwsProxyResponse awsProxyResponse)
         Map<String, String> headers = new LinkedHashMap<>();
         for (Map.Entry<String, List<String>> entry : multiValueHeaders.entrySet()) {
             List<String> headerValues = entry.getValue();
-            if (headerValues != null && headerValues.size() == 1) {
-                headers.put(entry.getKey(), headerValues.get(0));
+            if (headerValues != null) {
+                headerValues.stream().findFirst()
+                    .ifPresent(value -> headers.put(entry.getKey(), value));
             }
         }
         return headers;