利用反射给hibernate实体类赋值

今天做到一个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;
	}
以上

猜你喜欢

转载自blog.csdn.net/qq_36804701/article/details/80366699
今日推荐