package com.ruoyi.common.utils;
|
|
import java.util.*;
|
import java.util.function.Function;
|
import java.util.stream.Collectors;
|
|
public class ArrayListUtils {
|
|
/**
|
* 根据ID将两个List拆分为：仅List1独有、仅List2独有、两者共有
|
*
|
* @param list1 列表1
|
* @param list2 列表2
|
* @param idExtractor ID提取函数
|
* @return Map包含三个Key: "onlyIn1", "onlyIn2", "common"
|
*/
|
public static <T, K> Map<String, List<T>> splitListsById(List<T> list1, List<T> list2, Function<T, K> idExtractor) {
|
// 1. 提取ID集合，用于快速判断存在性
|
Set<K> ids1 = list1.stream().map(idExtractor).collect(Collectors.toSet());
|
Set<K> ids2 = list2.stream().map(idExtractor).collect(Collectors.toSet());
|
|
// 2. 计算ID层面的差集和交集
|
Set<K> onlyIn1Ids = new HashSet<>(ids1);
|
onlyIn1Ids.removeAll(ids2); // List1独有的ID
|
|
Set<K> onlyIn2Ids = new HashSet<>(ids2);
|
onlyIn2Ids.removeAll(ids1); // List2独有的ID
|
|
Set<K> commonIds = new HashSet<>(ids1);
|
commonIds.retainAll(ids2); // 共有的ID
|
|
// 3. 根据ID过滤原始对象列表
|
List<T> onlyIn1 = list1.stream()
|
.filter(item -> onlyIn1Ids.contains(idExtractor.apply(item)))
|
.collect(Collectors.toList());
|
|
List<T> onlyIn2 = list2.stream()
|
.filter(item -> onlyIn2Ids.contains(idExtractor.apply(item)))
|
.collect(Collectors.toList());
|
|
// 注意：共有部分通常合并两个列表中的元素，或者只取其中一个。
|
// 这里策略是：保留list1和list2中所有ID重合的对象（如果业务要求去重或合并对象属性，需额外处理）
|
List<T> common = new ArrayList<>();
|
common.addAll(list1.stream()
|
.filter(item -> commonIds.contains(idExtractor.apply(item)))
|
.collect(Collectors.toList()));
|
// common.addAll(list2.stream()
|
// .filter(item -> commonIds.contains(idExtractor.apply(item)))
|
// .collect(Collectors.toList()));
|
|
Map<String, List<T>> result = new HashMap<>();
|
result.put("onlyIn1", onlyIn1);
|
result.put("onlyIn2", onlyIn2);
|
result.put("common", common);
|
|
return result;
|
}
|
}
|
