From 293ccd74254e668f18a0f97df376b68ce601d13a Mon Sep 17 00:00:00 2001
From: Oleg Zhurakousky <ozhurakousky@vmware.com>
Date: Thu, 17 Jun 2021 10:42:39 +0200
Subject: [PATCH] Add support for handling Supplier to FunctionAroundWrapper

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

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 dd5ce016e..d33fab87d 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
@@ -40,6 +40,9 @@ public abstract class FunctionAroundWrapper implements BiFunction<Object, Functi
 		if (input instanceof Message) {
 			return this.doApply((Message<byte[]>) input, targetFunction);
 		}
+		else if (targetFunction.isSupplier()) {
+			return this.doApply(null, targetFunction);
+		}
 		return targetFunction.apply(input);
 	}