package com.qianwen.core.tool.beans;
|
|
import java.beans.PropertyDescriptor;
|
import java.util.HashMap;
|
import java.util.Iterator;
|
import java.util.Map;
|
import com.qianwen.core.tool.utils.ReflectUtil;
|
import org.springframework.asm.ClassVisitor;
|
import org.springframework.asm.Label;
|
import org.springframework.asm.Type;
|
import org.springframework.cglib.core.ClassEmitter;
|
import org.springframework.cglib.core.CodeEmitter;
|
import org.springframework.cglib.core.Constants;
|
import org.springframework.cglib.core.EmitUtils;
|
import org.springframework.cglib.core.MethodInfo;
|
import org.springframework.cglib.core.ObjectSwitchCallback;
|
import org.springframework.cglib.core.ReflectUtils;
|
import org.springframework.cglib.core.Signature;
|
import org.springframework.cglib.core.TypeUtils;
|
/**
|
* beanmap 发射器
|
*/
|
class BladeBeanMapEmitter extends ClassEmitter {
|
private static final Type BEAN_MAP = TypeUtils.parseType(BladeBeanMap.class.getName());
|
private static final Type FIXED_KEY_SET = TypeUtils.parseType("org.springframework.cglib.beans.FixedKeySet");
|
private static final Signature CSTRUCT_OBJECT = TypeUtils.parseConstructor("Object");
|
private static final Signature CSTRUCT_STRING_ARRAY = TypeUtils.parseConstructor("String[]");
|
private static final Signature BEAN_MAP_GET = TypeUtils.parseSignature("Object get(Object, Object)");
|
private static final Signature BEAN_MAP_PUT = TypeUtils.parseSignature("Object put(Object, Object, Object)");
|
private static final Signature KEY_SET = TypeUtils.parseSignature("java.util.Set keySet()");
|
private static final Signature NEW_INSTANCE = new Signature("newInstance", BEAN_MAP, new Type[]{Constants.TYPE_OBJECT});
|
private static final Signature GET_PROPERTY_TYPE = TypeUtils.parseSignature("Class getPropertyType(String)");
|
|
public BladeBeanMapEmitter(ClassVisitor v, String className, Class type, int require) {
|
super(v);
|
begin_class(46, 1, className, BEAN_MAP, null, "<generated>");
|
EmitUtils.null_constructor(this);
|
EmitUtils.factory_method(this, NEW_INSTANCE);
|
generateConstructor();
|
/*
|
Map<? extends String, ? extends PropertyDescriptor> getters = makePropertyMap(ReflectUtil.getBeanGetters(type));
|
Map<? extends String, ? extends PropertyDescriptor> setters = makePropertyMap(ReflectUtil.getBeanSetters(type));
|
*/
|
Map<String, PropertyDescriptor> getters = makePropertyMap(ReflectUtil.getBeanGetters(type));
|
Map<String, PropertyDescriptor> setters = makePropertyMap(ReflectUtil.getBeanSetters(type));
|
|
Map<String, PropertyDescriptor> allProps = new HashMap<>(32);
|
allProps.putAll(getters);
|
allProps.putAll(setters);
|
if (require != 0) {
|
Iterator<String> it = allProps.keySet().iterator();
|
while (it.hasNext()) {
|
String name = it.next();
|
if (((require & 1) != 0 && !getters.containsKey(name)) || ((require & 2) != 0 && !setters.containsKey(name))) {
|
it.remove();
|
getters.remove(name);
|
setters.remove(name);
|
}
|
}
|
}
|
generateGet(type, getters);
|
generatePut(type, setters);
|
String[] allNames = getNames(allProps);
|
generateKeySet(allNames);
|
generateGetPropertyType(allProps, allNames);
|
end_class();
|
}
|
|
private Map<String, PropertyDescriptor> makePropertyMap(PropertyDescriptor[] props) {
|
Map<String, PropertyDescriptor> names = new HashMap<>(16);
|
for (PropertyDescriptor prop : props) {
|
String propName = prop.getName();
|
if (!"class".equals(propName)) {
|
names.put(propName, prop);
|
}
|
}
|
return names;
|
}
|
|
private String[] getNames(Map<String, PropertyDescriptor> propertyMap) {
|
return (String[]) propertyMap.keySet().toArray(new String[0]);
|
}
|
|
private void generateConstructor() {
|
CodeEmitter e = begin_method(1, CSTRUCT_OBJECT, null);
|
e.load_this();
|
e.load_arg(0);
|
e.super_invoke_constructor(CSTRUCT_OBJECT);
|
e.return_value();
|
e.end_method();
|
}
|
|
private void generateGet(Class type, final Map<String, PropertyDescriptor> getters) {
|
final CodeEmitter e = begin_method(1, BEAN_MAP_GET, null);
|
e.load_arg(0);
|
e.checkcast(Type.getType(type));
|
e.load_arg(1);
|
e.checkcast(Constants.TYPE_STRING);
|
EmitUtils.string_switch(e, getNames(getters), 1, new ObjectSwitchCallback() { // from class: org.springblade.core.tool.beans.BladeBeanMapEmitter.1
|
public void processCase(Object key, Label end) {
|
PropertyDescriptor pd = (PropertyDescriptor) getters.get(key);
|
MethodInfo method = ReflectUtils.getMethodInfo(pd.getReadMethod());
|
e.invoke(method);
|
e.box(method.getSignature().getReturnType());
|
e.return_value();
|
}
|
|
public void processDefault() {
|
e.aconst_null();
|
e.return_value();
|
}
|
});
|
e.end_method();
|
}
|
|
private void generatePut(Class type, final Map<String, PropertyDescriptor> setters) {
|
final CodeEmitter e = begin_method(1, BEAN_MAP_PUT, null);
|
e.load_arg(0);
|
e.checkcast(Type.getType(type));
|
e.load_arg(1);
|
e.checkcast(Constants.TYPE_STRING);
|
EmitUtils.string_switch(e, getNames(setters), 1, new ObjectSwitchCallback() { // from class: org.springblade.core.tool.beans.BladeBeanMapEmitter.2
|
public void processCase(Object key, Label end) {
|
PropertyDescriptor pd = (PropertyDescriptor) setters.get(key);
|
if (pd.getReadMethod() == null) {
|
e.aconst_null();
|
} else {
|
MethodInfo read = ReflectUtils.getMethodInfo(pd.getReadMethod());
|
e.dup();
|
e.invoke(read);
|
e.box(read.getSignature().getReturnType());
|
}
|
e.swap();
|
e.load_arg(2);
|
MethodInfo write = ReflectUtils.getMethodInfo(pd.getWriteMethod());
|
e.unbox(write.getSignature().getArgumentTypes()[0]);
|
e.invoke(write);
|
e.return_value();
|
}
|
|
public void processDefault() {
|
}
|
});
|
e.aconst_null();
|
e.return_value();
|
e.end_method();
|
}
|
|
private void generateKeySet(String[] allNames) {
|
declare_field(10, "keys", FIXED_KEY_SET, null);
|
CodeEmitter e = begin_static();
|
e.new_instance(FIXED_KEY_SET);
|
e.dup();
|
EmitUtils.push_array(e, allNames);
|
e.invoke_constructor(FIXED_KEY_SET, CSTRUCT_STRING_ARRAY);
|
e.putfield("keys");
|
e.return_value();
|
e.end_method();
|
CodeEmitter e2 = begin_method(1, KEY_SET, null);
|
e2.load_this();
|
e2.getfield("keys");
|
e2.return_value();
|
e2.end_method();
|
}
|
|
private void generateGetPropertyType(final Map allProps, String[] allNames) {
|
final CodeEmitter e = begin_method(1, GET_PROPERTY_TYPE, null);
|
e.load_arg(0);
|
EmitUtils.string_switch(e, allNames, 1, new ObjectSwitchCallback() { // from class: org.springblade.core.tool.beans.BladeBeanMapEmitter.3
|
public void processCase(Object key, Label end) {
|
PropertyDescriptor pd = (PropertyDescriptor) allProps.get(key);
|
EmitUtils.load_class(e, Type.getType(pd.getPropertyType()));
|
e.return_value();
|
}
|
|
public void processDefault() {
|
e.aconst_null();
|
e.return_value();
|
}
|
});
|
e.end_method();
|
}
|
}
|
