java8 stream中Collectors.toMap空指針問題及解決
Collectors.toMap空指針問題
在工作中遇到了一個List轉(zhuǎn)Map的時候的一個NullPointException.
情形很簡單,問題出在Collectors.toMap,當(dāng)key值沖突的時候理論上會按照我們的代碼來替換value,但是這里有個小坑
list.stream().collect(Collectors.toMap(it -> it.getCategoryId(), it -> it.getCategoryImage() ,(k1,k2) -> k2));
可以看到map在key值沖突merge的時候會要求新的value不能為null.
這意味著,只要傳入了(k1,k2) -> k2處理key沖突的function,那么當(dāng)value里存在Null的時候必然會拋NullPointException

Collectors.toMap的坑
按照常規(guī)思維,往一個map里put一個已經(jīng)存在的key,會把原有的key對應(yīng)的value值覆蓋,然而通過一次線上問題,發(fā)現(xiàn)Java8中的Collectors.toMap反其道而行之,它默認(rèn)給拋異常,拋異常...
線上業(yè)務(wù)代碼出現(xiàn)Duplicate Key的異常,影響了業(yè)務(wù)邏輯,查看拋出異常部分的代碼,類似以下寫法:
Map<Integer, String> map = list.stream().collect(Collectors.toMap(Person::getId, Person::getName));
然后list里面有id相同的對象,結(jié)果轉(zhuǎn)map的時候居然直接拋異常了。。查源碼發(fā)現(xiàn)toMap方法默認(rèn)使用了個throwingMerger
public static <T, K, U>
Collector<T, ?, Map<K,U>> toMap(Function<? super T, ? extends K> keyMapper,
? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? Function<? super T, ? extends U> valueMapper) {
? ? return toMap(keyMapper, valueMapper, throwingMerger(), HashMap::new);
}
?
?
private static <T> BinaryOperator<T> throwingMerger() {
? ? return (u,v) -> { throw new IllegalStateException(String.format("Duplicate key %s", u)); };
}那么這個throwingMerger是哪里用的呢?
public static <T, K, U, M extends Map<K, U>>
Collector<T, ?, M> toMap(Function<? super T, ? extends K> keyMapper,
? ? ? ? ? ? ? ? ? ? ? ? ? ? Function<? super T, ? extends U> valueMapper,
? ? ? ? ? ? ? ? ? ? ? ? ? ? BinaryOperator<U> mergeFunction,
? ? ? ? ? ? ? ? ? ? ? ? ? ? Supplier<M> mapSupplier) {
? ? BiConsumer<M, T> accumulator
? ? ? ? ? ? = (map, element) -> map.merge(keyMapper.apply(element),
? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? valueMapper.apply(element), mergeFunction);
? ? return new CollectorImpl<>(mapSupplier, accumulator, mapMerger(mergeFunction), CH_ID);
}這里傳進去的是HashMap,所以最終走的是HashMap的merge方法。merge方法里面有這么一段代碼:
if (old != null) {
? ? V v;
? ? if (old.value != null)
? ? ? ? v = remappingFunction.apply(old.value, value);
? ? else
? ? ? ? v = value;
? ? if (v != null) {
? ? ? ? old.value = v;
? ? ? ? afterNodeAccess(old);
? ? }
? ? else
? ? ? ? removeNode(hash, key, null, false, true);
? ? return v;
}相信只看變量名就能知道這段代碼啥意思了。。如果要put的key已存在,那么就調(diào)用傳進來的方法。而throwingMerger的做法就是拋了個異常。所以到這里就可以知道寫的代碼為什么呲了。。
如果不想拋異常的話,自己傳進去一個方法即可,上述代碼可以改成:
Map<Integer, String> map = list.stream().collect(Collectors.toMap(Person::getId, Person::getName,(oldValue, newValue) -> newValue));
這樣就做到了使用新的value替換原有value。
寫代碼調(diào)方法時,多看源碼實現(xiàn),注意踩坑!
以上為個人經(jīng)驗,希望能給大家一個參考,也希望大家多多支持腳本之家。
相關(guān)文章
MybatisPlus自定義Sql實現(xiàn)多表查詢的示例
這篇文章主要介紹了MybatisPlus自定義Sql實現(xiàn)多表查詢的示例,文中通過示例代碼介紹的非常詳細(xì),對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2020-08-08
SpringBoot實現(xiàn)權(quán)限驗證的示例步驟
權(quán)限驗證是一種用于控制對系統(tǒng)資源和操作的訪問的機制。它允許開發(fā)人員定義誰可以執(zhí)行特定操作或訪問特定資源,并確保只有經(jīng)過授權(quán)的用戶才能執(zhí)行這些操作,這篇文章主要介紹了SpringBoot實現(xiàn)權(quán)限驗證,需要的朋友可以參考下2023-08-08
springboot集成PageHelper分頁失效的原因及解決
項目啟動初期,在集成mybatis的分頁插件,自定義封裝了一個分頁的工具類,方便后期項目的擴展,結(jié)果無法分頁了,怎么設(shè)置搞都沒辦法正常分頁,所以本文將給大家介紹一下springboot集成PageHelper分頁失效的原因及解決,需要的朋友可以參考下2023-10-10
詳解如何在SpringBoot中實現(xiàn)優(yōu)雅關(guān)閉
這篇文章主要介紹了如何在SpringBoot中實現(xiàn)優(yōu)雅關(guān)閉,SpringBoot應(yīng)用程序的關(guān)閉可以是崩潰,也可以是手動關(guān)閉的,Shutdown、Crash 和 Graceful 之間的區(qū)別在于,它控制決定了我們可以用這個事件做什么,本文中,一起研究下Spring Boot提供的開箱即用功能之一:優(yōu)雅關(guān)閉2024-09-09
基于SpringBoot實現(xiàn)自定義插件的流程詳解
在SpringBoot中,插件是一種擴展機制,它可以幫助我們在應(yīng)用程序中快速地添加一些額外的功能,在本文中,我們將介紹如何使用 SpringBoot實現(xiàn)自定義插件,需要的朋友可以參考下2023-06-06
Spring Boot整合MybatisPlus逆向工程(MySQL/PostgreSQL)
MyBatis-Plus是MyBatis的增強工具,本文主要介紹了Spring Boot整合MybatisPlus逆向工程(MySQL/PostgreSQL),具有一定的參考價值,感興趣的可以了解一下2021-07-07

