Ensure tracer and scheduler are clean after test execution

amarziali · amarziali · commit 0c57e15f1baa · 2025-09-11T10:16:24.000+02:00
diff --git a/communication/src/main/java/datadog/communication/ddagent/SharedCommunicationObjects.java b/communication/src/main/java/datadog/communication/ddagent/SharedCommunicationObjects.java
@@ -159,7 +159,7 @@ public DDAgentFeaturesDiscovery featuresDiscovery(Config config) {
               ret.discover(); // safe to run on same thread
             } else {
               // avoid performing blocking I/O operation on application thread
-              AgentTaskScheduler.INSTANCE.execute(ret::discoverIfOutdated);
+              AgentTaskScheduler.getInstance().execute(ret::discoverIfOutdated);
             }
           }
           featuresDiscovery = ret;
diff --git a/communication/src/main/java/datadog/communication/monitor/DDAgentStatsDConnection.java b/communication/src/main/java/datadog/communication/monitor/DDAgentStatsDConnection.java
@@ -87,8 +87,8 @@ private void scheduleConnect() {
         log.debug(
             "Scheduling StatsD connection in {} seconds - {}", remainingDelay, statsDAddress());
       }
-      AgentTaskScheduler.INSTANCE.scheduleWithJitter(
-          ConnectTask.INSTANCE, this, remainingDelay, SECONDS);
+      AgentTaskScheduler.getInstance()
+          .scheduleWithJitter(ConnectTask.INSTANCE, this, remainingDelay, SECONDS);
     } else {
       doConnect();
     }
@@ -168,8 +168,8 @@ private void doConnect() {
               log.debug(
                   "Scheduling StatsD connection in {} seconds - {}", RETRY_DELAY, statsDAddress());
             }
-            AgentTaskScheduler.INSTANCE.scheduleWithJitter(
-                ConnectTask.INSTANCE, this, RETRY_DELAY, SECONDS);
+            AgentTaskScheduler.getInstance()
+                .scheduleWithJitter(ConnectTask.INSTANCE, this, RETRY_DELAY, SECONDS);
           } else {
             log.debug("Max retries have been reached. Will not attempt again.");
           }
