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

Performance improvements related to SCST-2303

Browse Source
This commit is contained in:
Oleg Zhurakousky
2022-03-22 13:53:48 +01:00
parent 5ef56a90f0
commit 2ce45726ab
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
spring-cloud-function-context/src/main/java/org/springframework/cloud/function/context/catalog/SimpleFunctionRegistry.java
View File

@@ -1128,6 +1128,12 @@ public class SimpleFunctionRegistry implements FunctionRegistry {
if (!(output instanceof Publisher) && this.enhancer != null) { if (!(output instanceof Publisher) && this.enhancer != null) {
output = enhancer.apply(output); output = enhancer.apply(output);
} }
if (this.getTarget() instanceof PassThruFunction) { // scst-2303
Map<String, Object> headersMap = (Map<String, Object>) ReflectionUtils
.getField(SimpleFunctionRegistry.this.headersField, ((Message) output).getHeaders());
headersMap.put(MessageHeaders.CONTENT_TYPE, contentType[0]);
return messageConverter.toMessage(((Message) output).getPayload(), ((Message) output).getHeaders());
}
if (ObjectUtils.isEmpty(contentType) && !(output instanceof Publisher)) { if (ObjectUtils.isEmpty(contentType) && !(output instanceof Publisher)) {
return output; return output;