From 0cb5483d0ae4e93c22dbd176447b3cec9746c20b Mon Sep 17 00:00:00 2001
From: Oleg Zhurakousky <ozhurakousky@vmware.com>
Date: Fri, 15 Oct 2021 19:21:37 +0200
Subject: [PATCH] Relax type rules for outputMessageEnricher

---
 .../function/context/catalog/SimpleFunctionRegistry.java    | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/spring-cloud-function-context/src/main/java/org/springframework/cloud/function/context/catalog/SimpleFunctionRegistry.java b/spring-cloud-function-context/src/main/java/org/springframework/cloud/function/context/catalog/SimpleFunctionRegistry.java
index 7dcc73168..b63632f30 100644
--- a/spring-cloud-function-context/src/main/java/org/springframework/cloud/function/context/catalog/SimpleFunctionRegistry.java
+++ b/spring-cloud-function-context/src/main/java/org/springframework/cloud/function/context/catalog/SimpleFunctionRegistry.java
@@ -387,7 +387,7 @@ public class SimpleFunctionRegistry implements FunctionRegistry, FunctionInspect
 		 * It is not intended to remain here and will be removed as soon as particular elements
 		 * of stream will be refactored to address this.
 		 */
-		private Function<Object, Message> enhancer;
+		private Function<Object, Object> enhancer;
 
 		FunctionInvocationWrapper(FunctionInvocationWrapper function) {
 			this.skipOutputConversion = function.skipOutputConversion;
@@ -432,7 +432,7 @@ public class SimpleFunctionRegistry implements FunctionRegistry, FunctionInspect
 		 * It is not intended to remain here and will be removed as soon as particular elements
 		 * of stream will be refactored to address this.
 		 */
-		public Function<Object, Message> getEnhancer() {
+		public Function<Object, Object> getEnhancer() {
 			return this.enhancer;
 		}
 
@@ -443,7 +443,7 @@ public class SimpleFunctionRegistry implements FunctionRegistry, FunctionInspect
 		 * It is not intended to remain here and will be removed as soon as particular elements
 		 * of stream will be refactored to address this.
 		 */
-		public void setEnhancer(Function<Object, Message> enhancer) {
+		public void setEnhancer(Function<Object, Object> enhancer) {
 			this.enhancer = enhancer;
 		}