Skip to content

Commit 7f36c74

Browse files
committed
feat: Add more helpers to bytecode bridge
1 parent 44188ec commit 7f36c74

File tree

2 files changed

+7
-1
lines changed
  • dd-java-agent

2 files changed

+7
-1
lines changed

dd-java-agent/agent-bootstrap/src/main/java/datadog/trace/bootstrap/instrumentation/api/Java8BytecodeBridge.java

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -29,4 +29,9 @@ public static Context getContextFrom(Object carrier) {
2929
public static Context detachContextFrom(Object carrier) {
3030
return Context.detachFrom(carrier);
3131
}
32+
33+
/** @see AgentSpan#fromContext(Context) */
34+
public static AgentSpan spanFromContext(Context context) {
35+
return AgentSpan.fromContext(context);
36+
}
3237
}

dd-java-agent/instrumentation/servlet/request-2/src/main/java/datadog/trace/instrumentation/servlet2/Servlet2Advice.java

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
package datadog.trace.instrumentation.servlet2;
22

3+
import static datadog.trace.bootstrap.instrumentation.api.Java8BytecodeBridge.spanFromContext;
34
import static datadog.trace.bootstrap.instrumentation.decorator.HttpServerDecorator.DD_SPAN_ATTRIBUTE;
45
import static datadog.trace.instrumentation.servlet2.Servlet2Decorator.DECORATE;
56

@@ -97,7 +98,7 @@ public static void stopSpan(
9798
if (scope == null) {
9899
return;
99100
}
100-
final AgentSpan span = AgentSpan.fromContext(scope.context());
101+
final AgentSpan span = spanFromContext(scope.context());
101102

102103
if (response instanceof HttpServletResponse) {
103104
DECORATE.onResponse(

0 commit comments

Comments
 (0)