文章目录
- 一、问题
- 二、问题示例
- 三、原因
- 四、解决方法
- 4.1、方案一
- 4.2、方案二(推荐)
- 4.3、方案三
一、问题
发现Collectors.toMap的一个坑,key可以为null,但value不能为null,否则抛空指针异常nullPointerException。
二、问题示例
报错示例如下:
import lombok.AllArgsConstructor;
import lombok.Data;import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;public class Test {public static void main(String[] args){List<User> list = new ArrayList<>();User u1 = new User("张三",null);list.add(u1);//key:姓名 value:性别Map<String,String> userMap = list.stream().collect(Collectors.toMap(User::getName, User::getSex,(o1,o2)->o1));System.out.println(userMap);}
}@Data
@AllArgsConstructor
class User{//姓名private String name;//性别private String sex;
}
报错信息:
Exception in thread "main" java.lang.NullPointerExceptionat java.util.HashMap.merge(HashMap.java:1225)at java.util.stream.Collectors.lambda$toMap$58(Collectors.java:1320)at java.util.stream.ReduceOps$3ReducingSink.accept(ReduceOps.java:169)at java.util.ArrayList$ArrayListSpliterator.forEachRemaining(ArrayList.java:1382)at java.util.stream.AbstractPipeline.copyInto(AbstractPipeline.java:481)at java.util.stream.AbstractPipeline.wrapAndCopyInto(AbstractPipeline.java:471)at java.util.stream.ReduceOps$ReduceOp.evaluateSequential(ReduceOps.java:708)at java.util.stream.AbstractPipeline.evaluate(AbstractPipeline.java:234)at java.util.stream.ReferencePipeline.collect(ReferencePipeline.java:499)at com.springboot.store.Test.main(Test.java:17)
三、原因
由上面截图,可以找到是空指针报错,然后根据代码提示,
原来是HashMap.merge这个方法
at java.util.HashMap.merge(HashMap.java:1225)
所以顺着Collectors.toMap这个方法一直找下去
四、解决方法
4.1、方案一
使用Optional类处理value,优点是可以继续使用Collectors.toMap,缺点是为null的value会被改,而这可能违反业务上的期望。
//解决方案一,使用Optional类处理null 若value为空,设置默认值""
Map<String, String> map02 = list.stream().collect(Collectors.toMap(User::getName, s -> Optional.ofNullable(s.getSex()).orElse(""), (a, b) -> b));
System.out.println(map02);
输出结果:
{张三=}
注意:方案一中将null修改为了空引号,一定程度上还是修改了数据
4.2、方案二(推荐)
使用 Stream#collect(Supplier<R> supplier,BiConsumer<R, ? super T> accumulator,BiConsumer<R, R> combiner)
方法,可以保留为null的value。所以个人推荐方案二。
以上两个解决方法的代码如下:
import lombok.AllArgsConstructor;
import lombok.Data;import java.util.*;
import java.util.stream.Collectors;public class Test {public static void main(String[] args){List<User> list = new ArrayList<>();User u1 = new User("张三",null);list.add(u1);//key:姓名 value:性别//Map<String,String> userMap = list.stream().collect(Collectors.toMap(User::getName, User::getSex,(o1,o2)->o1));//System.out.println(userMap);//解决方案二,直接使用collect()方法进行规约操作,关于这里collect方法的使用可以看这篇文章 https://www.cnblogs.com/sueyyyy/p/13502116.htmlHashMap<String, String> map03 = list.stream().collect(HashMap::new, (map, item) -> map.put(item.getName(), item.getSex()), HashMap::putAll);System.out.println(map03);}
}@Data
@AllArgsConstructor
class User{//姓名private String name;//性别private String sex;
}
输出结果:
{张三=null}
4.3、方案三
与方案一类似,当value为null是返回空字符串。一定程度上修改了value值。
//解决方案三,使用Objects.nonNull处理null
Map<String, String> map03 = list.stream().collect(Collectors.toMap(User::getName, s -> Objects.nonNull(s.getSex()) ? s.getSex() : "", (a, b) -> b));System.out.println(map03);