一、问题

发现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.NullPointerException
	at 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.html
        HashMap<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);