Generator 代码实现
import java.lang.reflect.Field;
import java.util.HashMap;
import java.util.Map;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
public class ResultMapGenerator {
private static Pattern humpPattern = Pattern.compile("[A-Z]");
private static final Map<String, String> TYPE_MAP = new HashMap<>();
static {
TYPE_MAP.put("string", "VARCHAR");
TYPE_MAP.put("boolean", "BIT");
TYPE_MAP.put("byte", "TINYINT");
TYPE_MAP.put("short", "SMALLINT");
TYPE_MAP.put("integer", "INTEGER");
TYPE_MAP.put("long", "BIGINT");
TYPE_MAP.put("float", "REAL");
TYPE_MAP.put("double", "DOUBLE");
TYPE_MAP.put("date", "DATE");
TYPE_MAP.put("timestamp", "TIMESTAMP");
TYPE_MAP.put("time", "TIME");
TYPE_MAP.put("bigdecimal", "DECIMAL");
}
public static String generate(Class<?> clazz){
String pkgName = clazz.getName();
String clazzName = clazz.getSimpleName();
String resultMapId = Character.toLowerCase(clazzName.charAt(0))+clazzName.substring(1)+"Map";
StringBuilder resultMap = new StringBuilder();
resultMap.append("<resultMap id=\"");
resultMap.append(resultMapId);
resultMap.append("\" type=\"");
resultMap.append(pkgName);
resultMap.append("\">\n");
Field[] fields = clazz.getDeclaredFields();
for(Field f : fields){
String property = f.getName();
String javaType = f.getType().getSimpleName();
if("serialVersionUID".equals(property)){
continue;
}
resultMap.append(" <result column=\"");
resultMap.append(property2Column(property));
resultMap.append("\" jdbcType=\"");
resultMap.append(javaType2jdbcType(javaType.toLowerCase()));
resultMap.append("\" property=\"");
resultMap.append(property);
resultMap.append("\" />\n");
}
resultMap.append("</resultMap>");
return resultMap.toString();
}
private static String property2Column(String property){
Matcher matcher = humpPattern.matcher(property);
StringBuffer sb = new StringBuffer();
while(matcher.find()){
matcher.appendReplacement(sb, "_"+matcher.group(0).toLowerCase());
}
matcher.appendTail(sb);
return sb.toString();
}
private static String javaType2jdbcType(String javaType){
String jdbcType = TYPE_MAP.get(javaType);
return jdbcType==null ? "UNKNOWN" : jdbcType;
}
}
具体使用
String resultMap = ResultMapGenerator.generate(Entity.class);
System.out.println(resultMap);