Spring Boot 通用对象列表比较和去重
1、前言
在Excel批量导入数据时，常常需要与数据库中已存在数据的比较，并且需要考虑导入数据重复的可能性。
导入的行数据，一般有一个实体类（对象）与之对应，往往这个实体类在数据库中的字段要比导入数据更多，如主键ID字段，这个ID字段一般不会出现在导入行数据中，此时导入的对象使用其它的唯一键来识别，如导入某个单位的用户数据，用用户名或手机号来唯一识别用户，而不会有用户ID字段（此时新用户的用户ID还有待系统来分配）。
批量导入数据，往往需要与已存在对象数据进行比较，新的对象使用insert操作，已有对象使用update操作。但如果逐条数据，都先查询数据库，再决定是insert，还是update，是非常低效的，代码也会很臃肿。使用类似Mysql的ON DUPLICATE KEY UPDATE ，是一种解决方案，但在使用全局ID时，可能存在ID主键与其它唯一主键冲突的情况，导致update失败。
因此，稳妥的作法，是比较导入对象集与存在对象集，比较的结果产生了4个集合：新增数据集合、属性值完全相同的对象集合、对象键值相同但属性有变化的对象集合、剩余对象集合。
一般批量导入数据，会有一个限定，如导入用户数据，限定导入某一个单位的，这样比较的数据集就不会太大。
因此，对象列表的比较，是批量数据导入的基础性功能。另外，导入数据，首先要消除数据中重复的对象，这些重复的对象数据是异常数据。
2、解决方案
对象列表比较，应支持通用实体类对象，这样可实现代码的复用，消除低效重复的针对具体实体类的代码。因此需要支持泛型，且使用反射机制。
另外，对象比较，是一组属性字段值比较，包括键值与普通属性值，键值用于识别对象。
代码如下：
package com.abc.example.common.utils;
import java.lang.reflect.Field;
import java.util.List;
import java.util.Map;
import lombok.extern.slf4j.Slf4j;
/**
* @className : ObjectListUtil
* @description : 对象列表工具类
* @summary :
* @history :
* ------------------------------------------------------------------------------
* date version modifier remarks
* ------------------------------------------------------------------------------
* 2021/08/15 1.0.0 sheng.zheng 初版
*
*/
@Slf4j
public class ObjectListUtil {
/**
*
* @methodName : compareTwoList
* @description : 比较两个对象列表，新列表与旧列表对比，根据比较的属性字段字典，比较结果
* 1、新的对象；2、键值相同，属性不同，需要修改的对象；3、属性完全相同
* 这样有新增对象列表，修改对象列表，相同对象列表，剩余对象列表
* @param <T> : 泛型类型T
* @param fieldMap : 比较字段字典，key为字段名称，value为是否键值，1表示键值字段，0为普通字段。
* @param newList : 新列表，要求键值无重复
* @param oldList : 旧列表，要求键值无重复，在方法中将被改变
* @param addList : 新增对象列表
* @param sameList : 相同对象列表
* @param updateList: 修改对象列表
* @param remainderList: 剩余对象列表
* @history :
* ------------------------------------------------------------------------------
* date version modifier remarks
* ------------------------------------------------------------------------------
* 2021/08/15 1.0.0 sheng.zheng 初版
*
*/
public static <T> void compareTwoList(Map<String,Integer> fieldMap,
List<T> newList,List<T> oldList,
List<T> addList, List<T> sameList,
List<T> updateList, List<T> remainderList) {
// 遍历新列表
for(int i = 0; i < newList.size(); i++) {
T newItem = newList.get(i);
// 标记对象是否匹配
boolean found = false;
// 遍历旧列表，倒序
for (int j = oldList.size() - 1; j >= 0; j--) {
T oldItem = oldList.get(j);
// 比较两个对象
int compare = compareTwoItem(fieldMap,newItem,oldItem);
if (compare == 1) {
// 如果两个对象相同，加入相同列表中
sameList.add(newItem);
// 从列表中移除
oldList.remove(j);
found = true;
// 结束本轮遍历
break;
}else if(compare == 2) {
// pass
}else if(compare == 3) {
// 匹配对象，但属性不同，加入修改表中
updateList.add(newItem);
// 从列表中移除
oldList.remove(j);
found = true;
// 结束本轮遍历
break;
}else {
// 发生异常
return;
}
}
if (found == false) {
// 如果本轮遍历，未找到匹配项，加入新增列表中
addList.add(newItem);
}
}
// oldList中剩余的项，加入剩余列表中
for(int i = 0; i < oldList.size(); i++) {
T oldItem = oldList.get(i);
remainderList.add(oldItem);
}
}
/**
*
* @methodName : compareTwoItem
* @description : 比较两个相同类型的对象
* @param <T> : 泛型类型T
* @param fieldMap : 比较字段字典，key为字段名称，value为是否键值，1表示键值字段。
* @param newItem : 新的对象
* @param oldItem : 旧的对象
* @return : 返回值定义如下：
* 0 : 数据处理异常
* 1 : 对象完全相同（比较字段的字段值都相同）
* 2 : 对象不同（键值字段的字段值存在不相同）
* 3 : 对象相同，属性不同（键值字段的字段值相同，但属性值不同）
* @history :
* ------------------------------------------------------------------------------
* date version modifier remarks
* ------------------------------------------------------------------------------
* 2021/08/15 1.0.0 sheng.zheng 初版
*
*/
@SuppressWarnings("unchecked")
public static <T> int compareTwoItem(Map<String,Integer> fieldMap,T newItem, T oldItem) {
int retCode = 1;
try {
for (Map.Entry<String,Integer> entry : fieldMap.entrySet()) {
// 取得字段名称
String fieldName = entry.getKey();
Integer keyFlag = entry.getValue();
// 取得新对象的当前字段值
Class<T> newClazz = (Class<T>) newItem.getClass();
Field newField = newClazz.getDeclaredField(fieldName);
newField.setAccessible(true);
Object newValue = newField.get(newItem);
// 取得旧对象的当前字段值
Class<T> oldClazz = (Class<T>) oldItem.getClass();
Field oldField = oldClazz.getDeclaredField(fieldName);
oldField.setAccessible(true);
Object oldValue = oldField.get(oldItem);
// 比较两个属性字段的值
if (!newValue.equals(oldValue)) {
// 如果字段值不相等
if (keyFlag == 1) {
// 如果为键值字段，表示两个对象
return 2;
}else {
// 非键值字段，表示有属性发生改变
retCode = 3;
}
}
}
}catch (NoSuchFieldException e) {
e.printStackTrace();
log.error(e.getMessage());
return 0;
}catch (IllegalAccessException e) {
e.printStackTrace();
log.error(e.getMessage());
return 0;
}
return retCode;
}
/**
*
* @methodName : removeDuplicate
* @description : 对象列表去重
* @param <T> : 泛型类型T
* @param fieldMap : 比较字段字典，key为字段名称，value为是否键值，1表示键值字段。
* @param inputList : 对象列表，将被去重
* @param dupList : 重复的多余对象列表，将被去重
* @history :
* ------------------------------------------------------------------------------
* date version modifier remarks
* ------------------------------------------------------------------------------
* 2021/08/15 1.0.0 sheng.zheng 初版
*
*/
public static <T> void removeDuplicate(Map<String,Integer> fieldMap,List<T> inputList,List<T> dupList){
// 开始比较下标
int pos = 0;
while (true) {
if (inputList.size() -1 < pos) {
break;
}
// 标记对象是否重复
boolean found = false;
// 被比较对象
T compItem = inputList.get(pos);
// 遍历列表
for (int i = pos + 1; i < inputList.size(); i++) {
// 比较对象
T newItem = inputList.get(i);
int compare = compareTwoItem(fieldMap,newItem,compItem);
if (compare == 1 || compare == 3) {
// 键值相同，为重复对象
found = true;
// 结束本次比较
break;
}
}
if (found == true) {
// 对象重复
dupList.add(compItem);
// 移除重复项
inputList.remove(pos);
// 注意，移除后，当前位置不变
}else {
// 不重复，处理下一个
pos ++;
}
}
}
}
3、调用方法
调用方法如下：
// 去重处理
// 字段字典
Map<String,Integer> fieldMap = new HashMap<String,Integer>();
// 手机号和用户名作为对象识别属性
fieldMap.put("phoneNumber", 1);
fieldMap.put("userName", 1);
List<UserInfo> dupList = new ArrayList<UserInfo>();
ObjectListUtil.removeDuplicate(fieldMap,importUserList,dupList);
// dupList可以作为导入错误数据输出用
// 查询数据库中存在的数据，orgId为导入附加参数，表示单位ID
List<UserInfo> dbUserList = userDao.selectItemsByOrgId(orgId);
// 对比新旧数据列表
List<UserInfo> addList = new ArrayList<UserInfo>();
List<UserInfo> updateList = new ArrayList<UserInfo>();
List<UserInfo> sameList = new ArrayList<UserInfo>();
List<UserInfo> remainderList = new ArrayList<UserInfo>();
// 对象的其它属性字段，非键值
fieldMap.put("age", 0);
fieldMap.put("address", 0);
fieldMap.put("height", 0);
ObjectListUtil.compareTwoList(fieldMap, importUserList, dbUserList,
addList, sameList, updateList, remainderList);
// 对addList执行insert操作，可以批量insert
// 对updateList执行update操作，逐个update
