packagecom.springboot.demo.utils;importorg.springframework.jdbc.core.RowMapper;importjava.lang.reflect.Method;importjava.sql.ResultSet;importjava.sql.SQLException;importjava.util.HashMap;importjava.util.Map;public classRowMapperFactory {public static RowMapper getInstance(Classclazz) {returngetRowMapper(clazz);
}public static RowMapper getRowMapper(Classclazz) {return new RowMapper() {private AutoRowMapper rm = newAutoRowMapper(clazz);
@Overridepublic K mapRow(ResultSet rs, int rowNum) throwsSQLException {try{returnrm.mapRow(rs, rowNum, clazz.newInstance());
}catch (InstantiationException |IllegalAccessException e) {
e.printStackTrace();return null;
}
}
};
}/*** 这是一个用于简化 rowMapper 的编写辅助类。
*
* 本类会自动跳过 entity 中存在,但是在查询语句中不存在的列。
*
*@param entity type*/
private static class AutoRowMapper{private Map colName2SetterMap = new HashMap<>(); //数据列名到 setter 名称的映射
private Map setterMap = new HashMap<>(); //setter 名称到 setter 方法的映射
private Map setterParamMap = new HashMap<>(); //setter 名称到 setter 参数类型的映射
/*** 初始化
*
*@paramt 实体类的类对象*/
protected AutoRowMapper(Classt) {
Method[] ms=t.getMethods();for (int i = 0; i < ms.length; i++) {
String name=ms[i].getName();if (name.startsWith("set") && ms[i].getParameterCount() == 1) {
setterMap.put(name, ms[i]);
setterParamMap.put(name, ms[i].getGenericParameterTypes()[0].getTypeName());
colName2SetterMap.put(setterToColName(name), name);
}
}
}/*** 在子类中,要实现的 RowMapper.mapRow 中掉用此方法。
*
*@paramrs 结果集
*@paramrowNum 结果集行号
*@paramt entity 对象,用于装填查询结果行数据
*@return传入的 entity 对象*/
public T mapRow(ResultSet rs, introwNum, T t) {for(String col : colName2SetterMap.keySet()) {try{int index = rs.findColumn(col); //如果找不到列,就会抛出异常
String setterName =colName2SetterMap.get(col);
inject(setterMap.get(setterName), setterParamMap.get(setterName), rs, index, t);
}catch(SQLException ex) {continue;
}
}returnt;
}/*** 把 setter 名称转换为列名。如 setCreatedOn --> created_on*/
private staticString setterToColName(String setterName) {
String property= setterName.substring(3, setterName.length());
StringBuilder sb= new StringBuilder().append(property.charAt(0));for(int i = 1; i < property.length(); i++) {char c =property.charAt(i);if(Character.isUpperCase(c)) {
sb.append("");
}
sb.append(c);
}returnsb.toString().toLowerCase();
}/*** 把指定列按类型注入 entity.
*
* 目前支持的类字段类型有:
*
* java.lang.Boolean boolean
* java.lang.Byte byte
* java.lang.Long long
* java.lang.Integer int
* java.lang.Short short
* java.lang.Float float
* java.lang.Double double
* java.lang.Date
* java.lang.String
* java.sql.Blob
* java.math.BigDecimal
*
*/private void inject(Method getter, String fieldType, ResultSet rs, intindex, T t) {try{switch(fieldType) {case "java.lang.Boolean": //布尔值
case "boolean":
getter.invoke(t, rs.getBoolean(index));break;case "java.lang.Byte": //字节
case "byte":
getter.invoke(t, rs.getByte(index));break;case "java.lang.Long": //Long
case "long":
getter.invoke(t, rs.getLong(index));break;case "java.lang.Integer": //Int
case "int":
getter.invoke(t, rs.getInt(index));break;case "java.lang.Short": //Short
case "short":
getter.invoke(t, rs.getShort(index));break;case "java.lang.Float": //Float
case "float":
getter.invoke(t, rs.getFloat(index));break;case "java.lang.Double": //Double
case "double":
getter.invoke(t, rs.getDouble(index));break;case "java.lang.Date":
getter.invoke(t, rs.getDate(index));break;case "java.lang.String":
getter.invoke(t, rs.getString(index));break;case "java.sql.Blob":
getter.invoke(t, rs.getBlob(index));break;case "java.math.BigDecimal":
getter.invoke(t, rs.getBigDecimal(index));break;default:
getter.invoke(t, rs.getObject(index));break;
}
}catch(Exception ex) {
ex.printStackTrace();
}
}
}
}