Java8 Stream distinct 去重

• 参考
• 总结

Stream distinct 去重

1，distinct去重原理：类实现hashCode()和equals()方法

2，filter根据属性过滤去重：支持单属性与多重属性去重。
不足：多重属性去重必须提前已知属性个数。
核心代码：

// 去重predict方法
public static  Predicate distinctByKey(Function keyExtractor){
        Map seen = new ConcurrentHashMap<>();
        return t -> seen.putIfAbsent(keyExtractor.apply(t), Boolean.TRUE) == null;
}

// 单属性去重 按照age去重
        students.stream().filter(distinctByKey(Student::getAge)).forEach(System.out::println);
        

        System.out.println("---------cut line------------");

        // 多属性去重 按照name+age+type去重
        students.stream().filter(distinctByKey(Student::getName)).filter(distinctByKey(Student::getAge))
                .filter(distinctByKey(Student::getType)).forEach(System.out::println);
        
    }


//根据多个实体属性去重
List list3 = list.stream().filter(distinctByKey((p) -> (p.get属性名称1()))).filter(distinctByKey((p) -> (p.get属性名称2()))).collect(Collectors.toList());

3，TreeSet去重：定义Comparator比较器
核心代码：

// --------单属性去重 按照age去重---------
        TreeSet treeSet = students.stream().collect(Collectors.toCollection(() -> new TreeSet<>(Comparator.comparing(Student::getAge))));
        treeSet.forEach(System.out::println);
        
        System.out.println("-----------cut line--------");

        // 想把结果再转化成List，可以使用 collectingAndThen()方法 该方法先获取数据，然后再对数据进行操作
        List unique = students.stream().collect(
                Collectors.collectingAndThen(
                        Collectors.toCollection(() -> new TreeSet<>(Comparator.comparing(Student::getAge))), ArrayList::new)
        );
        unique.forEach(System.out::println);
        

        System.out.println("-----------cut line--------");

        // --------多属性去重 按照name+age+type去重--------
        List unique1 = students.stream().collect(
                Collectors.collectingAndThen(
                        Collectors.toCollection(() -> new TreeSet<>(Comparator.comparing(o -> o.getName() + "-" + o.getAge() + "-" + o.getType()))), ArrayList::new)
        );
        unique1.forEach(System.out::println);
        

比较器写法

@Test
public void test(){
    // 第一种
    Comparator comparator = (o1, o2) -> o1.getPrice() - o2.getPrice();
    int compare = comparator.compare(new Book("Java", 20.0), new Book("C++", 20.0));
    
    // 第二种
    Comparator comparing = Comparator.comparing(book -> book.getPrice());
    int compare2 = comparing.compare(new Book("Java", 20.0), new Book("C++", 20.0));

    // 输出是结果一样的
    System.out.println("compare = " + compare);
    System.out.println("compara2 = " + compare2);
}