文章目錄

• 簡介
• Collector介紹
• 自定義Collector
• 總結

怎么在java中創建一個自定義的collector

簡介

在之前的java collectors文章里面，我們講到了stream的collect方法可以調用Collectors里面的toList()或者toMap()方法，將結果轉換為特定的集合類。

今天我們介紹一下怎么自定義一個Collector。

Collector介紹

我們先看一下Collector的定義：

Collector接口需要實現supplier(),accumulator(),combiner(),finisher(),characteristics()這5個接口。

同時Collector也提供了兩個靜態of方法來方便我們創建一個Collector實例。

我們可以看到兩個方法的參數跟Collector接口需要實現的接口是一一對應的。

下面分別解釋一下這幾個參數：

• supplier

Supplier是一個函數，用來創建一個新的可變的集合。換句話說Supplier用來創建一個初始的集合。accumulator

• accumulator

accumulator定義了累加器，用來將原始添加到集合中。

• combiner

combiner用來將兩個集合合并成一個。

• finisher

finisher將集合轉換為最終的集合類型。

• characteristics

characteristics表示該集合的特征。這個不是必須的參數。

有了這幾個參數，我們接下來看看怎么使用這些參數來構造一個自定義Collector。

自定義Collector

我們利用Collector的of方法來創建一個不變的Set：

public static <T> Collector<T, Set<T>, Set<T>> toImmutableSet() {return Collector.of(HashSet::new, Set::add,(left, right) -> {left.addAll(right);return left;}, Collections::unmodifiableSet);}

上面的例子中，我們HashSet::new作為supplier，Set::add作為accumulator，自定義了一個方法作為combiner，最后使用Collections::unmodifiableSet將集合轉換成不可變集合。

上面我們固定使用HashSet::new作為初始集合的生成方法，實際上，上面的方法可以更加通用：

public static <T, A extends Set<T>> Collector<T, A, Set<T>> toImmutableSet(Supplier<A> supplier) {return Collector.of(supplier,Set::add, (left, right) -> {left.addAll(right);return left;}, Collections::unmodifiableSet);}

上面的方法，我們將supplier提出來作為一個參數，由外部來定義。

看下上面兩個方法的測試：

@Testpublic void toImmutableSetUsage(){Set<String> stringSet1=Stream.of("a","b","c","d").collect(ImmutableSetCollector.toImmutableSet());log.info("{}",stringSet1);Set<String> stringSet2=Stream.of("a","b","c","d").collect(ImmutableSetCollector.toImmutableSet(LinkedHashSet::new));log.info("{}",stringSet2);}

輸出：

INFO com.flydean.ImmutableSetCollector - [a, b, c, d] INFO com.flydean.ImmutableSetCollector - [a, b, c, d]

總結

本文介紹了Collector和自定義Collector的實例，希望能對大家有所幫助。

本文的例子https://github.com/ddean2009/learn-java-streams/tree/master/customCollector

更多精彩內容且看：

• 區塊鏈從入門到放棄系列教程-涵蓋密碼學,超級賬本,以太坊,Libra,比特幣等持續更新
• Spring Boot 2.X系列教程:七天從無到有掌握Spring Boot-持續更新
• Spring 5.X系列教程:滿足你對Spring5的一切想象-持續更新
• java程序員從小工到專家成神之路（2020版）-持續更新中,附詳細文章教程

歡迎關注我的公眾號:程序那些事，更多精彩等著您！
更多內容請訪問 www.flydean.com

總結

以上是生活随笔為你收集整理的怎么在java中创建一个自定义的collector的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。