Java8利用Stream实现列表去重的方法详解

Java8利用Stream实现列表去重的方法详解

目录一. Stream 的distinct()方法1.1 对于 String 列表的去重1.2 对于实体类列表的去重二. 根据 List

一. Stream 的distinct()方法

distinct()是java 8 中 Stream 提供的方法，返回的是由该流中不同元素组成的流。distinct()使用 hashCode() 和 eqauls() 方法来获取不同的元素。

因此，需要去重的类必须实现 hashCode() 和 equals() 方法。换句话讲，我们可以通过重写定制的 hashCode() 和 equals() 方法来达到某些特殊需求的去重。

distinct() 方法声明如下：

Stream distinct();

1.1 对于 String 列表的去重

因为 String 类已经覆写了 equals() 和 hashCode() 方法，所以可以去重成功。

@Test

public void listDistinctByStreamDistinct() {

// 1. 对于 String 列表去重

List stringList = new ArrayList() {{

add("A");

aTjnPDIdd("A");

add("B");

add("B");

add("C");

}};

out.print("去重前：");

for (String s : stringList) {

out.print(s);

}

out.println();

stringList = stringList.stream().distinct().collect(Collectors.toList());

out.print("去重后：");

for (String s : stringList) {

out.print(s);

}

out.println();

}

结果如下：

去重前：AABBC去重后：ABC

1.2 对于实体类列表的去重

注：代码中我们使用了 Lombok 插件的 @Data注解，可自动覆写 equals() 以及 hashCode() 方法。

/**

* 定义一个实体类

*/

@Data

public class Student {

private String stuNo;

private String name;

}

@Test

public void listDistinctByStreamDistinct() throws jsonProcessingException {

ObjectMapper objectMapper = new ObjectMapper();

// 1. 对于 Student 列表去重

List studentList = getStudentList();

out.print("去重前：");

out.println(objectMapper.writeValueAsString(studentList));

studentList = studentList.stream().distinct().collect(Collectors.toList());

out.print("去重后：");

out.println(objectMapper.writeValueAsString(studentList));

}

结果如下：

去重前：[{"stuNo":"001","name":"Tom"},{"stuNo":"002","name":"Mike"},{"stuNo":"001","name":"Tom"}]去重后：[{"stuNo":"001","name":"Tom"},{"stuNo":"002","name":"Mike"}]

二. 根据 List

2.1 新建一个列表出来

@Test

public void distinctByProperty1() throws JsonProcessingException {

// 这里第一种方法我们通过新创建一个只有不同元素列表来实现根据对象某个属性去重

ObjectMapper objectMapper = new ObjectMapper();

List studentList = getStudentList();

out.print("去重前 :");

out.println(objectMapper.writeValueAsString(studentList));

studentList = studentList.stream().distinct().collect(Collectors.toList());

out.print("distinct去重后:");

out.println(objectMapper.writeValueAsString(studentList));

// 这里我们引入了两个静态方法，以及通过 TreeSet<> 来达到获取不同元素的效果

// 1. import static java.util.stream.Collectors.collectingAndThen;

// 2. import static java.util.stream.Collectors.toCollection;

studentList = studentList.stream().collect(

collectingAndThen(

toCollection(() -> new TreeSet<>(Comparator.comparing(Student::getName))), ArrayList::new)

);

out.print("根据名字去重后 :");

out.println(objectMapper.writeValueAsString(studentList));

}

结果如下：

去重前        :[{"stuNo":"001","name":"Tom"},{"stuNo":"001","name":"Tom"},{"stuNo":"003","name":"Tom"}]distinct去重后:[{"stuNo":"001","name":"Tom"},{"stuNo":"003","name":"Tom"}]根据名字去重后 :[{"stuNo":"001","name":"Tom"}]

2.2 通过 filter() 方法

我们首先创建一个方法作为 Stream.filter() 的参数，其返回类型为 Predicate，原理就是判断一个元素能否加入到 Set 中去，代码如下：

private static Predicate distinctByKey(Function super T, ?> keyExtractor) {

Set

return t -> seen.add(keyExtractor.apply(t));

}

使用如下：

@Test

public void distinctByProperty2() throws JsonProcessingException {

// 这里第二种方法我们通过过滤来实现根据对象某个属性去重

ObjectMapper objectMapper = new ObjectMapper();

List studentList = getStudentList();

out.print("去重前 :");

out.println(objectMapper.writeValueAsString(studentList));

studentList = studentList.stream().distinct().collect(Collectors.toList());

out.print("distinct去重后:");

out.println(objectMapper.writeValueAsString(studentList));

// 这里我们将 distinctByKey() 方法作为 filter() 的参数，过滤掉那些不能加入到 set 的元素

studentList = studentList.stream().filter(distinctByKey(Student::getName)).collect(Collectors.toList());

out.print("根据名字去重后 :");

out.println(objectMapper.writeValueAsString(studentList));

}

结果如下：

去重前        :[{"stuNo":"001","name":"Tom"},{"stuNo":"001","name":"Tom"},{"stuNo":"003","name":"Tom"}]distinct去重后:[{"stuNo":"001","name":"Tom"},{"stuNo":"003","name":"Tom"}]根据名字去重后 :[{"stuNo":"001","name":"Tom"}]

版权声明：本文内容由网络用户投稿，版权归原作者所有，本站不拥有其著作权，亦不承担相应法律责任。如果您发现本站中有涉嫌抄袭或描述失实的内容，请联系我们jiasou666@gmail.com 处理，核实后本网站将在24小时内删除侵权内容。