使用字节伙伴在java级别进行仪器测试

Question

我有一个线程池示例的代码，如下所示

public class RunThreads{
static final int MAX_TASK = 3;
public static void main(String[] args)
{
    Runnable r1 = new Task("task 1");
    Runnable r2 = new Task("task 2");
    Runnable r3 = new Task("task 3");
    Runnable r4 = new Task("task 4");
    Runnable r5 = new Task("task 5");
    ExecutorService pool = Executors.newFixedThreadPool(MAX_TASK);
    pool.execute(r1);
    pool.execute(r2);
    pool.execute(r3);
    pool.execute(r4);
    pool.execute(r5);
    pool.shutdown();
}}

和

class Task implements Runnable{
private String name;
public Task(String s)
{
    name = s;
}
public void run()
{
    try
    {
        for (int i = 0; i<=5; i++)
        {
            if (i==0)
            {
                Date d = new Date();
                SimpleDateFormat ft = new SimpleDateFormat("hh:mm:ss");
                System.out.println("Initialization Time for"
                        + " task name - "+ name +" = " +ft.format(d));
                //prints the initialization time for every task
            }
            else
            {
                Date d = new Date();
                SimpleDateFormat ft = new SimpleDateFormat("hh:mm:ss");
                System.out.println("Executing Time for task name - "+
                        name +" = " +ft.format(d));
                // prints the execution time for every task
            }
            Thread.sleep(1000);
        }
        System.out.println(name+" complete");
    }
    catch(InterruptedException e)
    {
        e.printStackTrace();
    }
}}

i 为仪器 java 创建一个小型代理ThreadPoolExecutor

如下所示

public class Agent {
public static void premain(String arguments, Instrumentation instrumentation) {
    new AgentBuilder.Default()
            .with(new AgentBuilder.InitializationStrategy.SelfInjection.Eager())
            .type((ElementMatchers.nameContains("ThreadPoolExecutor")))
            .transform(
                    new AgentBuilder.Transformer.ForAdvice()
                            .include(MonitorInterceptor.class.getClassLoader())
                            .advice(ElementMatchers.any(), MonitorInterceptor.class.getName())
            ).installOn(instrumentation);
}}

我们可以使用 Byte Buddy 来检测像 ThreadPoolExecutor 这样的 java 类吗？当我调试线程池执行器类工作时。但是当我使用代理尝试此操作时，线程池执行器类永远不会起作用。

编辑这是我的监视器拦截器

public class MonitorInterceptor {

@Advice.OnMethodEnter
static void enter(@Advice.Origin String method) throws Exception {
    System.out.println(method);
}

编辑

new AgentBuilder.Default()
            .with(new AgentBuilder.InitializationStrategy.SelfInjection.Eager())
            .with(AgentBuilder.Listener.StreamWriting.toSystemError())
            .ignore(none())
            .type((ElementMatchers.nameContains("ThreadPoolExecutor")))
            .transform((builder, typeDescription, classLoader, module) -> builder
                    .constructor(ElementMatchers.any())
                    .intercept(Advice.to(MyAdvice.class))
                    .method(ElementMatchers.any())
                    .intercept(Advice.to(MonitorInterceptor.class))
            ).installOn(instrumentation);

Answer 1

除非显式配置，否则 Byte Buddy 不会检测核心 Java 类。您可以通过显式设置不排除此类类的忽略匹配器来更改此设置。

在这种情况下，使用 Advice 时不需要配置初始化策略。

您可能还想限制建议的范围，现在您可以拦截任何方法或构造函数。

为了找出问题所在，您还可以定义一个 AgentBuilder.Listener 来接收错误通知。

Answer 2

使用拉斐尔·温特豪德的答案，我解决了这个问题。我创建一个代理如下，

new AgentBuilder.Default()
                .ignore(ElementMatchers.none())
                .type(ElementMatchers.nameContains("ThreadPoolExecutor"))
                .transform((builder, type, classLoader, module) -> builder
                        .visit(Advice.to(ThreadPoolExecutorAdvice.class).on(ElementMatchers.any()))
                ).installOn(instrumentation); 

使用这个我们可以检测Java类。像这样形成这个构造函数给出

java.util.concurrent.ThreadPoolExecutor$Worker(java.util.concurrent.ThreadPoolExecutor,java.lang.Runnable)

但是在代码构造函数中不是那样的，它是

public ThreadPoolExecutor(int corePoolSize,
                              int maximumPoolSize,
                              long keepAliveTime,
                              TimeUnit unit,
                              BlockingQueue<Runnable> workQueue) {
        this(corePoolSize, maximumPoolSize, keepAliveTime, unit, workQueue,
             Executors.defaultThreadFactory(), defaultHandler);
    }

所以我期待类似的东西

java.util.concurrent.ThreadPoolExecutor(int,int,long,java.util.concurrent.TimeUnit,java.util.concurrent.BlockingQueue)

我使用 javassist 来获取构造函数并按照字节伙伴给出的内容获取它。因此，使用 .ignore(ElementMatchers.none()) 和 visit(Advice.to(ThreadPoolExecutorAdvice.class).on(ElementMatchers.any()))我们可以在 Java 级类中获得所有构造函数和方法