JAVA基础篇（16）-实际项目中获取集合的交集/差集

image.png

需求：用户会上送一个新集合，需要和DB中旧集合比较。

新集合有，旧集合没有的数据会保存。
新集合有，旧集合有的数据会更新。
新集合没有，旧集合有的数据会删除。

上述需求可以转换为两个集合取差集/交集。

JDK为我们提供了对应的API方法：

    public static void main(String[] args) {

        List<String> newList = new ArrayList<>();
        newList.add("A");
        newList.add("B");
        newList.add("C");
        newList.add("D");

        List<String> oldList = new ArrayList<>();
        oldList.add("C");
        oldList.add("D");
        oldList.add("E");
        oldList.add("F");
}

求差集：

//最终得到的数据是C，D（需要插入的数据）
newList.removeAll(oldList);
System.out.println(JSON.toJSONString(newList));

//最终得到的数据是E,F（需要删除的数据）
oldList.removeAll(newList);
System.out.println(JSON.toJSONString(oldList));

求并集：

//最终得到的数据是C,D（需要更新的数据）
oldList.retainAll(newList);
System.out.println(JSON.toJSONString(oldList));

但是在实际开发中，JDK提供的方法并不能完全的满足业务需要。

业务场景需求：

上送的新集合，新集合中的元素主键id必须为空才会去保存。
新集合和旧集合只要主键id相同，那么就是一条记录，就需要去更新（若是使用ArrayList求并集的方法，那么必须重写equals方法，只要id相同那么就是true）。
获取差集/并集后不能去修改原集合。

//集合关系的包装类
@Data
public class ListRelationshipWrapper<T> {
    //需要插入的数据
    private List<T> insertInfos;
    //需要更新的数据
    private List<T> updateInfos;
    //需要删除的数据
    private List<T> deleteInfos;
}

public abstract class AbstractListRelationship {
    //获取到实体的主键
    protected abstract <ID>  ID renderKey();
    //根据id判断是否能插入
    protected abstract boolean checkIdSave();
    //判断实体是否相等
    protected abstract <T extends  AbstractListRelationship> boolean renderEquals(T data);
    //获取集合的关系
    public static  ListRelationshipWrapper getListRelationship(List<? extends AbstractListRelationship> oldData,
                                             List<? extends AbstractListRelationship> newData){
        ListRelationshipWrapper<AbstractListRelationship> wrapper = new ListRelationshipWrapper<>();
        List<AbstractListRelationship> insertInfos =new ArrayList();
        Map<Object,AbstractListRelationship> newDataMap = new HashMap<>();
        //新集合（1，3）
        for(AbstractBeanDifference data : newData){
            Object o = data.renderKey();
            if(data.checkIdSave()){
                //若是能插入，保存到insertDatas集合中。
                insertDatas.add(data);
            }else{
                //以id为key，保存对象
                newDataMap.put(o,data);
            }
        }
        wrapper.setInsertDatas(insertDatas);

        List<AbstractBeanDifference> deleteDatas = new ArrayList();
        List<AbstractBeanDifference> updateDatas = new ArrayList();
        //旧集合
        for(AbstractBeanDifference data : oldData){
            //包含id，则更新
            if (newDataMap.containsKey(oldData)) {
                //两个对象不相等，才去更新
                if (!data.renderEquals(newDataMap.get(oldData)))
                    updateDatas.add(newDataMap.get(oldData));
            } else {
                deleteDatas.add(newDataMap.get(oldData));
            }
        }
        wrapper.setDeleteDatas(deleteDatas);
        wrapper.setUpdateDatas(updateDatas);
        return differenceWrapper;
    }

}

JAVA基础篇（16）-实际项目中获取集合的交集/差集

推荐阅读