如何使用 bytebuddy 在现有类中定义新方法并在同一类中的现有方法中添加对它的调用？

Question

我有一堂课，如下所示

public class HelloWorld{
  public void sayHelloWorld(){
    System.out.println("Hello World");
  }
}

现在我想使用 bytebuddy 向HelloWorld类添加另一个方法，并在sayHelloWorld中添加对新方法的调用。因此，假设在 bytebuddy 施展魔法之后，类将会是这样的。（我知道 bytebuddy 使用字节码而不是 java 源文件。下面的代码仅用于说明目的。）

public class HelloWorld{
  public void sayHelloWorld(){
    System.out.println("Hello World");
    sayHelloAgain()
  }
  public void sayHelloAgain(){
    System.out.println("Hello Again")
  }
}

1. 首先，bytebuddy 可以做到这一点吗？
2. 其次，如果可以的话，我该怎么做？我了解到bytebuddy可以用来重新定义方法，但不能修改方法体。这是真的？

如果有人能阐明这一点，那就太好了。蒂亚！

Answer 1

Byte Buddy 允许您以多种方式执行此操作。最直接的方法是定义一个拦截器来实现sayHelloAgainByte Buddy 创建委托：

public class HelloAgainDelegate {
  public static void sayHelloAgain() {
    System.out.println("Hello again");
  }
}

然后，您可以在重新定义的类上定义该方法，并对该sayHelloWorld方法进行变基以首先调用原始方法，然后调用另一个方法：

Class<?> type = new ByteBuddy()
  .rebase(HelloWorld.class)
  .defineMethod("sayHelloAgain", void.class, Visibility.PUBLIC)
  .intercept(MethodDelegation.to(HelloAgainDelegate.class))
  .method(named("sayHelloWorld"))
  .intercept(SuperMethodCall.INSTANCE
    .andThen(MethodCall.invoke(named("sayHelloAgain"))))
  .make()
  .load(HelloWorld.class.getClassLoader(), 
        ClassLoadingStrategy.Default.CHILD_FIRST)
  .getLoaded();

Object instance = type.newInstance();
type.getMethod("sayHelloWorld").invoke(instance);
type.getMethod("sayHelloAgain").invoke(instance);

在 Java 代码中，重新基化的类看起来像这样：

public class HelloWorld {
  synthetic void sayHelloWorld$origin() {
    System.out.println("Hello World");
  }  

  public void sayHelloWorld() {
    sayHelloWorld$origin();
    sayHelloAgain();
  }

  public void sayHelloAgain() {
    HelloAgainInterceptor.sayHelloAgain();
  }
}

如果此委托不适合您，您还可以使用Advice模板类内联代码：

class HelloAgainAdvice {
  @Advice.OnMethodExit
  static void sayHelloAgain() {
    System.out.println("Hello again");
  }
}

MethodDelegation您可以通过 来使用此类，而不是Advice.to(HelloAgainAdvice.class)。复制代码时，您将无法设置断点，但重新定义的类将是独立的。