diff --git a/dd-java-agent/instrumentation/reactor-netty-1.0/src/main/java/datadog/trace/instrumentation/reactor/netty/CaptureConnectSpan.java b/dd-java-agent/instrumentation/reactor-netty-1.0/src/main/java/datadog/trace/instrumentation/reactor/netty/CaptureConnectSpan.java index 351a37affd3..68cb2c55208 100644 --- a/dd-java-agent/instrumentation/reactor-netty-1.0/src/main/java/datadog/trace/instrumentation/reactor/netty/CaptureConnectSpan.java +++ b/dd-java-agent/instrumentation/reactor-netty-1.0/src/main/java/datadog/trace/instrumentation/reactor/netty/CaptureConnectSpan.java @@ -2,6 +2,7 @@ import static datadog.trace.bootstrap.instrumentation.api.AgentTracer.activeSpan; +import datadog.context.Context; import datadog.trace.bootstrap.instrumentation.api.AgentSpan; import java.util.function.Function; import reactor.core.publisher.Mono; @@ -11,6 +12,7 @@ public class CaptureConnectSpan implements Function, Mono> { static final String CONNECT_SPAN = "datadog.connect.span"; + static final String CONNECT_CONTEXT = "datadog.connect.context"; @Override public Mono apply(Mono mono) { @@ -18,7 +20,7 @@ public Mono apply(Mono mono) { context -> { final AgentSpan span = activeSpan(); if (null != span) { - return context.put(CONNECT_SPAN, span); + return context.put(CONNECT_SPAN, span).put(CONNECT_CONTEXT, Context.current()); } else { return context; } diff --git a/dd-java-agent/instrumentation/reactor-netty-1.0/src/main/java/datadog/trace/instrumentation/reactor/netty/TransferConnectSpan.java b/dd-java-agent/instrumentation/reactor-netty-1.0/src/main/java/datadog/trace/instrumentation/reactor/netty/TransferConnectSpan.java index 5d07e5c3c26..b0a4926e865 100644 --- a/dd-java-agent/instrumentation/reactor-netty-1.0/src/main/java/datadog/trace/instrumentation/reactor/netty/TransferConnectSpan.java +++ b/dd-java-agent/instrumentation/reactor-netty-1.0/src/main/java/datadog/trace/instrumentation/reactor/netty/TransferConnectSpan.java @@ -2,8 +2,10 @@ import static datadog.trace.bootstrap.instrumentation.api.AgentTracer.captureSpan; import static datadog.trace.instrumentation.netty41.AttributeKeys.CONNECT_PARENT_CONTINUATION_ATTRIBUTE_KEY; +import static datadog.trace.instrumentation.reactor.netty.CaptureConnectSpan.CONNECT_CONTEXT; import static datadog.trace.instrumentation.reactor.netty.CaptureConnectSpan.CONNECT_SPAN; +import datadog.context.Context; import datadog.trace.bootstrap.instrumentation.api.AgentScope.Continuation; import datadog.trace.bootstrap.instrumentation.api.AgentSpan; import java.util.function.BiConsumer; @@ -14,7 +16,13 @@ public class TransferConnectSpan implements BiConsumer