如何在Java中实现包装装饰器？

Question

问题是创建现有对象的动态增强版本。

我无法修改的Class。相反，我必须：

• 子类化
• 将现有对象包装在新对象中 Class
• 将所有原始方法调用委托给包装的对象
• 实现另一个接口定义的所有方法

要添加到现有对象的接口是：

public interface EnhancedNode {

  Node getNode();
  void setNode(Node node);

  Set getRules();
  void setRules(Set rules);

  Map getGroups();
  void setGroups(Map groups);

}

使用Byte Buddy，我设法继承了类并实现了我的接口。问题是委派给包装的对象。我发现做到这一点的唯一方法是使用反射速度太慢（我在应用程序上负担很重，性能至关重要）。

到目前为止，我的代码是：

Class<? extends Node> proxyType = new ByteBuddy()
     .subclass(node.getClass(), ConstructorStrategy.Default.IMITATE_SUPER_TYPE_PUBLIC)
     .method(anyOf(finalNode.getClass().getMethods())).intercept(MethodDelegation.to(NodeInterceptor.class))
     .defineField("node", Node.class, Visibility.PRIVATE)
     .implement(EnhancedNode.class).intercept(FieldAccessor.ofBeanProperty())
     .defineField("groups", Map.class, Visibility.PRIVATE)
     .implement(EnhancedNode.class).intercept(FieldAccessor.ofBeanProperty())
     .defineField("rules", Set.class, Visibility.PRIVATE)
     .implement(EnhancedNode.class).intercept(FieldAccessor.ofBeanProperty())
     .make()
     .load(getClass().getClassLoader(), ClassLoadingStrategy.Default.WRAPPER)
     .getLoaded();
enhancedClass = (Class<N>) proxyType;
EnhancedNode enhancedNode = (EnhancedNode) enhancedClass.newInstance();
enhancedNode.setNode(node);

Node子类/包装的对象在哪里。该NodeInterceptor转发调用的方法的getNode属性。

这里的代码NodeInterceptor：

public class NodeInterceptor {

  @RuntimeType
  public static Object intercept(@Origin Method method,
                               @This EnhancedNode proxy,
                               @AllArguments Object[] arguments)
        throws Exception {
      Node node = proxy.getNode();
      Object res;
      if (node != null) {
          res = method.invoke(method.getDeclaringClass().cast(node), arguments);
      } else {
          res = null;
      }
      return res;
  }
}

一切正常，但拦截方法太慢，我打算直接使用ASM添加Node的每种方法的实现，但我希望有一种使用字节伙伴的更简单方法。

Answer 1

您可能想使用Pipe而不是反射API：

public class NodeInterceptor {

  @RuntimeType
  public static Object intercept(@Pipe Function<Node, Object> pipe,
                                 @FieldValue("node") Node proxy) throws Exception {
      return proxy != null
        ? pipe.apply(proxy);
        : null;
  }
}

为了使用管道，首先需要安装它。如果您有可用的Java 8，则可以使用java.util.Function。否则，只需定义某种类型：

interface Function<T, S> { S apply(T t); }

你自己 类型的名称和方法无关。安装类型：

MethodDelegation.to(NodeInterceptor.class)
                .appendParameterBinder(Pipe.Binder.install(Function.class));

但是，您确定反射部分是应用程序性能问题的关键点吗？您是否正确缓存了生成的类，并且缓存有效地工作了？反射API比其声誉更快，尤其是因为使用字节伙伴倾向于隐含单态调用站点。

最后，一些一般性反馈。你在打电话

.implement(EnhancedNode.class).intercept(FieldAccessor.ofBeanProperty())

多次。这没有作用。另外，method.getDeclaringClass().cast(node)没有必要。反射API为您执行转换。