From 05416dbc30ecab8a2bda237923ef205c9a3bda06 Mon Sep 17 00:00:00 2001
From: Oleg Zhurakousky <ozhurakousky@vmware.com>
Date: Fri, 31 May 2024 15:53:05 +0200
Subject: [PATCH] GH-STREAM 2704 Fix Consumer CCE on Kotlin Consumer

Resolves #2704
---
 .../config/KotlinLambdaToFunctionAutoConfiguration.java     | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/spring-cloud-function-context/src/main/java/org/springframework/cloud/function/context/config/KotlinLambdaToFunctionAutoConfiguration.java b/spring-cloud-function-context/src/main/java/org/springframework/cloud/function/context/config/KotlinLambdaToFunctionAutoConfiguration.java
index 034a6b883..5c8828e87 100644
--- a/spring-cloud-function-context/src/main/java/org/springframework/cloud/function/context/config/KotlinLambdaToFunctionAutoConfiguration.java
+++ b/spring-cloud-function-context/src/main/java/org/springframework/cloud/function/context/config/KotlinLambdaToFunctionAutoConfiguration.java
@@ -123,7 +123,11 @@ public class KotlinLambdaToFunctionAutoConfiguration {
 			if (this.kotlinLambdaTarget instanceof Function1) {
 				return ((Function1) this.kotlinLambdaTarget).invoke(arg0);
 			}
-			return ((Function) this.kotlinLambdaTarget).apply(arg0);
+			else if (this.kotlinLambdaTarget instanceof Function) {
+				return ((Function) this.kotlinLambdaTarget).apply(arg0);
+			}
+			((Consumer) this.kotlinLambdaTarget).accept(arg0);
+			return null;
 		}
 
 		@Override