如何使用Java 8的Stream實作清單去重？-java教程-PHP中文網

一. Stream 的distinct()方法

distinct()是Java 8 中 Stream 提供的方法，傳回的是由此流中不同元素所組成的流。 distinct()使用 hashCode() 和 eqauls() 方法來取得不同的元素。

因此，需要去重的類別必須實作 hashCode() 和 equals() 方法。換句話說，我們可以透過重寫客製化的 hashCode() 和 equals() 方法來達到某些特殊需求的去重。

distinct() 方法宣告如下：

Stream<T> distinct();

登入後複製

1.1 對於String 清單的去重

因為String 類別已經覆寫了equals() 和hashCode() 方法，所以可以去重成功。

@Test
public void listDistinctByStreamDistinct() {
  // 1. 对于 String 列表去重
  List<String> stringList = new ArrayList<String>() {{
    add("A");
    add("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<Student> 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