Provide access to the captured Context from its Continuation (#9427)

Author: mcculls

dd-trace-core/src/main/java/datadog/trace/core/scopemanager/ScopeContinuation.java

@@ -122,6 +122,11 @@ public AgentSpan span() {
     return AgentSpan.fromContext(context);
   }
 
+  @Override
+  public Context context() {
+    return context;
+  }
+
   @Override
   public String toString() {
     return getClass().getSimpleName() + "@" + Integer.toHexString(hashCode()) + "->" + context;

internal-api/build.gradle.kts

@@ -84,6 +84,7 @@ val excludedClassesCoverage by extra(
     "datadog.trace.bootstrap.instrumentation.api.AgentPropagation",
     "datadog.trace.bootstrap.instrumentation.api.AgentPropagation.ContextVisitor",
     "datadog.trace.bootstrap.instrumentation.api.AgentScope",
+    "datadog.trace.bootstrap.instrumentation.api.AgentScope.Continuation",
     "datadog.trace.bootstrap.instrumentation.api.AgentSpan",
     "datadog.trace.bootstrap.instrumentation.api.AgentSpanContext",
     "datadog.trace.bootstrap.instrumentation.api.AgentTracer",

internal-api/src/main/java/datadog/trace/bootstrap/instrumentation/api/AgentScope.java

@@ -25,5 +25,10 @@ interface Continuation extends TraceScope.Continuation {
 
     /** Provide access to the captured span */
     AgentSpan span();
+
+    /** Provide access to the captured context */
+    default Context context() {
+      return span();
+    }
   }
 }