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

Update testing module to new agent api

This commit is contained in:
Trask Stalnaker 2019-10-19 15:30:40 -07:00
parent 41185662df
commit 299c396eab
1 changed files with 12 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
dd-java-agent/testing/src/main/groovy/datadog/trace/agent/test/utils/TraceUtils.groovy
View File

@ -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> T runUnderTrace(final String rootOperationName, final Callable<T> 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()
}
}