问题是创建现有对象的动态增强版本。
我无法修改对象的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 的每个方法的实现,但我希望使用 Byte Buddy 有更简单的方法。