From 195d853714894a9fb3f98e7662364f754a42a348 Mon Sep 17 00:00:00 2001
From: Tim Ysewyn <Tim.Ysewyn@me.com>
Date: Wed, 15 Dec 2021 11:00:46 +0100
Subject: [PATCH] Moved return statement out of log.isDebugEnabled() condition

---
 .../instrument/messaging/TraceFunctionAroundWrapper.java       | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/spring-cloud-sleuth-instrumentation/src/main/java/org/springframework/cloud/sleuth/instrument/messaging/TraceFunctionAroundWrapper.java b/spring-cloud-sleuth-instrumentation/src/main/java/org/springframework/cloud/sleuth/instrument/messaging/TraceFunctionAroundWrapper.java
index 7ef24829c..63ea201ac 100644
--- a/spring-cloud-sleuth-instrumentation/src/main/java/org/springframework/cloud/sleuth/instrument/messaging/TraceFunctionAroundWrapper.java
+++ b/spring-cloud-sleuth-instrumentation/src/main/java/org/springframework/cloud/sleuth/instrument/messaging/TraceFunctionAroundWrapper.java
@@ -47,6 +47,7 @@ import org.springframework.messaging.support.MessageHeaderAccessor;
  *
  * @author Marcin Grzejszczak
  * @author Oleg Zhurakousky
+ * @author Tim Ysewyn
  * @since 3.0.0
  */
 public class TraceFunctionAroundWrapper extends FunctionAroundWrapper
@@ -126,8 +127,8 @@ public class TraceFunctionAroundWrapper extends FunctionAroundWrapper
 			if (log.isDebugEnabled()) {
 				log.debug("Target function [" + targetFunction.getFunctionDefinition() + "] has raw input type ["
 						+ itemType + "] and should be [" + Message.class + "]. Will not wrap it.");
-				return targetFunction.apply(messageStream);
 			}
+			return targetFunction.apply(messageStream);
 		}
 		Publisher<Message> messagePublisher = messageStream;
 		if (FunctionTypeUtils.isMono(targetFunction.getInputType())) {