open-telemetry
/
opentelemetry-java-instrumentation
mirror of https://github.com/open-telemetry/opentelemetry-java-instrumentation.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Make spring controller spans to handle async

This commit is contained in:
Luca Abbati 2019-05-24 17:42:06 +02:00
parent 200a9063fa
commit 25e06b4965
No known key found for this signature in database
GPG Key ID: C901DDA2FFE14529
2 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
dd-java-agent/instrumentation/spring-web/src/main/java/datadog/trace/instrumentation/springweb/HandlerAdapterInstrumentation.java
View File

@ -14,6 +14,7 @@ import static net.bytebuddy.matcher.ElementMatchers.takesArguments;
import com.google.auto.service.AutoService; import com.google.auto.service.AutoService;
import datadog.trace.agent.tooling.Instrumenter; import datadog.trace.agent.tooling.Instrumenter;
import datadog.trace.context.TraceScope;
import io.opentracing.Scope; import io.opentracing.Scope;
import io.opentracing.util.GlobalTracer; import io.opentracing.util.GlobalTracer;
import java.lang.reflect.Method; import java.lang.reflect.Method;
@ -107,6 +108,9 @@ public final class HandlerAdapterInstrumentation extends Instrumenter.Default {
final String operationName = DECORATE.spanNameForClass(clazz) + "." + methodName; final String operationName = DECORATE.spanNameForClass(clazz) + "." + methodName;
final Scope scope = GlobalTracer.get().buildSpan(operationName).startActive(true); final Scope scope = GlobalTracer.get().buildSpan(operationName).startActive(true);
if (scope instanceof TraceScope) {
((TraceScope) scope).setAsyncPropagation(true);
}
DECORATE.afterStart(scope); DECORATE.afterStart(scope);
return scope; return scope;
} }

2
dd-trace-api/src/main/java/datadog/trace/context/TraceScope.java
View File

@ -2,7 +2,7 @@ package datadog.trace.context;
import java.io.Closeable; import java.io.Closeable;
/** An object when can propagate a datadog trace across multiple threads. */ /** An object that can propagate a datadog trace across multiple threads. */
public interface TraceScope extends Closeable { public interface TraceScope extends Closeable {
/** /**
* Prevent the trace attached to this TraceScope from reporting until the returned Continuation * Prevent the trace attached to this TraceScope from reporting until the returned Continuation