java8 stream中Collectors.toMap空指針問題及解決
Collectors.toMap空指針問題
在工作中遇到了一個List轉(zhuǎn)Map的時候的一個NullPointException.
情形很簡單,問題出在Collectors.toMap,當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,那么當value里存在Null的時候必然會拋NullPointException
Collectors.toMap的坑
按照常規(guī)思維,往一個map里put一個已經(jīng)存在的key,會把原有的key對應的value值覆蓋,然而通過一次線上問題,發(fā)現(xiàn)Java8中的Collectors.toMap反其道而行之,它默認給拋異常,拋異常...
線上業(yè)務代碼出現(xiàn)Duplicate Key的異常,影響了業(yè)務邏輯,查看拋出異常部分的代碼,類似以下寫法:
Map<Integer, String> map = list.stream().collect(Collectors.toMap(Person::getId, Person::getName));
然后list里面有id相同的對象,結(jié)果轉(zhuǎn)map的時候居然直接拋異常了。。查源碼發(fā)現(xiàn)toMap方法默認使用了個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)文章
SpringBoot解析指定Yaml配置文件的實現(xiàn)過程
這篇文章主要介紹了SpringBoot解析指定Yaml配置文件,本文給大家介紹的非常詳細,對大家的學習或工作具有一定的參考借鑒價值,需要的朋友可以參考下2023-03-03java實現(xiàn)pgsql自動更新創(chuàng)建時間與更新時間的兩種方式小結(jié)
本文主要介紹了java實現(xiàn)pgsql自動更新創(chuàng)建時間與更新時間的兩種方式小結(jié),主要包括通過數(shù)據(jù)庫自身實現(xiàn)以及通過mybatisplus的TableField注解添加,具有一定的參考價值,感興趣的可以了解一下2024-01-01java比較器Comparable接口與Comaprator接口的深入分析
本篇文章是對java比較器Comparable接口與Comaprator接口進行了詳細的分析介紹,需要的朋友參考下2013-06-06