Spring Boot 通用对象列表比较和去重

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

hmoban主题是根据ripro二开的主题,极致后台体验,无插件,集成会员系统
自学咖网 » Spring Boot 通用对象列表比较和去重