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

Question

我有一个如下所示的课程

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

现在，我想使用bytebuddy向HelloWorld类添加另一个方法，并在sayHelloWorld中添加对新方法的调用。所以假设在比特巴迪表演了魔法之后，这个班级会变成这样。（我知道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允许您以各种方式执行此操作。最直接的方法是定义一个拦截器，该拦截器实现sayHelloAgain，Byte 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");
  }
}

您可以通过建议使用这个类，而不是方法委派。到（HelloAgainAdvice.class）。复制代码时，您将无法设置断点，但重新定义的类将是自包含的。