背景
码哥很早之前基于工作需要, 写了个List对象的交、并、补的求解方法(两个对象List根据属性取交集和差集),发现很多小伙伴都有类似的需求, 刚好最近码哥有空了,就着手把之前的简陋方法封装下, 让它更通用些~
闲话少说, 直接上代码
public final class ListCompareUtil {
private static final String ERROR_MESSAGE = "输入列表或比较条件不能为null";
private ListCompareUtil() {
throw new IllegalStateException("ListCompareUtil类不可实例化");
}
/**
* 比较两个列表,返回在新列表中而不在现有列表中的元素
*
* @param <T> 列表元素的类型
* @param newList 新列表
* @param existingList 现有列表
* @param condition 用于比较列表元素的条件
* @return 在新列表中而不在现有列表中的元素列表
*/
public static <T> List<T> getDifferenceList(List<T> newList, List<T> existingList, BiPredicate<T, T> condition) {
if (newList == null || existingList == null || condition == null) {
throw new IllegalArgumentException(ERROR_MESSAGE);
}
List<T> resultList = newList.stream()
.filter(item -> existingList.stream().noneMatch(existingItem -> condition.test(item, existingItem)))
.collect(Collectors.toList());
// 返回一个不可修改的列表,防止外部修改结果列表
return Collections.unmodifiableList(resultList);
}
/**
* 求两个列表的并集
*
* @param <T> 列表元素的类型
* @param newList 新列表
* @param existingList 现有列表
* @param condition 用于比较列表元素的条件
* @return 两个列表的并集
*/
public static <T> List<T> getUnionList(List<T> newList, List<T> existingList, BiPredicate<T, T> condition) {
if (newList == null || existingList == null || condition == null) {
throw new IllegalArgumentException(ERROR_MESSAGE);
}
List<T> resultList = new ArrayList<>(newList);
existingList.forEach(existingItem -> {
if (newList.stream().noneMatch(newItem -> condition.test(newItem, existingItem))) {
resultList.add(existingItem);
}
});
// 返回一个不可修改的列表,防止外部修改结果列表
return Collections.unmodifiableList(resultList);
}
/**
* 求两个列表的交集
*
* @param <T> 列表元素的类型
* @param newList 新列表
* @param existingList 现有列表
* @param condition 用于比较列表元素的条件
* @return 两个列表的交集
*/
public static <T> List<T> getIntersectionList(List<T> newList, List<T> existingList, BiPredicate<T, T> condition) {
if (newList == null || existingList == null || condition == null) {
throw new IllegalArgumentException(ERROR_MESSAGE);
}
List<T> resultList = newList.stream()
.filter(item -> existingList.stream().anyMatch(existingItem -> condition.test(item, existingItem)))
.collect(Collectors.toList());
// 返回一个不可修改的列表,防止外部修改结果列表
return Collections.unmodifiableList(resultList);
}
}
简单的使用方式示例
@Test
public void getIntersectionList() {
List<String> existList = Arrays.asList("apple", "banana", "orange", "grape");
List<String> newList = Arrays.asList("apple", "grape", "mango");
List<String> diffResult = ListCompareUtil.getIntersectionList(newList, existList, String::equals);
System.out.println(diffResult);
}
@Test
public void getUnionList() {
List<String> existList = Arrays.asList("apple", "banana", "orange", "grape");
List<String> newList = Arrays.asList("apple", "grape", "mango");
List<String> diffResult = ListCompareUtil.getUnionList(newList, existList, String::equals);
System.out.println(diffResult);
}
@Test
public void getDifferenceList() {
List<String> existList = Arrays.asList("apple", "banana", "orange", "grape");
List<String> newList = Arrays.asList("apple", "grape", "mango");
List<String> diffResult = ListCompareUtil.getDifferenceList(newList, existList, String::equals);
System.out.println(diffResult);
}
比对方式可自己定义, 比如多个属性的比对, 老规矩, 给需要的伙伴一个简单的示例, 自行调整
@Test
public void test(){
List<Person> newList = Arrays.asList(
new Person("Alice", 25),
new Person("Bob", 30),
new Person("Charlie", 22)
);
List<Person> existingList = Arrays.asList(
new Person("Alice", 25),
new Person("David", 28)
);
BiPredicate<Person, Person> condition = (user1, user2) ->
user1.getName().equals(user2.getName()) && user1.getAge() == user2.getAge();
List<Person> differenceList = ListCompareUtil.getDifferenceList(newList, existingList, condition);
List<Person> unionList = ListCompareUtil.getUnionList(newList, existingList, condition);
List<Person> intersectionList = ListCompareUtil.getIntersectionList(newList, existingList, condition);
System.out.println("Difference List:");
differenceList.forEach(user -> System.out.println(user.getName() + " " + user.getAge()));
System.out.println("Union List:");
unionList.forEach(user -> System.out.println(user.getName() + " " + user.getAge()));
System.out.println("Intersection List:");
intersectionList.forEach(user -> System.out.println(user.getName() + " " + user.getAge()));
}
感谢
不知不觉粉丝过5000了, 感谢大家关注~
如果你觉得有用, 记得三连, 不得不说, 有反馈时更新的会更勤快~