快捷導(dǎo)航

Java8-Stream流操作List去重問題

更新時(shí)間：2022年11月11日 15:21:17 作者：徐壽春

這篇文章主要介紹了Java8-Stream流操作List去重問題，具有很好的參考價(jià)值，希望對大家有所幫助。如有錯(cuò)誤或未考慮完全的地方，望不吝賜教

Java8Stream流操作List去重

根據(jù)屬性去重整體去重使用

distinct

ArrayList<LabelInfoDTO> collect = labelInfoDTOS.stream().
collect(Collectors.collectingAndThen(Collectors.toCollection(() -> new TreeSet<>(Comparator.comparing(LabelInfoDTO::getLabelCode))), ArrayList::new));

List列表運(yùn)用Java8的stream流按某字段去重

問題

項(xiàng)目中經(jīng)常會遇到列表去重的問題，一般可使用Java8的stream()流提供的distinct()方法：list.stream().distinct()。

list的類型為List<String>、List<Integer>，list里的元素為簡單包裝類型。

或者List<Xxx>，其中Xxx為自定義對象類型，重寫equals和hashCode方法，可根據(jù)業(yè)務(wù)情況來實(shí)現(xiàn)，如id相同即認(rèn)為對象相等。

有時(shí)會遇到這種情況，需要對按對象里的某字段來去重。

例如：

@NoArgsConstructor
@AllArgsConstructor
@Data
class Book {
?
? ? public static Book of(Long id, String name, String createTime) {
? ? ? ? return new Book(id, name, Date.from(LocalDateTime.parse(createTime, DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss")).atZone(ZoneId.systemDefault()).toInstant()));
? ? }
?
? ? private Long id;
?
? ? private String name;
?
? ? private Date createTime;
}

現(xiàn)在我們要按name字段來去重，假設(shè)list如下：

List<Book> books = new ArrayList<>();
books.add(Book.of(1L, "Thinking in Java", "2021-06-29 17:13:14"));
books.add(Book.of(2L, "Hibernate in action", "2021-06-29 18:13:14"));
books.add(Book.of(3L, "Thinking in Java", "2021-06-29 19:13:14"));

思路

1、重寫B(tài)ook類的equals和hashCode方法，以name來判斷比較是否相同，然后用stream的distinct方法來去重

代碼：

class Book {
? ? ...
?
? ? @Override
? ? public String toString() {
? ? ? ? return String.format("(%s,%s,%s)", id, name, DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss").format(createTime.toInstant().atZone(ZoneId.systemDefault()).toLocalDateTime()));
? ? }
?
? ? @Override
? ? public boolean equals(Object o) {
? ? ? ? if (this == o) return true;
? ? ? ? if (o == null || getClass() != o.getClass()) return false;
? ? ? ? Book book = (Book) o;
? ? ? ? return Objects.equals(name, book.name);
? ? }
}
?
List<Book> distinctNameBooks1 = books.stream().distinct().collect(Collectors.toList());
System.out.println(distinctNameBooks1);

總結(jié)：

通過重寫equals和hashCode方法，按實(shí)際需求來比較，可直接使用stream的distinct方法去重，比較方便；

有時(shí)對象類不方便或者不能修改，如它已實(shí)現(xiàn)好或者是引用的三方包不能修改，該方法不能靈活地按字段來去重。

2、通過Collectors.collectingAndThen的Collectors.toCollection，里面用TreeSet在構(gòu)造函數(shù)中指定字段

代碼：

List<Book> distinctNameBooks2 = books.stream().collect(Collectors.collectingAndThen(Collectors.toCollection(() -> new TreeSet<>(Comparator.comparing(o -> o.getName()))), ArrayList::new));
System.out.println(distinctNameBooks2);

總結(jié)：

使用stream流提供的方法，代碼很簡潔，但不足是雖然實(shí)現(xiàn)了去重效果，但list里的順序變化了，而有的場景需要保持順序。

3、通過stream的filter方法來去重，定義一個(gè)去重方法，參數(shù)為Function類型，返回值為Predicate類型

代碼：

public static <T> Predicate<T> distinctByKey(Function<? super T, Object> keyExtractor) {
? ? Map<Object, Boolean> map = new HashMap<>();
? ? return t -> map.putIfAbsent(keyExtractor.apply(t), Boolean.TRUE) == null;
}
?
List<Book> distinctNameBooks3 = books.stream().filter(distinctByKey(o -> o.getName())).collect(Collectors.toList());
System.out.println(distinctNameBooks3);

總結(jié)：

通過封裝定義一個(gè)去重方法，配合filter方法可靈活的按字段去重，保持了原列表的順序，不足之處是內(nèi)部定義了一個(gè)HashMap，有一定內(nèi)存占用，并且多了一個(gè)方法定義。

4、通過stream的filter方法來去重，不定義去重方法，在外面創(chuàng)建HashMap

代碼：

Map<Object, Boolean> map = new HashMap<>();
List<Book> distinctNameBooks4 = books.stream().filter(i -> map.putIfAbsent(i.getName(), Boolean.TRUE) == null).collect(Collectors.toList());
System.out.println(distinctNameBooks4);

總結(jié)：

仍然是配合filter方法實(shí)現(xiàn)去重，沒有單獨(dú)創(chuàng)建方法，臨時(shí)定義一個(gè)HashMap，保持了原列表的順序，不足之處是有一定內(nèi)存占用。

PS：暫時(shí)沒找到stream流原生支持的可按某字段去重并且保持原列表順序的方法

以上為個(gè)人經(jīng)驗(yàn)，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。

您可能感興趣的文章: