反射实现合并两个java对象的值
2021-03-09 本文已影响0人
青玉_f18c
不废话,直接上代码,要求jdk8以上:
package com.qiyi.bi.predict.common.utils;
import com.qiyi.bi.predict.pojo.entity.ProjectEntity;
import org.apache.commons.lang3.reflect.FieldUtils;
import org.apache.commons.lang3.reflect.MethodUtils;
import org.apache.commons.lang3.tuple.ImmutablePair;
import org.apache.commons.lang3.tuple.Pair;
import java.lang.reflect.Field;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.time.LocalDate;
import java.util.Date;
import java.util.function.BiFunction;
/**
* @Description: ...
* @Author: Li Yalei - Robin
* @Date: 2021/3/9 11:50
*/
public class ObjectUtils extends org.apache.commons.lang3.ObjectUtils {
/**
* 合并 o1, o2 两个对象
* 当o1, o2其中有一个为null时,返回返回非null那个对象; 全为null时,返回null
* 都不为null时,将o2的字段值合并到 o1中。
*
* 合并规则:
* 1. 值相同,保留o1中的字段值
* 2. 值不同,a.有一个不为null,保留非null的那个值
* b.都不为null,由用户决定如何合并(调用operationWhenConflict方法,方法返回值为合并后的值,将赋值给o1的当前字段)
*
* 注意:合并的属性需要提供 set, get方法
* @param o1
* @param o2
* @param operationWhenConflict 当发生字段不一致时候用户的操作
* Field:发生冲突的字段,Pair: o1,o2中对应的冲突字段值
*
* @param <T> T: 待合并的对象,V:对象中的字段的值
*/
public static <T,V> T merge(T o1, T o2, BiFunction<Field, Pair<V,V>, V> operationWhenConflict) throws IllegalAccessException, InvocationTargetException {
if (o1 == null || o2 == null ? true : false) {
return o1 == null ? o2 : o1;
}
Class<?> clazz = o1.getClass();
//两者都不为null
for (Field field : FieldUtils.getAllFieldsList(clazz)) {
Method getMethod = getGetMethod(clazz, field.getName());
Method setMethod = getSetMethod(clazz, field.getName(), new Class[]{field.getType()});
if(getMethod==null || setMethod==null){
//没有set, get方法,跳过
continue;
}
// 值不一致
field.setAccessible(true);
if(notEqual(field.get(o1), field.get(o2))){
Object v1 = field.get(o1);;
Object v2 = field.get(o2);
if (v1 == null || v2 == null ? true : false) {
// 两者有且仅有一个为null
field.set(o1, v1 == null ? v2 : v1);
continue;
}
//两者都不为null
field.set(o1, operationWhenConflict.apply(field, new ImmutablePair<>((V)v1, (V)v2)));
}
}
return o1;
}
public static Method getGetMethod(Class<?> clazz, String fieldName) throws IllegalAccessException{
return setGetMethod(clazz, fieldName, "get");
}
public static Method getSetMethod(Class<?> clazz, String fieldName, Class<?>[] args) throws IllegalAccessException{
return setGetMethod(clazz, fieldName, "set", args);
}
private static Method setGetMethod(Class<?> clazz, String fieldName, String prefix, Class<?>... args) throws IllegalAccessException{
if(fieldName==null || "".equals(fieldName)){
throw new IllegalAccessException("The filed name isn't present, name="+fieldName);
}
String methodName = prefix;
if(fieldName.length()==1){
methodName += (fieldName.charAt(0)+"").toUpperCase();
}else {
methodName += (fieldName.charAt(0)+"").toUpperCase()+fieldName.substring(1);
}
Method method = MethodUtils.getAccessibleMethod(clazz, methodName, args);
return method;
}
public static void main(String[] args) throws Exception {
ProjectEntity p1 = new ProjectEntity();
p1.setProjectName("项目米");
p1.setPuyuId("123");
ProjectEntity p2 = new ProjectEntity();
p2.setEndTime(LocalDate.now());
p2.setPuyuId("112");
p2.setCreatedAt(new Date());
ProjectEntity p = merge(p1, p2, ((field, pair) -> {
return pair.getLeft() + "," + pair.getRight();
}));
}
}