java - 如何使用 javassist/CGLib 修改/装饰从第 3 方 API 返回的对象

Question

我有一个第 3 方 API 调用，它返回以下对象:

public class A {
  protected void common() {
     System.out.println("common is called in A");
  }

  public void test1() {
    common();
    System.out.println("test1 is called in A");
  }
  public void test2() {
    common();
    System.out.println("test2 is called in A");
  }
}

但我想修改其行为，如以下 ModifiedA 显示:

public class ModifiedA extends A {
  @Override
  protected void common() {
     super.common();
     System.out.println("common is called in ModifiedA");
  }
}

所以我想做的是:

A a = 3rdPartyAPI_call();

//
// Now I'd like to get a ModifiedA which has changed common() behavior.
//

如何使用 javassist/CGLIB 来完成此任务？

一个简单的方法可能是喜欢这样:

public class ModifiedA extends A {
  private A a;
  public ModifiedA(final A a) {
     this.a = a;
  }
  //
  // Override every public method in A
  //

  @Override
  protected void common() {
     super.common();
     System.out.println("common is called in ModifiedA");
  }
}

但是由于 A 的定义来自第 3 方，非常复杂并且可能会发生变化，所以我想使用代理来执行此操作？

感谢您提前发表评论。

Answer 1

您可以使用 CGLib 来实现委托(delegate)者模式，而无需重写所有方法。根据风格，有几种不同的方法可以实现这一点，但这里有一种与您的示例类似的方法。

您可以使用 cglib Enhancer 包装实例:

public static <T> T wrapInstance(final T original) {
    Enhancer enhancer = new Enhancer();
    enhancer.setSuperclass(original.getClass());
    enhancer.setCallback(new MethodInterceptor() {
        @Override
        public Object intercept(Object obj, Method method, Object[] args, MethodProxy proxy) throws Throwable {
            Object returnValue = proxy.invoke(original, args);
            if (method.getName().equals("common")) {
                System.out.println("common is called");
            }
            return returnValue;
        }
    });
    return (T) enhancer.create();
}