记Stream的toMap方法引发的问题

    有时候需要将List中的对象转化为Map的key-value形式方便快速获取对应的value,例如要将元素中的id和name分别作为map的key和value,一般来说需要遍历List,一个个构造map,不美观也不方便,JAVA8中提供了Stream,其中的toMap方法可以快速的晚上上面的操作。

    使用方式:

Map map = list.stream().collect(Collectors.toMap(Object::getId,Object::getName));

返回的map对象,key为id,value为name

 

注意问题:

1.重复的key问题

     往map进行put操作的时候即便是put一个重复的key也不会有问题,但是在这里不能这么使用。

     对于重复key问题,源码中有一个说明,意思是若key存在重复,会抛出IllegalStateException异常,应使用toMap(Function, Function, BinaryOperator)方法。

记Stream的toMap方法引发的问题_第1张图片

    所以如果我们的key可能存在重复的key,我们应该使用另一个toMap方法,按照说明使用这个方法如果存在重复的key,会调用equals方法来对code进行对应,如果相等则会按照提供的方法进行取值,下面的写法就提供了一个取值的方法 (k1,k2) -> k1,如果k1和k2相等,取k1作为key

记Stream的toMap方法引发的问题_第2张图片

Map map= list.stream().collect(Collectors.toMap(Object::getId,Object::getName,(k1,k2) -> k1));

2.value为null问题

    HashMap在使用的时候是允许key和value都为null的,但在这里key可以为null,但是value不允许为null,来看源码的实现,在进行合并的时候,对value进行了非空的校验,不允许vallue为空,所以在toMap之前要先把value为空的数据过滤掉,Stream提供了一个filter的过滤方法

记Stream的toMap方法引发的问题_第3张图片

    下面就只过滤出name不为空的数据,这样就可以避免value为空的情况,同理,这个filter也可以过滤符合其他条件的数据,不需要我们再去遍历list将剔除无用的数据。

Map map= list.stream().filter(t -> StringUtils.isNotBlank(t.getName())).collect(Collectors.toMap(Object::getId,Object::getName));

 

你可能感兴趣的:(JAVA,学习随笔)