java - 如何使用 bytebuddy 向类添加私有(private)字段

Question

我有一堂这样的课

public class SampleClass {

    public void startMethod(int no){
    //some works!!
}
public void endMethod(String name){
    //some works!!
}    
}

我正在使用Advice来获取此方法的指标。

public class TimerAdvice {

@Advice.OnMethodEnter
static void enter(@Advice.Origin String method){

    if (method.equals("startMethod")) {
        //Metrics works
    }

}

@Advice.OnMethodExit
static void exit(@Advice.Origin String method){
    if (method.equals("endMethod")) {
       //Metrics works
        }
    }
}
}

我想将 @Advice.OnMethodEnter 中的指标计时器(或任何字段)添加到 SampleClass 并从 @Advice.OnMethodExit 获取它这是可以做到的吗？我想我的问题已经很清楚了。

Answer 1

您必须为每个建议方法定义两个类，然后将每个建议应用于您想要更改的方法。您可以使用 Byte Buddy 的类型构建器 API 定义字段:

new ByteBuddy()
  .redefine(SampleClass.class)
  .visit(Advice.to(TimerAdviceStart.class).on(named("startMethod"))
  .visit(Advice.to(TimerAdviceEnd.class).on(named("endMethod"))
  .defineField("foo", long.class, Visibility.PRIVATE)
  .make();

您现在可以使用 @Advice.Field 读取和写入 foo 字段。但是，由于线程问题，您应该小心。

此外，对于重新定义 Java 代理的类，无法执行此操作，因为 JVM 禁止向已加载的类添加字段。