java8stream中Collectors.toMap空指針問題及解決

第java8stream中Collectors.toMap空指針問題及解決目錄Collectors.toMap空指針問題Collectors.toMap的坑

Collectors.toMap空指針問題

在工作中遇到了一個List轉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一個已經存在的key，會把原有的key對應的value值覆蓋，然而通過一次線上問題，發(fā)現Java8中的Collectors.toMap反其道而行之，它默認給拋異常，拋異常...

線上業(yè)務代碼出現DuplicateKey的異常，影響了業(yè)務邏輯，查看拋出異常部分的代碼，類似以下寫法：

MapInteger,Stringmap=list.stream().collect(Collectors.toMap(Person::getId,Person::getName));

然后list里面有id相同的對象，結果轉map的時候居然直接拋異常了。。查源碼發(fā)現toMap方法默認使用了個throwingMerger

publicstaticT,K,U

CollectorT,,MapK,UtoMap(FunctionsuperT,extendsKkeyMapper,

FunctionsuperT,extendsUvalueMapper){

returntoMap(keyMapper,valueMapper,throwingMerger(),HashMap::new);

privatestaticTBinaryOperatorTthrowingMerger(){

return(u,v)-{thrownewIllegalStateException(String.format("Duplicatekey%s",u));};

}

那么這個throwingMerger是哪里用的呢？

publicstaticT,K,U,MextendsMapK,U

CollectorT,,MtoMap(FunctionsuperT,extendsKkeyMapper,

FunctionsuperT,extendsUvalueMapper,

BinaryOperatorUmergeFunction,

SupplierMmapSupplier){

BiConsumerM,Taccumulator

=(map,element)-map.merge(keyMapper.apply(element),

valueMapper.apply(element),mergeFunction);

returnnewCollectorImpl(mapSupplier,accumulator,mapMerger(mergeFunction),CH_ID);

}

這里傳進去的是HashMap，所以最終走的是HashMap的merge方法。merge方法里面有這么一段代碼：

if(old!=null){

Vv;

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);

returnv;

}

相信只看變量名就能知道這段代碼啥意思了。。如果要put的key已存在，那么就調用傳進來的方法。而throwingMerger的做法就是拋了個異常。所以到這里就可以知道寫的代碼為什么呲了。。

如果不想拋異常的話，自己傳進去一個方法即可，上述代碼可以改成：

MapInteger,Stringmap=list.stream().collect(Collectors.toMap(Person::getId,Person::get