有时候需要将List中的对象转化为Map的key-value形式方便快速获取对应的value,例如要将元素中的id和name分别作为map的key和value,一般来说需要遍历List,一个个构造map,不美观也不方便,JAVA8中提供了Stream,其中的toMap方法可以快速的晚上上面的操作。
使用方式:
Map<String,Object> 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)方法。
所以如果我们的key可能存在重复的key,我们应该使用另一个toMap方法,按照说明使用这个方法如果存在重复的key,会调用equals方法来对code进行对应,如果相等则会按照提供的方法进行取值,下面的写法就提供了一个取值的方法 (k1,k2) -> k1,如果k1和k2相等,取k1作为key
Map<String, String> 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的过滤方法
下面就只过滤出name不为空的数据,这样就可以避免value为空的情况,同理,这个filter也可以过滤符合其他条件的数据,不需要我们再去遍历list将剔除无用的数据。
Map<String, String> map= list.stream().filter(t -> StringUtils.isNotBlank(t.getName())).collect(Collectors.toMap(Object::getId,Object::getName));