如何使用ByteBuddy创建动态代理

Question

在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。

Answer 1

在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，它更灵活，通常更快，并允许您按需调用超级方法。转发检测也可以使用MethodCall或Forwarding检测。您可以在相应的类javadoc中找到详细信息。