通过Stream流找出集合中对象中某个属性重复的值.
在日常的开发中,经常会碰到对数据库查出来的数据还要进行二次的业务逻辑处理的情况,有时候需要对一个对象集合中的重复的元素进行剔除. 下面是一个demo. 可以找出对象List中,某个属性重复的值.
public class StreamTest { public static void main(String[] args) { ArrayList<Bean> list = new ArrayList<>(); Bean b1 = new Bean("200021", "张三", "男"); Bean b2 = new Bean("200025", "李四", "女"); Bean b3 = new Bean("200023", "王二麻子", "男"); Bean b4 = new Bean("200021", "狗蛋", "男"); Bean b5 = new Bean("200026", "哈皮", "女"); CollectionUtils.addAll(list,b1,b2,b3,b4,b5); List<String> repeatCode = list.stream().collect(Collectors.groupingBy(Bean::getCode, Collectors.counting())).entrySet().stream().filter(entry -> entry.getValue() > 1).map(Map.Entry::getKey).collect(Collectors.toList()); System.out.println("repeatCode = " + repeatCode); System.out.println("list = " + list); if (CollectionUtils.isNotEmpty(repeatCode)) { repeatCode.forEach(merId -> extractRepeat(list, merId)); } System.out.println("list = " + list); } public static void extractRepeat( List<Bean> compInfos, String repeatCode) { Optional<Bean> first = compInfos.stream().filter(o -> o.getCode().equals(repeatCode)).findFirst(); if (first.isPresent()) { Bean bean = first.get(); compInfos.removeIf(next -> next.getCode().equals(repeatCode));//先将所有符合这个重复的code的对象从源集合中删除. compInfos.add(bean);//再将取出来的第一个添加进去. } } } @AllArgsConstructor @Data class Bean{ String code; String name; String sex; }
讲解
上述repeatCode中的就是list中所有Bean中重复的code.
Collectors.groupingBy(Bean::getCode, Collectors.counting()) 这段代码的作用就是统计每个Code出现的次数.展开的完整写法为
for (Bean bean : list) { map.merge(bean.getCode(), 1L, Long::sum); }
然后通过Stream的filter将出现次数大于1的值放到list中.最后就是通过遍历将集合中所有code等于重复code的对象剔除.然后
再加入一个需要的对象.就完成了剔除某一个值重复的对象.