From d9a0add8b8fa823d175f6508e8be3bd240eb202d Mon Sep 17 00:00:00 2001
From: Oleg Zhurakousky <ozhurakousky@vmware.com>
Date: Mon, 11 Jan 2021 10:20:18 +0100
Subject: [PATCH] Fix getItemType() in SimpleFunctionRegistry to account for
 Message<?>

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

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 fb8efd33c..e12cbcae1 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,6 +387,9 @@ public class SimpleFunctionRegistry implements FunctionRegistry, FunctionInspect
 			if (FunctionTypeUtils.isPublisher(type) || FunctionTypeUtils.isMessage(type) || FunctionTypeUtils.isTypeCollection(type)) {
 				type = FunctionTypeUtils.getGenericType(type);
 			}
+			if (FunctionTypeUtils.isMessage(type)) {
+				type = FunctionTypeUtils.getGenericType(type);
+			}
 			return type;
 		}