From 033ab385796db0d938a15130f2a663c4819659c5 Mon Sep 17 00:00:00 2001 From: Dave Syer Date: Mon, 17 Sep 2018 14:47:59 +0100 Subject: [PATCH] Fix message processor in isolated function --- .../springframework/cloud/function/web/RequestProcessor.java | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/spring-cloud-function-web/src/main/java/org/springframework/cloud/function/web/RequestProcessor.java b/spring-cloud-function-web/src/main/java/org/springframework/cloud/function/web/RequestProcessor.java index 36998076a..dbe7d5623 100644 --- a/spring-cloud-function-web/src/main/java/org/springframework/cloud/function/web/RequestProcessor.java +++ b/spring-cloud-function-web/src/main/java/org/springframework/cloud/function/web/RequestProcessor.java @@ -195,8 +195,9 @@ public class RequestProcessor { BodyBuilder builder = ResponseEntity.ok(); if (inspector.isMessage(handler)) { result = Flux.from(result) - .doOnNext(value -> addHeaders(builder, (Message) value)) - .map(message -> MessageUtils.unpack(handler, message).getPayload()); + .map(message -> MessageUtils.unpack(handler, message)) + .doOnNext(value -> addHeaders(builder, value)) + .map(message -> message.getPayload()); } else { builder.headers(HeaderUtils.sanitize(request.headers()));