前言

在實(shí)際項(xiàng)目中我們經(jīng)常會(huì)用到 List 轉(zhuǎn) Map 操作，在過(guò)去我們可能使用的是 for 循環(huán)遍歷的方式。舉個(gè)例子：

先定義類：

// 簡(jiǎn)單對(duì)象 @Accessors(chain = true) // 鏈?zhǔn)椒椒?@lombok.Dataclass User { private String id; private String name;}

然后有這樣一個(gè) List：

List<User> userList = Lists.newArrayList(new User().setId('A').setName('張三'),new User().setId('B').setName('李四'),new User().setId('C').setName('王五'));

我們希望轉(zhuǎn)成 Map 的格式為：

A-> 張三 B-> 李四 C-> 王五

過(guò)去的做法（循環(huán)）：

Map<String, String> map = new HashMap<>();for (User user : userList) { map.put(user.getId(), user.getName());}使用 Java8 特性

Java8 中新增了 Stream 特性，使得我們?cè)谔幚砑喜僮鲿r(shí)更方便了。

以上述例子為例，我們可以一句話搞定：

userList.stream().collect(Collectors.toMap(User::getId, User::getName));

當(dāng)然，如果希望得到 Map 的 value 為對(duì)象本身時(shí)，可以這樣寫(xiě)：

userList.stream().collect(Collectors.toMap(User::getId, t -> t));

或：

userList.stream().collect(Collectors.toMap(User::getId, Function.identity()));關(guān)于 Collectors.toMap 方法

Collectors.toMap 有三個(gè)重載方法：

toMap(Function<? super T, ? extends K> keyMapper, Function<? super T, ? extends U> valueMapper);toMap(Function<? super T, ? extends K> keyMapper, Function<? super T, ? extends U> valueMapper,BinaryOperator<U> mergeFunction);toMap(Function<? super T, ? extends K> keyMapper, Function<? super T, ? extends U> valueMapper,BinaryOperator<U> mergeFunction, Supplier<M> mapSupplier);

參數(shù)含義分別是：

keyMapper：Key 的映射函數(shù) valueMapper：Value 的映射函數(shù) mergeFunction：當(dāng) Key 沖突時(shí)，調(diào)用的合并方法 mapSupplier：Map 構(gòu)造器，在需要返回特定的 Map 時(shí)使用

還是用上面的例子，如果 List 中 userId 有相同的，使用上面的寫(xiě)法會(huì)拋異常：

List<User> userList = Lists.newArrayList(new User().setId('A').setName('張三'),new User().setId('A').setName('李四'), // Key 相同 new User().setId('C').setName('王五'));userList.stream().collect(Collectors.toMap(User::getId, User::getName));// 異常：java.lang.IllegalStateException: Duplicate key 張三 at java.util.stream.Collectors.lambda$throwingMerger$114(Collectors.java:133) at java.util.HashMap.merge(HashMap.java:1245) at java.util.stream.Collectors.lambda$toMap$172(Collectors.java:1320) at java.util.stream.ReduceOps$3ReducingSink.accept(ReduceOps.java:169) at java.util.ArrayList$ArrayListSpliterator.forEachRemaining(ArrayList.java:1374) at java.util.stream.AbstractPipeline.copyInto(AbstractPipeline.java:481) at java.util.stream.AbstractPipeline.wrapAndCopyInto(AbstractPipeline.java:471) at java.util.stream.ReduceOps$ReduceOp.evaluateSequential(ReduceOps.java:708) at java.util.stream.AbstractPipeline.evaluate(AbstractPipeline.java:234) at java.util.stream.ReferencePipeline.collect(ReferencePipeline.java:499) at Test.toMap(Test.java:17) ...

這時(shí)就需要調(diào)用第二個(gè)重載方法，傳入合并函數(shù)，如：

userList.stream().collect(Collectors.toMap(User::getId, User::getName, (n1, n2) -> n1 + n2));// 輸出結(jié)果：A-> 張三李四 C-> 王五

第四個(gè)參數(shù)（mapSupplier）用于自定義返回 Map 類型，比如我們希望返回的 Map 是根據(jù) Key 排序的，可以使用如下寫(xiě)法：

List<User> userList = Lists.newArrayList(new User().setId('B').setName('張三'),new User().setId('A').setName('李四'),new User().setId('C').setName('王五'));userList.stream().collect( Collectors.toMap(User::getId, User::getName, (n1, n2) -> n1, TreeMap::new));// 輸出結(jié)果：A-> 李四 B-> 張三 C-> 王五總結(jié)

到此這篇關(guān)于Java8中List轉(zhuǎn)Map(Collectors.toMap) 的文章就介紹到這了,更多相關(guān)Java8 List轉(zhuǎn)Map(Collectors.toMap)內(nèi)容請(qǐng)搜索好吧啦網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持好吧啦網(wǎng)！