Skip to content

Commit 4905729

Browse files
Remove custom scope manager support (#8164)
* chore(ot): Remove custom scope manager support * chore(core): Remove custom scope manager support
1 parent 54d4a6f commit 4905729

File tree

11 files changed

+58
-901
lines changed

11 files changed

+58
-901
lines changed

dd-trace-core/src/main/java/datadog/trace/core/CoreTracer.java

+10-23
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,6 @@
5151
import datadog.trace.bootstrap.instrumentation.api.AgentHistogram;
5252
import datadog.trace.bootstrap.instrumentation.api.AgentPropagation;
5353
import datadog.trace.bootstrap.instrumentation.api.AgentScope;
54-
import datadog.trace.bootstrap.instrumentation.api.AgentScopeManager;
5554
import datadog.trace.bootstrap.instrumentation.api.AgentSpan;
5655
import datadog.trace.bootstrap.instrumentation.api.AgentSpanLink;
5756
import datadog.trace.bootstrap.instrumentation.api.AgentTracer;
@@ -165,7 +164,7 @@ public static CoreTracerBuilder builder() {
165164
/** Sampler defines the sampling policy in order to reduce the number of traces for instance */
166165
final Sampler initialSampler;
167166
/** Scope manager is in charge of managing the scopes from which spans are created */
168-
final AgentScopeManager scopeManager;
167+
final ContinuableScopeManager scopeManager;
169168

170169
final MetricsAggregator metricsAggregator;
171170

@@ -295,7 +294,7 @@ public static class CoreTracerBuilder {
295294
private SingleSpanSampler singleSpanSampler;
296295
private HttpCodec.Injector injector;
297296
private HttpCodec.Extractor extractor;
298-
private AgentScopeManager scopeManager;
297+
private ContinuableScopeManager scopeManager;
299298
private Map<String, ?> localRootSpanTags;
300299
private Map<String, ?> defaultSpanTags;
301300
private Map<String, String> serviceNameMappings;
@@ -356,11 +355,6 @@ public CoreTracerBuilder extractor(HttpCodec.Extractor extractor) {
356355
return this;
357356
}
358357

359-
public CoreTracerBuilder scopeManager(AgentScopeManager scopeManager) {
360-
this.scopeManager = scopeManager;
361-
return this;
362-
}
363-
364358
public CoreTracerBuilder localRootSpanTags(Map<String, ?> localRootSpanTags) {
365359
this.localRootSpanTags = tryMakeImmutableMap(localRootSpanTags);
366360
return this;
@@ -497,7 +491,6 @@ public CoreTracer build() {
497491
singleSpanSampler,
498492
injector,
499493
extractor,
500-
scopeManager,
501494
localRootSpanTags,
502495
defaultSpanTags,
503496
serviceNameMappings,
@@ -529,7 +522,6 @@ private CoreTracer(
529522
final SingleSpanSampler singleSpanSampler,
530523
final HttpCodec.Injector injector,
531524
final HttpCodec.Extractor extractor,
532-
final AgentScopeManager scopeManager,
533525
final Map<String, ?> localRootSpanTags,
534526
final Map<String, ?> defaultSpanTags,
535527
final Map<String, String> serviceNameMappings,
@@ -632,16 +624,13 @@ private CoreTracer(
632624
: Monitoring.DISABLED;
633625

634626
traceWriteTimer = performanceMonitoring.newThreadLocalTimer("trace.write");
635-
if (scopeManager == null) {
636-
this.scopeManager =
637-
new ContinuableScopeManager(
638-
config.getScopeDepthLimit(),
639-
config.isScopeStrictMode(),
640-
profilingContextIntegration,
641-
healthMetrics);
642-
} else {
643-
this.scopeManager = scopeManager;
644-
}
627+
628+
scopeManager =
629+
new ContinuableScopeManager(
630+
config.getScopeDepthLimit(),
631+
config.isScopeStrictMode(),
632+
profilingContextIntegration,
633+
healthMetrics);
645634

646635
externalAgentLauncher = new ExternalAgentLauncher(config);
647636

@@ -1108,9 +1097,7 @@ public DataStreamsCheckpointer getDataStreamsCheckpointer() {
11081097

11091098
@Override
11101099
public void addScopeListener(final ScopeListener listener) {
1111-
if (scopeManager instanceof ContinuableScopeManager) {
1112-
((ContinuableScopeManager) scopeManager).addScopeListener(listener);
1113-
}
1100+
this.scopeManager.addScopeListener(listener);
11141101
}
11151102

11161103
@Override

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

+2-9
Original file line numberDiff line numberDiff line change
@@ -11,13 +11,13 @@
1111
import datadog.trace.api.scopemanager.ExtendedScopeListener;
1212
import datadog.trace.api.scopemanager.ScopeListener;
1313
import datadog.trace.bootstrap.instrumentation.api.AgentScope;
14-
import datadog.trace.bootstrap.instrumentation.api.AgentScopeManager;
1514
import datadog.trace.bootstrap.instrumentation.api.AgentSpan;
1615
import datadog.trace.bootstrap.instrumentation.api.AgentTracer;
1716
import datadog.trace.bootstrap.instrumentation.api.ProfilerContext;
1817
import datadog.trace.bootstrap.instrumentation.api.ProfilingContextIntegration;
1918
import datadog.trace.bootstrap.instrumentation.api.ScopeSource;
2019
import datadog.trace.bootstrap.instrumentation.api.ScopeState;
20+
import datadog.trace.bootstrap.instrumentation.api.ScopeStateAware;
2121
import datadog.trace.core.monitor.HealthMetrics;
2222
import datadog.trace.relocate.api.RatelimitedLogger;
2323
import datadog.trace.util.AgentTaskScheduler;
@@ -37,7 +37,7 @@
3737
* from being reported even if all related spans are finished. It also delegates to other
3838
* ScopeInterceptors to provide additional functionality.
3939
*/
40-
public final class ContinuableScopeManager implements AgentScopeManager {
40+
public final class ContinuableScopeManager implements ScopeStateAware {
4141
static final Logger log = LoggerFactory.getLogger(ContinuableScopeManager.class);
4242
static final RatelimitedLogger ratelimitedLog = new RatelimitedLogger(log, 1, MINUTES);
4343
static final long iterationKeepAlive =
@@ -81,18 +81,15 @@ public ContinuableScopeManager(
8181
this.profilingContextIntegration = profilingContextIntegration;
8282
}
8383

84-
@Override
8584
public AgentScope activate(final AgentSpan span, final ScopeSource source) {
8685
return activate(span, source.id(), false, /* ignored */ false);
8786
}
8887

89-
@Override
9088
public AgentScope activate(
9189
final AgentSpan span, final ScopeSource source, boolean isAsyncPropagating) {
9290
return activate(span, source.id(), true, isAsyncPropagating);
9391
}
9492

95-
@Override
9693
public AgentScope.Continuation captureSpan(final AgentSpan span) {
9794
AbstractContinuation continuation =
9895
new SingleContinuation(this, span, ScopeSource.INSTRUMENTATION.id());
@@ -170,7 +167,6 @@ ContinuableScope continueSpan(
170167
return scope;
171168
}
172169

173-
@Override
174170
public void closePrevious(final boolean finishSpan) {
175171
ScopeStack scopeStack = scopeStack();
176172

@@ -189,7 +185,6 @@ public void closePrevious(final boolean finishSpan) {
189185
}
190186
}
191187

192-
@Override
193188
public AgentScope activateNext(final AgentSpan span) {
194189
ScopeStack scopeStack = scopeStack();
195190

@@ -220,12 +215,10 @@ public AgentScope activateNext(final AgentSpan span) {
220215
return scope;
221216
}
222217

223-
@Override
224218
public AgentScope active() {
225219
return scopeStack().active();
226220
}
227221

228-
@Override
229222
public AgentSpan activeSpan() {
230223
final ContinuableScope active = scopeStack().active();
231224
return active == null ? null : active.span;

0 commit comments

Comments
 (0)