issue-525:Fix the issue where ttlagent does not work when adding othe…

…r agent
alibaba · Jul 6, 2023 · 0892946 · 0892946
1 parent d2407c2
commit 0892946
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/ttl-agent/src/main/java/com/alibaba/ttl3/agent/TtlAgent.java b/ttl-agent/src/main/java/com/alibaba/ttl3/agent/TtlAgent.java
@@ -14,6 +14,10 @@
 import java.util.ArrayList;
 import java.util.List;
 import java.util.Map;
+import java.util.PriorityQueue;
+import java.util.concurrent.PriorityBlockingQueue;
+import java.util.concurrent.ScheduledThreadPoolExecutor;
+import java.util.concurrent.ThreadPoolExecutor;
 
 /**
  * TTL Java Agent.
@@ -218,6 +222,8 @@ public static void premain(final String agentArgs, @NonNull final Instrumentatio
 
             final ClassFileTransformer transformer = new TtlTransformer(transformletList, isLogClassTransform());
             inst.addTransformer(transformer, true);
+
+            inst.retransformClasses(ThreadPoolExecutor.class, ScheduledThreadPoolExecutor.class, PriorityBlockingQueue.class, PriorityQueue.class);
             logger.info("[TtlAgent.premain] add Transformer " + transformer.getClass().getName() + " success");
 
             logger.info("[TtlAgent.premain] end");