SpringCloud/spring-cloud-function
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Ensure user set headers in API Gateway are properly propagated to message headers

Browse Source
This commit is contained in:
Oleg Zhurakousky
2021-05-18 16:33:04 +02:00
parent 25387b83be
commit c07aa97d15
2 changed files with 25 additions and 75 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
spring-cloud-function-adapters/spring-cloud-function-adapter-aws/src/main/java/org/springframework/cloud/function/adapter/aws/AWSLambdaUtils.java
View File

@@ -121,6 +121,12 @@ final class AWSLambdaUtils {
messageBuilder = MessageBuilder.withPayload(body).copyHeaders(requestMap);
}
}
Object providedHeaders = requestMap.remove("headers");
if (providedHeaders != null && providedHeaders instanceof Map) {
messageBuilder.removeHeader("headers");
messageBuilder.copyHeaders((Map<String, Object>) providedHeaders);
}
}
else if (request instanceof Iterable) {
messageBuilder = MessageBuilder.withPayload(request);
@@ -131,6 +137,8 @@ final class AWSLambdaUtils {
if (awsContext != null) {
messageBuilder.setHeader("aws-context", awsContext);
}
logger.info("Incoming request headers: " + headers);
return messageBuilder.copyHeaders(headers).build();
}