ByteBuddy:在构造类时使用新定义的字段进行拦截

时间:2017-01-11 08:56:51

标签: java subclassing byte-buddy

我正在查看其他人的一些ByteBuddy代码。他使用ByteBuddy生成运行时子类,用作代理,将运行时的某些管理代码实现到特定对象中。

Class<? extends T> newSubClass = new ByteBuddy(ClassFileVersion.ofThisVm())
                .subclass(classType)
                .defineField("_core", Object.class, Visibility.PUBLIC) //<---
                .method(ElementMatchers.isDeclaredBy(classType))
                .intercept(InvocationHandlerAdapter.of((proxy, method, m_args) -> {
                    //TODO: Need to replace core with _core as core is a function argument and will make it bound
                    return proxyHandler(core, method, m_args); //<--
                }))
                .make()
                .load(roleType.getClassLoader(), ClassLoadingStrategy.Default.WRAPPER)
                .getLoaded();


T proxy = ReflectionHelper.newInstance(newSubClass, args);
newSubClass.getField("_core").set(proxy, core);

为了不将core对象直接绑定到lambda中,我想使用新定义的字段_core,这样我就可以重用生成的类(并且不会为每次调用函数重新生成它)。  有没有办法实现这个目标?

提前致谢。

1 个答案:

答案 0 :(得分:2)

您可以像定义方法一样定义自定义构造函数。定义构造函数的一个重点是,您需要另一个构造函数调用作为其第一条指令。您可以使用MethodCall::invoke调用构造函数,您可以将其与FieldAccessor::ofField结合使用。

这样,您可以定义类似于以下内容的类:

new ByteBuddy(ClassFileVersion.ofThisVm())
  .subclass(classType, ConstructorStrategy.Default.NO_CONSTRUCTORS)
  .defineConstructor(Visibility.PUBLIC)
    .withParameter(InvocationHandler.class)
    .intercept(MethodCall.invoke(classType.getDeclaredConstructor())
    .andThen(FieldAccessor.ofField("_core").setsArgumentAt(0)))
  .defineField("_core", InvocationHandler.class, Visibility.PUBLIC)
  .method(ElementMatchers.isDeclaredBy(classType))
    .intercept(InvocationHandlerAdapter.toField("_core"))
  .make();

这样,您可以为每个实例设置自定义InvocationHandler。如果您只想将状态存储在_core字段中并从拦截器访问此字段,请查看MethodDelegation

new ByteBuddy(ClassFileVersion.ofThisVm())
  .subclass(classType, ConstructorStrategy.Default.NO_CONSTRUCTORS)
  .defineConstructor(Visibility.PUBLIC)
    .withParameter(Object.class)
    .intercept(MethodCall.invoke(classType.getDeclaredConstructor())
    .andThen(FieldAccessor.ofField("_core").setsArgumentAt(0)))
  .defineField("_core", Object.class, Visibility.PUBLIC)
  .method(ElementMatchers.isDeclaredBy(classType))
    .intercept(MethodDelegation.to(MyHandler.class))
  .make();

public class MyHandler {
  @RuntimeType
  public static Object intercept(@FieldValue("_core") Object value) { ... }
}

您可能需要的其他注释包括@This@AllArguments@Origin@SuperCall。您需要的越少,代理的效率就越高。特别是@AllArguments由于其分配要求而价格昂贵。

请注意,在这种情况下,您的字段仅在超级构造函数调用之后设置。此外,您假设超类型中存在默认构造函数。或者,您可以实现自定义ConstructorStrategy

至于缓存,请查看Byte Buddy的TypeCache