diff --git a/dd-java-agent/agent-bootstrap/src/main/java/datadog/trace/bootstrap/Agent.java b/dd-java-agent/agent-bootstrap/src/main/java/datadog/trace/bootstrap/Agent.java
@@ -513,7 +513,7 @@ public static void startDatadogTracer(InitializationTelemetry initTelemetry) thr
 
   private static void registerLogManagerCallback(final ClassLoadCallBack callback) {
     // one minute fail-safe in case the class was unintentionally loaded during premain
-    AgentTaskScheduler.INSTANCE.schedule(callback, 1, TimeUnit.MINUTES);
+    AgentTaskScheduler.getInstance().schedule(callback, 1, TimeUnit.MINUTES);
     try {
       final Class<?> agentInstallerClass = AGENT_CLASSLOADER.loadClass(AGENT_INSTALLER_CLASS_NAME);
       final Method registerCallbackMethod =
@@ -526,7 +526,7 @@ private static void registerLogManagerCallback(final ClassLoadCallBack callback)
 
   private static void registerMBeanServerBuilderCallback(final ClassLoadCallBack callback) {
     // one minute fail-safe in case the class was unintentionally loaded during premain
-    AgentTaskScheduler.INSTANCE.schedule(callback, 1, TimeUnit.MINUTES);
+    AgentTaskScheduler.getInstance().schedule(callback, 1, TimeUnit.MINUTES);
     try {
       final Class<?> agentInstallerClass = AGENT_CLASSLOADER.loadClass(AGENT_INSTALLER_CLASS_NAME);
       final Method registerCallbackMethod =
@@ -785,7 +785,7 @@ private static void startCrashTracking() {
       if (forceEarlyStart) {
         initializeCrashTrackingDefault();
       } else {
-        AgentTaskScheduler.INSTANCE.execute(Agent::initializeCrashTrackingDefault);
+        AgentTaskScheduler.getInstance().execute(Agent::initializeCrashTrackingDefault);
       }
     } else {
       // for Java 8 we are relying on JMX to give us the process PID
@@ -796,8 +796,8 @@ private static void startCrashTracking() {
 
   private static void scheduleJmxStart(final int jmxStartDelay) {
     if (jmxStartDelay > 0) {
-      AgentTaskScheduler.INSTANCE.scheduleWithJitter(
-          new JmxStartTask(), jmxStartDelay, TimeUnit.SECONDS);
+      AgentTaskScheduler.getInstance()
+          .scheduleWithJitter(new JmxStartTask(), jmxStartDelay, TimeUnit.SECONDS);
     } else {
       startJmx();
     }
@@ -843,8 +843,8 @@ When getJmxStartDelay() is set to 0 we will attempt to initialize the JMX subsys
           */
           if (getJmxStartDelay() == 0) {
             log.debug("Waiting for profiler initialization");
-            AgentTaskScheduler.INSTANCE.scheduleWithJitter(
-                PROFILER_INIT_AFTER_JMX, 500, TimeUnit.MILLISECONDS);
+            AgentTaskScheduler.getInstance()
+                .scheduleWithJitter(PROFILER_INIT_AFTER_JMX, 500, TimeUnit.MILLISECONDS);
           } else {
             log.debug("Initializing profiler");
             PROFILER_INIT_AFTER_JMX.run();
diff --git a/dd-java-agent/agent-builder/src/main/java/datadog/trace/agent/tooling/AgentInstaller.java b/dd-java-agent/agent-builder/src/main/java/datadog/trace/agent/tooling/AgentInstaller.java
@@ -83,11 +83,12 @@ public static void installBytebuddyAgent(final Instrumentation inst) {
       }
       int poolCleaningInterval = InstrumenterConfig.get().getResolverResetInterval();
       if (poolCleaningInterval > 0) {
-        AgentTaskScheduler.INSTANCE.scheduleAtFixedRate(
-            SharedTypePools::clear,
-            poolCleaningInterval,
-            Math.max(poolCleaningInterval, 10),
-            TimeUnit.SECONDS);
+        AgentTaskScheduler.getInstance()
+            .scheduleAtFixedRate(
+                SharedTypePools::clear,
+                poolCleaningInterval,
+                Math.max(poolCleaningInterval, 10),
+                TimeUnit.SECONDS);
       }
     } else if (DEBUG) {
       log.debug("No target systems enabled, skipping instrumentation.");
diff --git a/dd-java-agent/agent-debugger/src/main/java/com/datadog/debugger/agent/SourceFileTrackingTransformer.java b/dd-java-agent/agent-debugger/src/main/java/com/datadog/debugger/agent/SourceFileTrackingTransformer.java
@@ -29,7 +29,7 @@ public class SourceFileTrackingTransformer implements ClassFileTransformer {
 
   private final ClassesToRetransformFinder finder;
   private final Queue<SourceFileItem> queue = new ConcurrentLinkedQueue<>();
-  private final AgentTaskScheduler scheduler = AgentTaskScheduler.INSTANCE;
+  private final AgentTaskScheduler scheduler = AgentTaskScheduler.getInstance();
   private final AtomicInteger queueSize = new AtomicInteger(0);
   private AgentTaskScheduler.Scheduled<Runnable> scheduled;
   // this field MUST only be used in flush() calling thread
diff --git a/dd-java-agent/agent-debugger/src/main/java/com/datadog/debugger/codeorigin/DefaultCodeOriginRecorder.java b/dd-java-agent/agent-debugger/src/main/java/com/datadog/debugger/codeorigin/DefaultCodeOriginRecorder.java
@@ -48,7 +48,7 @@ public class DefaultCodeOriginRecorder implements CodeOriginRecorder {
   private AgentTaskScheduler scheduler;
 
   public DefaultCodeOriginRecorder(Config config, ConfigurationUpdater configurationUpdater) {
-    this(config, configurationUpdater, AgentTaskScheduler.INSTANCE);
+    this(config, configurationUpdater, AgentTaskScheduler.getInstance());
   }
 
   public DefaultCodeOriginRecorder(
diff --git a/dd-java-agent/agent-debugger/src/main/java/com/datadog/debugger/exception/AbstractExceptionDebugger.java b/dd-java-agent/agent-debugger/src/main/java/com/datadog/debugger/exception/AbstractExceptionDebugger.java
@@ -94,7 +94,8 @@ public void handleException(Throwable t, AgentSpan span) {
           if (!applyConfigAsync) {
             applyExceptionConfiguration(fingerprint);
           } else {
-            AgentTaskScheduler.INSTANCE.execute(() -> applyExceptionConfiguration(fingerprint));
+            AgentTaskScheduler.getInstance()
+                .execute(() -> applyExceptionConfiguration(fingerprint));
           }
           break;
         } else {
diff --git a/dd-java-agent/agent-debugger/src/main/java/com/datadog/debugger/sink/DebuggerSink.java b/dd-java-agent/agent-debugger/src/main/java/com/datadog/debugger/sink/DebuggerSink.java
@@ -36,7 +36,7 @@ public class DebuggerSink {
   private final String tags;
   private final AtomicLong highRateDropped = new AtomicLong();
   private final int uploadFlushInterval;
-  private final AgentTaskScheduler lowRateScheduler = AgentTaskScheduler.INSTANCE;
+  private final AgentTaskScheduler lowRateScheduler = AgentTaskScheduler.getInstance();
   private volatile AgentTaskScheduler.Scheduled<DebuggerSink> lowRateScheduled;
   private volatile AgentTaskScheduler.Scheduled<DebuggerSink> flushIntervalScheduled;
   private volatile long currentLowRateFlushInterval = LOW_RATE_INITIAL_FLUSH_INTERVAL;
diff --git a/dd-java-agent/agent-debugger/src/main/java/com/datadog/debugger/symbol/SymDBEnablement.java b/dd-java-agent/agent-debugger/src/main/java/com/datadog/debugger/symbol/SymDBEnablement.java
@@ -58,7 +58,7 @@ public void accept(ConfigKey configKey, byte[] content, PollingRateHinter pollin
       SymDbRemoteConfigRecord symDb = deserializeSymDb(content);
       if (symDb.isUploadSymbols()) {
         // can be long, make it async
-        AgentTaskScheduler.INSTANCE.execute(this::startSymbolExtraction);
+        AgentTaskScheduler.getInstance().execute(this::startSymbolExtraction);
       } else {
         stopSymbolExtraction();
       }
diff --git a/dd-java-agent/agent-debugger/src/main/java/com/datadog/debugger/symbol/SymbolAggregator.java b/dd-java-agent/agent-debugger/src/main/java/com/datadog/debugger/symbol/SymbolAggregator.java
@@ -64,11 +64,11 @@ public SymbolAggregator(
 
   public void start() {
     flushRemainingScopeScheduled =
-        AgentTaskScheduler.INSTANCE.scheduleAtFixedRate(
-            this::flushRemainingScopes, this, 0, 1, TimeUnit.SECONDS);
+        AgentTaskScheduler.getInstance()
+            .scheduleAtFixedRate(this::flushRemainingScopes, this, 0, 1, TimeUnit.SECONDS);
     scanJarsScheduled =
-        AgentTaskScheduler.INSTANCE.scheduleAtFixedRate(
-            this::scanQueuedJars, this, 0, 1, TimeUnit.SECONDS);
+        AgentTaskScheduler.getInstance()
+            .scheduleAtFixedRate(this::scanQueuedJars, this, 0, 1, TimeUnit.SECONDS);
   }
 
   public void stop() {
diff --git a/dd-java-agent/agent-iast/src/main/java/com/datadog/iast/IastSystem.java b/dd-java-agent/agent-iast/src/main/java/com/datadog/iast/IastSystem.java
@@ -102,7 +102,7 @@ public static void start(
     if (VERBOSITY != Verbosity.OFF) {
       IastMetricCollector.register(new IastMetricCollector());
     }
-    final Reporter reporter = new Reporter(config, AgentTaskScheduler.INSTANCE);
+    final Reporter reporter = new Reporter(config, AgentTaskScheduler.getInstance());
     final boolean globalContext = config.getIastContextMode() == GLOBAL;
     final IastContext.Provider contextProvider = contextProvider(iast, globalContext);
     if (overheadController == null) {
@@ -111,7 +111,7 @@ public static void start(
               globalContext ? UNLIMITED : config.getIastRequestSampling(),
               config.getIastMaxConcurrentRequests(),
               globalContext,
-              AgentTaskScheduler.INSTANCE);
+              AgentTaskScheduler.getInstance());
     }
     IastContext.Provider.register(contextProvider);
     final Dependencies dependencies =
diff --git a/dd-java-agent/agent-iast/src/main/java/com/datadog/iast/taint/TaintedMap.java b/dd-java-agent/agent-iast/src/main/java/com/datadog/iast/taint/TaintedMap.java
@@ -76,7 +76,11 @@ static TaintedMap build(final int capacity) {
   static TaintedMap buildWithPurge(final int capacity, int maxAge, TimeUnit maxAgeUnit) {
     final TaintedMapImpl map =
         new TaintedMapImpl(
-            capacity, DEFAULT_MAX_BUCKET_SIZE, maxAge, maxAgeUnit, AgentTaskScheduler.INSTANCE);
+            capacity,
+            DEFAULT_MAX_BUCKET_SIZE,
+            maxAge,
+            maxAgeUnit,
+            AgentTaskScheduler.getInstance());
     return IastSystem.DEBUG ? new Debug(map) : map;
   }
 
@@ -132,7 +136,7 @@ class TaintedMapImpl implements TaintedMap, Runnable {
      */
     TaintedMapImpl(
         final int capacity, final int maxBucketSize, final int maxAge, final TimeUnit maxAgeUnit) {
-      this(capacity, maxBucketSize, maxAge, maxAgeUnit, AgentTaskScheduler.INSTANCE);
+      this(capacity, maxBucketSize, maxAge, maxAgeUnit, AgentTaskScheduler.getInstance());
     }
 
     TaintedMapImpl(
@@ -348,7 +352,7 @@ public void put(@Nonnull final TaintedObject entry) {
       delegate.put(entry);
       final long putOps = puts.updateAndGet(current -> current == Long.MAX_VALUE ? 0 : current + 1);
       if (putOps % COMPUTE_STATISTICS_INTERVAL == 0 && LOGGER.isDebugEnabled()) {
-        AgentTaskScheduler.INSTANCE.execute(this::computeStatistics);
+        AgentTaskScheduler.getInstance().execute(this::computeStatistics);
       }
     }
 
diff --git a/dd-java-agent/agent-tooling/src/main/java/datadog/trace/agent/tooling/WeakMaps.java b/dd-java-agent/agent-tooling/src/main/java/datadog/trace/agent/tooling/WeakMaps.java
@@ -14,12 +14,13 @@ public class WeakMaps {
   public static <K, V> WeakMap<K, V> newWeakMap() {
     final WeakConcurrentMap<K, V> map = new WeakConcurrentMap<>(false, true);
     if (!Platform.isNativeImageBuilder()) {
-      AgentTaskScheduler.INSTANCE.weakScheduleAtFixedRate(
-          MapCleaningTask.INSTANCE,
-          map,
-          CLEAN_FREQUENCY_SECONDS,
-          CLEAN_FREQUENCY_SECONDS,
-          TimeUnit.SECONDS);
+      AgentTaskScheduler.getInstance()
+          .weakScheduleAtFixedRate(
+              MapCleaningTask.INSTANCE,
+              map,
+              CLEAN_FREQUENCY_SECONDS,
+              CLEAN_FREQUENCY_SECONDS,
+              TimeUnit.SECONDS);
     }
     return new Adapter<>(map);
   }
diff --git a/dd-trace-core/src/main/java/datadog/trace/common/metrics/ConflatingMetricsAggregator.java b/dd-trace-core/src/main/java/datadog/trace/common/metrics/ConflatingMetricsAggregator.java
@@ -203,12 +203,13 @@ public void start() {
     sink.register(this);
     thread.start();
     cancellation =
-        AgentTaskScheduler.INSTANCE.scheduleAtFixedRate(
-            new ReportTask(),
-            this,
-            reportingInterval,
-            reportingInterval,
-            reportingIntervalTimeUnit);
+        AgentTaskScheduler.getInstance()
+            .scheduleAtFixedRate(
+                new ReportTask(),
+                this,
+                reportingInterval,
+                reportingInterval,
+                reportingIntervalTimeUnit);
     log.debug("started metrics aggregator");
   }
 
diff --git a/dd-trace-core/src/main/java/datadog/trace/common/metrics/OkHttpSink.java b/dd-trace-core/src/main/java/datadog/trace/common/metrics/OkHttpSink.java
@@ -83,8 +83,8 @@ public void accept(int messageCount, ByteBuffer buffer) {
     } else {
       if (asyncTaskStarted.compareAndSet(false, true)) {
         this.future =
-            AgentTaskScheduler.INSTANCE.scheduleAtFixedRate(
-                new Sender(enqueuedRequests), this, 1, 1, SECONDS);
+            AgentTaskScheduler.getInstance()
+                .scheduleAtFixedRate(new Sender(enqueuedRequests), this, 1, 1, SECONDS);
       }
       sendAsync(messageCount, buffer);
     }
diff --git a/dd-trace-core/src/main/java/datadog/trace/core/CoreTracer.java b/dd-trace-core/src/main/java/datadog/trace/core/CoreTracer.java
@@ -793,8 +793,8 @@ private CoreTracer(
     // Schedule the metrics aggregator to begin reporting after a random delay of 1 to 10 seconds
     // (using milliseconds granularity.) This avoids a fleet of traced applications starting at the
     // same time from sending metrics in sync.
-    AgentTaskScheduler.INSTANCE.scheduleWithJitter(
-        MetricsAggregator::start, metricsAggregator, 1, SECONDS);
+    AgentTaskScheduler.getInstance()
+        .scheduleWithJitter(MetricsAggregator::start, metricsAggregator, 1, SECONDS);
 
     if (dataStreamsMonitoring == null) {
       this.dataStreamsMonitoring =
diff --git a/dd-trace-core/src/main/java/datadog/trace/core/StatusLogger.java b/dd-trace-core/src/main/java/datadog/trace/core/StatusLogger.java
@@ -27,7 +27,7 @@ public final class StatusLogger extends JsonAdapter<Config>
     implements AgentTaskScheduler.Task<Config>, JsonAdapter.Factory {
 
   public static void logStatus(Config config) {
-    AgentTaskScheduler.INSTANCE.schedule(new StatusLogger(), config, 500, MILLISECONDS);
+    AgentTaskScheduler.getInstance().schedule(new StatusLogger(), config, 500, MILLISECONDS);
   }
 
   @Override
diff --git a/dd-trace-core/src/main/java/datadog/trace/core/datastreams/DefaultDataStreamsMonitoring.java b/dd-trace-core/src/main/java/datadog/trace/core/datastreams/DefaultDataStreamsMonitoring.java
@@ -128,8 +128,13 @@ public DefaultDataStreamsMonitoring(
   public void start() {
     checkDynamicConfig();
     cancellation =
-        AgentTaskScheduler.INSTANCE.scheduleAtFixedRate(
-            new ReportTask(), this, bucketDurationNanos, bucketDurationNanos, TimeUnit.NANOSECONDS);
+        AgentTaskScheduler.getInstance()
+            .scheduleAtFixedRate(
+                new ReportTask(),
+                this,
+                bucketDurationNanos,
+                bucketDurationNanos,
+                TimeUnit.NANOSECONDS);
     thread.start();
   }
 
diff --git a/dd-trace-core/src/main/java/datadog/trace/core/monitor/TracerHealthMetrics.java b/dd-trace-core/src/main/java/datadog/trace/core/monitor/TracerHealthMetrics.java
@@ -155,8 +155,8 @@ public class TracerHealthMetrics extends HealthMetrics implements AutoCloseable
   public void start() {
     if (started.compareAndSet(false, true)) {
       cancellation =
-          AgentTaskScheduler.INSTANCE.scheduleAtFixedRate(
-              new Flush(), this, interval, interval, units);
+          AgentTaskScheduler.getInstance()
+              .scheduleAtFixedRate(new Flush(), this, interval, interval, units);
     }
   }
 
diff --git a/dd-trace-core/src/main/java/datadog/trace/core/scopemanager/ContinuableScopeManager.java b/dd-trace-core/src/main/java/datadog/trace/core/scopemanager/ContinuableScopeManager.java
@@ -430,8 +430,9 @@ private static final class RootIterationCleaner
 
     public static void scheduleFor(Map<ScopeStack, ContinuableScope> rootIterationScopes) {
       long period = Math.min(iterationKeepAlive, 10_000);
-      AgentTaskScheduler.INSTANCE.scheduleAtFixedRate(
-          CLEANER, rootIterationScopes, iterationKeepAlive, period, TimeUnit.MILLISECONDS);
+      AgentTaskScheduler.getInstance()
+          .scheduleAtFixedRate(
+              CLEANER, rootIterationScopes, iterationKeepAlive, period, TimeUnit.MILLISECONDS);
     }
 
     @Override
diff --git a/dd-trace-core/src/test/groovy/datadog/trace/core/test/DDCoreSpecification.groovy b/dd-trace-core/src/test/groovy/datadog/trace/core/test/DDCoreSpecification.groovy
@@ -1,10 +1,12 @@
 package datadog.trace.core.test
 
+import static datadog.trace.util.AgentThreadFactory.AgentThread.TASK_SCHEDULER
+
 import datadog.trace.api.DDSpanId
 import datadog.trace.api.DDTraceId
 import datadog.trace.api.StatsDClient
-import datadog.trace.api.sampling.PrioritySampling
 import datadog.trace.api.datastreams.NoopPathwayContext
+import datadog.trace.api.sampling.PrioritySampling
 import datadog.trace.bootstrap.instrumentation.api.ProfilingContextIntegration
 import datadog.trace.common.writer.ListWriter
 import datadog.trace.core.CoreTracer
@@ -14,8 +16,26 @@ import datadog.trace.core.DDSpanContext
 import datadog.trace.core.propagation.PropagationTags
 import datadog.trace.core.tagprocessor.TagsPostProcessorFactory
 import datadog.trace.test.util.DDSpecification
+import datadog.trace.util.AgentTaskScheduler
+import spock.lang.Shared
+
+import java.util.concurrent.TimeUnit
 
 abstract class DDCoreSpecification extends DDSpecification {
+  @Shared
+  static List<CoreTracer> unclosedTracers = []
+
+
+  static class AutoCloseableCoreTracerBuilder extends CoreTracerBuilder {
+    @Override
+    CoreTracer build() {
+      def ret = super.build()
+      unclosedTracers.add(ret)
+      ret
+    }
+  }
+
+
 
   protected boolean useNoopStatsDClient() {
     return true
@@ -36,12 +56,20 @@ abstract class DDCoreSpecification extends DDSpecification {
     TagsPostProcessorFactory.reset()
   }
 
+  @Override
+  void cleanup() {
+    unclosedTracers.each {it.close()}
+    unclosedTracers.clear()
+    AgentTaskScheduler.getInstance().shutdown(10, TimeUnit.SECONDS)
+    AgentTaskScheduler.INSTANCE = new AgentTaskScheduler(TASK_SCHEDULER)
+  }
+
   protected CoreTracerBuilder tracerBuilder() {
-    def builder = CoreTracer.builder()
+    def builder = new AutoCloseableCoreTracerBuilder()
     if (useNoopStatsDClient()) {
       builder = builder.statsDClient(StatsDClient.NO_OP)
     }
-    return builder.strictTraceWrites(useStrictTraceWrites())
+    builder.strictTraceWrites(useStrictTraceWrites())
   }
 
   protected DDSpan buildSpan(long timestamp, CharSequence spanType, Map<String, Object> tags) {
diff --git a/internal-api/src/main/java/datadog/trace/api/sampling/AdaptiveSampler.java b/internal-api/src/main/java/datadog/trace/api/sampling/AdaptiveSampler.java
diff --git a/internal-api/src/main/java/datadog/trace/util/AgentTaskScheduler.java b/internal-api/src/main/java/datadog/trace/util/AgentTaskScheduler.java
diff --git a/internal-api/src/main/java/datadog/trace/util/TempLocationManager.java b/internal-api/src/main/java/datadog/trace/util/TempLocationManager.java
diff --git a/telemetry/src/main/java/datadog/telemetry/dependency/DependencyService.java b/telemetry/src/main/java/datadog/telemetry/dependency/DependencyService.java

Original file line number	Diff line number	Diff line change
`@@ -159,7 +159,7 @@ public DDAgentFeaturesDiscovery featuresDiscovery(Config config) {`
`159`	`159`	`ret.discover(); // safe to run on same thread`
`160`	`160`	`} else {`
`161`	`161`	`// avoid performing blocking I/O operation on application thread`
`162`		`- AgentTaskScheduler.INSTANCE.execute(ret::discoverIfOutdated);`
	`162`	`+ AgentTaskScheduler.getInstance().execute(ret::discoverIfOutdated);`
`163`	`163`	`}`
`164`	`164`	`}`
`165`	`165`	`featuresDiscovery = ret;`
Original file line number	Diff line number	Diff line change
`@@ -87,8 +87,8 @@ private void scheduleConnect() {`
`87`	`87`	`log.debug(`
`88`	`88`	`"Scheduling StatsD connection in {} seconds - {}", remainingDelay, statsDAddress());`
`89`	`89`	`}`
`90`		`- AgentTaskScheduler.INSTANCE.scheduleWithJitter(`
`91`		`- ConnectTask.INSTANCE, this, remainingDelay, SECONDS);`
	`90`	`+ AgentTaskScheduler.getInstance()`
	`91`	`+ .scheduleWithJitter(ConnectTask.INSTANCE, this, remainingDelay, SECONDS);`
`92`	`92`	`} else {`
`93`	`93`	`doConnect();`
`94`	`94`	`}`
`@@ -168,8 +168,8 @@ private void doConnect() {`
`168`	`168`	`log.debug(`
`169`	`169`	`"Scheduling StatsD connection in {} seconds - {}", RETRY_DELAY, statsDAddress());`
`170`	`170`	`}`
`171`		`- AgentTaskScheduler.INSTANCE.scheduleWithJitter(`
`172`		`- ConnectTask.INSTANCE, this, RETRY_DELAY, SECONDS);`
	`171`	`+ AgentTaskScheduler.getInstance()`
	`172`	`+ .scheduleWithJitter(ConnectTask.INSTANCE, this, RETRY_DELAY, SECONDS);`
`173`	`173`	`} else {`
`174`	`174`	`log.debug("Max retries have been reached. Will not attempt again.");`
`175`	`175`	`}`
Original file line number	Diff line number	Diff line change
`@@ -48,7 +48,7 @@ public class DefaultCodeOriginRecorder implements CodeOriginRecorder {`
`48`	`48`	`private AgentTaskScheduler scheduler;`
`49`	`49`
`50`	`50`	`public DefaultCodeOriginRecorder(Config config, ConfigurationUpdater configurationUpdater) {`
`51`		`- this(config, configurationUpdater, AgentTaskScheduler.INSTANCE);`
	`51`	`+ this(config, configurationUpdater, AgentTaskScheduler.getInstance());`
`52`	`52`	`}`
`53`	`53`
`54`	`54`	`public DefaultCodeOriginRecorder(`
Original file line number	Diff line number	Diff line change
`@@ -58,7 +58,7 @@ public void accept(ConfigKey configKey, byte[] content, PollingRateHinter pollin`
`58`	`58`	`SymDbRemoteConfigRecord symDb = deserializeSymDb(content);`
`59`	`59`	`if (symDb.isUploadSymbols()) {`
`60`	`60`	`// can be long, make it async`
`61`		`- AgentTaskScheduler.INSTANCE.execute(this::startSymbolExtraction);`
	`61`	`+ AgentTaskScheduler.getInstance().execute(this::startSymbolExtraction);`
`62`	`62`	`} else {`
`63`	`63`	`stopSymbolExtraction();`
`64`	`64`	`}`