From f1d72227c319268156256cbf2a745a663ba24b9c Mon Sep 17 00:00:00 2001 From: Oleg Zhurakousky Date: Wed, 28 Jul 2021 12:39:42 +0200 Subject: [PATCH] Fix previous commit --- .../messaging/TraceFunctionAroundWrapper.java | 18 ++++++++---------- 1 file changed, 8 insertions(+), 10 deletions(-) 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 b31cc3ad7..b2a0cbdc7 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 @@ -87,20 +87,18 @@ public class TraceFunctionAroundWrapper extends FunctionAroundWrapper protected Object doApply(Message message, SimpleFunctionRegistry.FunctionInvocationWrapper targetFunction) { /* - * This code is temporary to disable conditions for which this interceptor is not ready. - * For example, - * - it does not handle properly input or output of type Publisher - * - it wraps output in Message when function returns a Collection which it should not do. + * This code is temporary to disable conditions for which this interceptor is not + * ready. For example, - it does not handle properly input or output of type + * Publisher - it wraps output in Message when function returns a + * Collection which it should not do. * */ - if ( (!FunctionTypeUtils.isCollectionOfMessage(targetFunction.getOutputType()) && !targetFunction.isOutputTypePublisher()) - || (targetFunction.isSupplier() && !targetFunction.isOutputTypePublisher()) - ) { - return targetFunction.apply(message); // no instrumentation + if ((FunctionTypeUtils.isCollectionOfMessage(targetFunction.getOutputType()) + || targetFunction.isOutputTypePublisher()) + || (targetFunction.isSupplier() && targetFunction.isOutputTypePublisher())) { + return targetFunction.apply(message); // no instrumentation } - - MessageAndSpans invocationMessage = null; Span span; if (message == null && targetFunction.isSupplier()) { // Supplier