diff --git a/dd-java-agent/testing/src/main/groovy/datadog/trace/agent/test/utils/TraceUtils.groovy b/dd-java-agent/testing/src/main/groovy/datadog/trace/agent/test/utils/TraceUtils.groovy index 2c4d24800f..e7b78e07f8 100644 --- a/dd-java-agent/testing/src/main/groovy/datadog/trace/agent/test/utils/TraceUtils.groovy +++ b/dd-java-agent/testing/src/main/groovy/datadog/trace/agent/test/utils/TraceUtils.groovy @@ -3,13 +3,15 @@ package datadog.trace.agent.test.utils import datadog.opentracing.DDSpan import datadog.trace.agent.decorator.BaseDecorator import datadog.trace.agent.test.asserts.TraceAssert -import datadog.trace.context.TraceScope -import io.opentracing.Scope -import io.opentracing.util.GlobalTracer +import datadog.trace.instrumentation.api.AgentScope +import datadog.trace.instrumentation.api.AgentSpan import lombok.SneakyThrows import java.util.concurrent.Callable +import static datadog.trace.instrumentation.api.AgentTracer.activateSpan +import static datadog.trace.instrumentation.api.AgentTracer.startSpan + class TraceUtils { private static final BaseDecorator DECORATOR = new BaseDecorator() { @@ -28,17 +30,19 @@ class TraceUtils { @SneakyThrows static T runUnderTrace(final String rootOperationName, final Callable r) { - final Scope scope = GlobalTracer.get().buildSpan(rootOperationName).startActive(true) - DECORATOR.afterStart(scope) - ((TraceScope) scope).setAsyncPropagation(true) + final AgentSpan span = startSpan(rootOperationName) + DECORATOR.afterStart(span) + + AgentScope scope = activateSpan(span, true) + scope.setAsyncPropagation(true) try { return r.call() } catch (final Exception e) { - DECORATOR.onError(scope, e) + DECORATOR.onError(span, e) throw e } finally { - DECORATOR.beforeFinish(scope) + DECORATOR.beforeFinish(span) scope.close() } }