今天做到一个excel导入的业务,可以对excel进行配置,决定要导入哪些字段,但是配置项里只能拿到数据库的原始字段,然后我就做了一个利用原始字段通过反射给Entity类赋值的方法
下面是全部代码
public class ReflectUtils {
public static String getFieldColumn(Class clazz,String columnName) throws Exception {
Field[] field = clazz.getDeclaredFields();
for(Field f:field) {
boolean fieldHasAnno = f.isAnnotationPresent(Column.class);
if(fieldHasAnno) {
Column column = f.getAnnotation(Column.class);
if(column.name().equals(columnName))
return f.getName();
}
String fieldName =f.getName();
fieldName = "get"+fieldName.substring(0, 1).toUpperCase() + fieldName.substring(1);
Method method = getDeclaredMethod(clazz, fieldName);
if(method == null)
continue;
Column column = method.getAnnotation(Column.class);
if(column.name().equals(columnName))
return f.getName();
}
return "";
}
public static Method getDeclaredMethod(Class clazz,String methodName) {
try {
return clazz.getDeclaredMethod(methodName);
} catch (NoSuchMethodException e) {
}
return null;
}
public static <T> T setField(T t, String columnName, Object o) throws Exception{
String fieldName = getFieldColumn(t.getClass(), columnName);
Field f = t.getClass().getDeclaredField(fieldName);
f.setAccessible(true);
f.set(t, o);
return t;
}
public static void main(String[] args) throws Exception {
ProductModel p = new ProductModel();
setField(p, "partnum","产品1");
System.out.println(p.getPartNum());
}
}
下面的方法参数是类和原始字段名,遍历类中的变量,对比变量的column标签的name值,如果相同则返回对应的变量名
public static String getFieldColumn(Class clazz,String columnName) throws Exception {
Field[] field = clazz.getDeclaredFields();
for(Field f:field) {//遍历类中有的变量
boolean fieldHasAnno = f.isAnnotationPresent(Column.class);
if(fieldHasAnno) {//判断变量上是否有column标签,如果有那么判断是否跟原始值相同
Column column = f.getAnnotation(Column.class);
if(column.name().equals(columnName))//相同则返回
return f.getName();
}
String fieldName =f.getName();//如果变量上没有标签,那么判断get方法上有没有标签
fieldName = "get"+fieldName.substring(0, 1).toUpperCase() + fieldName.substring(1);
Method method = getDeclaredMethod(clazz, fieldName);//将获得get方法的方法封装,防止报错
if(method == null)
continue;
Column column = method.getAnnotation(Column.class);
if(column.name().equals(columnName))
return f.getName();
}
return "";
}
先调用getFieldColumn获得变量名,然后利用反射直接赋值
public static <T> T setField(T t, String columnName, Object o) throws Exception{
String fieldName = getFieldColumn(t.getClass(), columnName);
Field f = t.getClass().getDeclaredField(fieldName);
f.setAccessible(true);
f.set(t, o);
return t;
}
以上