From 0c92ecb498eb2ad9df647ca1d63da0214225b751 Mon Sep 17 00:00:00 2001
From: atyamlalithya <99050095+atyamlalithya@users.noreply.github.com>
Date: Fri, 10 Jun 2022 11:12:00 -0700
Subject: [PATCH] Update AWSLambdaUtils.java

Added this fix to preserve headers from original payload
---
 .../cloud/function/adapter/aws/AWSLambdaUtils.java              | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/spring-cloud-function-adapters/spring-cloud-function-adapter-aws/src/main/java/org/springframework/cloud/function/adapter/aws/AWSLambdaUtils.java b/spring-cloud-function-adapters/spring-cloud-function-adapter-aws/src/main/java/org/springframework/cloud/function/adapter/aws/AWSLambdaUtils.java
index 5fd4dda97..35dd5a537 100644
--- a/spring-cloud-function-adapters/spring-cloud-function-adapter-aws/src/main/java/org/springframework/cloud/function/adapter/aws/AWSLambdaUtils.java
+++ b/spring-cloud-function-adapters/spring-cloud-function-adapter-aws/src/main/java/org/springframework/cloud/function/adapter/aws/AWSLambdaUtils.java
@@ -121,7 +121,7 @@ final class AWSLambdaUtils {
 					logger.info("Incoming request is API Gateway v2.0");
 					messageBuilder = createMessageBuilderForPOJOFunction(objectMapper, (Map) request);
 				}
-				Object providedHeaders = ((Map) request).remove("headers");
+				Object providedHeaders = ((Map) request).get("headers");
 				if (providedHeaders != null && providedHeaders instanceof Map) {
 					messageBuilder = MessageBuilder.withPayload(request);
 					messageBuilder.removeHeader("headers");