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

GH-654 Fix how RoutingFunction is treated by RSocketListenerFunction

Browse Source
This commit is contained in:
Oleg Zhurakousky
2021-02-23 16:58:37 +01:00
parent 29115aeb64
commit 26c11b4463
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
spring-cloud-function-rsocket/src/main/java/org/springframework/cloud/function/rsocket/RSocketListenerFunction.java
View File

@@ -69,7 +69,12 @@ class RSocketListenerFunction implements Function<Message<Flux<byte[]>>, Publish
@SuppressWarnings({ "unchecked", "rawtypes" })
private Mono<Void> handle(Message<Flux<byte[]>> messageToProcess) {
if (this.targetFunction.isConsumer()) {
if (this.targetFunction.isRoutingFunction()) {
Flux<?> dataFlux = messageToProcess.getPayload()
.map((payload) -> MessageBuilder.createMessage(payload, messageToProcess.getHeaders()));
return dataFlux.doOnNext(this.targetFunction).then();
}
else if (this.targetFunction.isConsumer()) {
Flux<?> dataFlux =
messageToProcess.getPayload()
.map((payload) -> MessageBuilder.createMessage(payload, messageToProcess.getHeaders()));