Use AgentTaskScheduler.INSTANCE for background cleanup

DataDog · Nov 23, 2020 · faa701b · faa701b
1 parent 77c82dc
commit faa701b
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/dd-java-agent/agent-tooling/src/main/java/datadog/trace/agent/tooling/CaffeineWeakCache.java b/dd-java-agent/agent-tooling/src/main/java/datadog/trace/agent/tooling/CaffeineWeakCache.java
@@ -4,6 +4,7 @@
 import com.github.benmanes.caffeine.cache.Caffeine;
 import datadog.trace.api.Function;
 import datadog.trace.bootstrap.WeakCache;
+import datadog.trace.util.AgentTaskScheduler;
 import java.util.concurrent.TimeUnit;
 
 public class CaffeineWeakCache<K, V> implements WeakCache<K, V> {
@@ -22,6 +23,7 @@ public CaffeineWeakCache(long maxSize) {
             .weakKeys()
             .maximumSize(maxSize)
             .expireAfterAccess(10, TimeUnit.MINUTES)
+            .executor(AgentTaskScheduler.INSTANCE)
             .build();
   }