分享幾種 Java8 中通過 Stream 對列表進行去重的方法

本文轉(zhuǎn)載自:https://juejin.im/post/5cd6b719f265da03b2044d56,本站轉(zhuǎn)載出于傳遞更多信息之目的,版權(quán)歸原作者或者來源機構(gòu)所有。

1. Stream 的 distinct() 方法

distinct() 是Java 8 中 Stream 提供的方法,返回的是由該流中不同元素組成的流,distinct() 通過 hashCode()eqauls() 方法來判斷重復元素,因此,我們可以通過重寫 hashCode()equals() 方法來達到某些特殊需求的去重。

Stream<T> distinct();
1.1 對于 String 列表的去重

因為 String 類已經(jīng)覆寫了 equals()hashCode() 方法,所以可以去重成功。

// 1. 對于 String 列表去重
  List<String> stringList = new ArrayList<String>() {{
    add("A");
    add("A");
    add("B");
    add("B");
    add("C");
  }};
  logger.info("去重前:{}", stringList);
  stringList = stringList.stream().distinct().collect(Collectors.toList());
  logger.info("去重后:{}", stringList);

結(jié)果如下:


去重前:AABBC
去重后:ABC
1.2 對于實體類列表的去重

注:代碼中我們使用了 Lombok 插件的 @Data 注解,可自動覆寫 equalshashCode 方法。

/**
* 定義一個實體類
*/ 
@Data
public class Student {
 private String stuNo;
 private String name;
}
// 1. 對于 Student 列表去重
List<Student> studentList = getStudentList();
logger.info("去重前:{}", JSON.toJSONString(studentList));
studentList = studentList.stream().distinct().collect(Collectors.toList());
logger.info("去重后:{}", JSON.toJSONString(studentList));

結(jié)果如下:

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

2. 根據(jù) List<Object>Object 某個屬性去重

2.1 利用 TreeSet 通過 comparing 指定某個屬性來去重
List<Student> studentList = studentList.stream().collect(
      collectingAndThen(
        toCollection(() -> new TreeSet<>(Comparator.comparing(Student::getName))), ArrayList::new)
    );
2.2 通過 filter() 方法

我們首先創(chuàng)建一個方法作為 Stream.filter() 的參數(shù),其返回類型為 Predicate,原理就是判斷一個元素能否加入到 Set 中去,代碼如下:

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

使用如下

List<Student> studentList = studentList.stream().filter(distinctByKey(Student::getName)).collect(Collectors.toList());
最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時請結(jié)合常識與多方信息審慎甄別。
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務(wù)。

友情鏈接更多精彩內(nèi)容