如何使用ByteBuddy创建动态代理

时间:2016-10-27 18:56:45

标签: java proxy bytecode-manipulation byte-buddy jvm-bytecode

在Java中,可以使用InvocationHandler的实现创建动态代理。尽管进行了JVM优化,但使用反射总是会有一些调用方法的开销。

为了尝试解决这个问题,我尝试使用ByteBuddy在运行时创建代理类,但文档在这方面似乎不够清楚。

如何创建MethodCallProxy以便将方法调用转发给某个类实例?

修改

为了更好地澄清我的问题,我提供了一个我想要实现的例子:

我正在构建一个RPC系统。在方法调用的每一侧,我都有一个定义契约的接口(当调用者/被调用者都在JVM下运行时)。

@Contract
interface ISomeService {
    fun someMethod(arg0: String, arg1: SomePojo): PojoResult
}

在调用站点,我注入一个代理,拦截所有方法调用并将它们转发给被调用者。

ByteBuddy()
    .subclass(Any::class.java)
    .implement(serviceClass)

    // Service contract method delegation
    .method(isDeclaredBy(serviceClass)).intercept(
      MethodDelegation
          .to(ServiceProxyInterceptor())
          .filter(not(isDeclaredBy(Any::class.java)))
    )

    .make()
    .load(this)
    .loaded as Class<T>

最后,在被调用者处,我有几个处理程序,每个服务方法一个,负责解组调用参数并将它们转发到服务实现。

@Service
class SomeServiceImpl {
    fun someMethod(arg0: String, arg1: SomePojo): PojoResult {
        // ...
    }
}

我可以使用代码生成解决此问题,但生成的jar文件可能会变得非常大。因此,我想创建这些处理程序的通用版本,并在每个实例中附加一个代理,该代理拦截对ISomeService的每个方法调用,并将它们转发到SomeServiceImpl

1 个答案:

答案 0 :(得分:5)

在Byte Buddy中创建代理类的方法有很多种。具体方法取决于您的用例。最简单的方法可能是使用InvocationHandlerAdapter。鉴于您要为SomeClass创建代理,您可以使用以下命令创建代理:

Class<? extends SomeClass> proxy = new ByteBuddy()
  .subclass(SomeClass.class)
  .method(ElementMatchers.any())
  .intercept(InvocationHandlerAdapter.of(invocationHandler))
  .make()
  .load(SomeClass.class.getClassLoader());

如果要创建具有不同实例的委托的代理,您还需要定义一个字段。这可以通过以下说明完成:

Class<? extends SomeClass> proxy = new ByteBuddy()
  .subclass(SomeClass.class)
  .defineField("handler", InvocationHandler.class, Visibility.PUBLIC)
  .method(ElementMatchers.any())
  .intercept(InvocationHandlerAdapter.toField("handler"))
  .make()
  .load(SomeClass.class.getClassLoader());

您可以通过反射或通过实现setter接口(例如:

)来设置上述字段
interface HandlerSetter {
  InvocationHandler getHandler();
  void setHandler(InvocationHandler handler);
}

Class<? extends SomeClass> proxy = new ByteBuddy()
  .subclass(SomeClass.class)
  .defineField("handler", InvocationHandler.class, Visibility.PUBLIC)
  .implement(HandlerSetter.class)
  .intercept(FieldAccessor.ofField("handler"))
  .method(ElementMatchers.any())
  .intercept(InvocationHandlerAdapter.toField("handler"))
  .make()
  .load(SomeClass.class.getClassLoader());

您现在可以实例化该类并将该类转换为接口以设置处理程序。

InvocationHandler之外,还有许多其他方法可以创建代理。一种方法是使用MethodDelegation,它更灵活,通常更快,并允许您按需调用超级方法。也可以使用MethodCallForwarding工具应用转发文章。您可以在相应的类javadoc中找到详细信息。