java stream去重的几种方式

java stream去重的几种方式

• 使用 Stream 的 distinct() 方法
• 使用 collectingAndThen() 和 toCollection() 方法
• 使用 filter() 方法

使用 Stream 的 distinct() 方法

这个方法会根据元素的 hashCode() 和 equals() 方法来判断是否重复。如果是自定义的类，需要重写这两个方法。

示例:

//利用java8的stream去重
List uniqueList = list.stream()
					.distinct()
					.collect(Collectors.toList());
System.out.println(uniqueList.toString());

使用 collectingAndThen() 和 toCollection() 方法

这个方法可以根据元素的某个属性或者多个属性来去重，比如 name 或者 name+address。这个方法会使用 TreeSet 来排序元素，所以不能保持原来的顺序。

示例:

//根据name属性去重
List<User> lt = list.stream().collect(
        collectingAndThen(
                toCollection(() -> new TreeSet<>(Comparator.comparing(User::getName))),
                ArrayList::new)
);
System.out.println("去重后的:" + lt);

使用 filter() 方法

这个方法需要自定义一个 Predicate 函数，用一个 Set 来记录已经出现过的元素，然后过滤掉重复的元素。

示例:

//定义一个Predicate函数
private static <T> Predicate<T> distinctByKey(Function<? super T, ?> keyExtractor) {
    Set<Object> seen = ConcurrentHashMap.newKeySet();
    return t -> seen.add(keyExtractor.apply(t));
}

//根据age属性去重
list.stream().filter(distinctByKey(s -> s.getAge()))
        .forEach(System.out::println);