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

时间:2016-01-04 16:11:02

标签: java decorator bytecode byte-buddy

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

我无法修改对象的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是子类/ wrap的对象。 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的每个方法的实现,但我希望使用Byte Buddy有一种更简单的方法。

1 个答案:

答案 0 :(得分:5)

您可能希望使用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比它的声誉更快,特别是因为使用Byte Buddy倾向于暗示单形呼叫站点。

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

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

多次。这没有效果。此外,method.getDeclaringClass().cast(node)不是必需的。反射API为你做了演员。