From 4c7dfa7a735852d455b378832b03388a5da812d3 Mon Sep 17 00:00:00 2001
From: Marcin Grzejszczak <marcin.grzejszczak@gmail.com>
Date: Wed, 13 Oct 2021 15:35:55 +0200
Subject: [PATCH] Changes for consumer

---
 .../cloud/function/context/catalog/FunctionAroundWrapper.java  | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/spring-cloud-function-context/src/main/java/org/springframework/cloud/function/context/catalog/FunctionAroundWrapper.java b/spring-cloud-function-context/src/main/java/org/springframework/cloud/function/context/catalog/FunctionAroundWrapper.java
index eff1e8c8d..0a95c605c 100644
--- a/spring-cloud-function-context/src/main/java/org/springframework/cloud/function/context/catalog/FunctionAroundWrapper.java
+++ b/spring-cloud-function-context/src/main/java/org/springframework/cloud/function/context/catalog/FunctionAroundWrapper.java
@@ -34,14 +34,13 @@ import org.springframework.messaging.Message;
  */
 public abstract class FunctionAroundWrapper implements BiFunction<Object, FunctionInvocationWrapper, Object> {
 
-	@SuppressWarnings("unchecked")
 	@Override
 	public final Object apply(Object input, FunctionInvocationWrapper targetFunction) {
 		boolean isSkipOutputConversion = targetFunction.isSkipOutputConversion();
 		targetFunction.setSkipOutputConversion(true);
 		Object result = null;
 
-		if (input instanceof Message || targetFunction.isOutputTypePublisher()) {
+		if (input instanceof Message || targetFunction.isOutputTypePublisher() || targetFunction.isInputTypePublisher()) {
 			return this.doApply(input, targetFunction);
 		}
 		else if (targetFunction.isSupplier() && !targetFunction.isOutputTypePublisher()) {