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的坑

按照常规思维，往一个map里put一个已经存在的key，会把原有的key对应的value值覆盖，然而通过一次线上问题，发现Java8中的Collectors.toMap反其道而行之，它默认给抛异常，抛异常...

线上业务代码出现DuplicateKey的异常，影响了业务逻辑，查看抛出异常部分的代码，类似以下写法：

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

然后list里面有id相同的对象，结果转map的时候居然直接抛异常了。。查源码发现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