单元测试之比较两个实体的属性值

方式一：使用反射的方法

[code]package www.lxk.com;

import java.beans.Introspector;
import java.beans.PropertyDescriptor;
import java.lang.reflect.Method;
import java.sql.Timestamp;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Date;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

public class CompareFieldsUtils {
/**
* 比较两个实体属性值，返回一个map以有差异的属性名为key，value为一个list分别存obj1,obj2此属性名的值
*
* @param obj1  进行属性比较的对象1
* @param obj2  进行属性比较的对象2
* @param ignoreArr 选择忽略比较的属性数组
* @return 属性差异比较结果map
*/
public static Map<String, List<Object>> compareFields(Object obj1, Object obj2, String[] ignoreArr) {
try {
Map<String, List<Object>> map = new HashMap<String, List<Object>>();
List<String> ignoreList = null;
if (ignoreArr != null && ignoreArr.length > 0) {
// array转化为list
ignoreList = Arrays.asList(ignoreArr);
}
if (obj1.getClass() == obj2.getClass()) {// 只有两个对象都是同一类型的才有可比性
Class<? extends Object> clazz = obj1.getClass();
// 获取object的属性描述
PropertyDescriptor[] pds = Introspector.getBeanInfo(clazz, Object.class).getPropertyDescriptors();
for (PropertyDescriptor pd : pds) {// 这里就是所有的属性了
String name = pd.getName();// 属性名
if (ignoreList != null && ignoreList.contains(name)) {// 如果当前属性选择忽略比较，跳到下一次循环
continue;
}
Method readMethod = pd.getReadMethod();// get方法
// 在obj1上调用get方法等同于获得obj1的属性值
Object o1 = readMethod.invoke(obj1);
// 在obj2上调用get方法等同于获得obj2的属性值
Object o2 = readMethod.invoke(obj2);
if (o1 instanceof Timestamp) {
o1 = new Date(((Timestamp) o1).getTime());
}
if (o2 instanceof Timestamp) {
o2 = new Date(((Timestamp) o2).getTime());
}
if (o1 == null && o2 == null) {
continue;
} else if (o1 == null && o2 != null) {
List<Object> list = new ArrayList<Object>();
list.add(o1);
list.add(o2);
map.put(name, list);
continue;
}
if (!o1.equals(o2)) {// 比较这两个值是否相等,不等就可以放入map了
List<Object> list = new ArrayList<Object>();
list.add(o1);
list.add(o2);
map.put(name, list);
}
}
}
return map;
} catch (Exception e) {
e.printStackTrace();
return null;
}
}

}

测试代码：

[code]package www.lxk.com;

import java.util.List;
import java.util.Map;

import lombok.AllArgsConstructor;
import lombok.Data;

public class CompareFieldsTest {

public static void main(String[] args) {
Student student01 = new Student("001", "zhangsan", 28, "nanjing");
Student student02 = new Student("002", "李四", 28, "suzhou");

Map<String, List<Object>> map01 = CompareFieldsUtils.compareFields(student01, student02, null);
System.out.println(map01);

System.out.println("--------------------------------");

String [] ignoreArr = {"id"};
Map<String, List<Object>> map02 = CompareFieldsUtils.compareFields(student01, student02,ignoreArr);
System.out.println(map02);

}

}

@AllArgsConstructor
@Data
class Student {
private String id;
private String name;
private int age;
private String city;

}

结果如下：