java Stream 的 Collectors.groupingBy 支持 key 为 null

最新推荐文章于 2024-02-02 17:38:58 发布

御剑长歌

最新推荐文章于 2024-02-02 17:38:58 发布

阅读量3.9k

点赞数 4

分类专栏： Java 文章标签： java 开发语言后端

本文链接：https://blog.csdn.net/weixin_40392053/article/details/121684071

版权

Java 专栏收录该内容

29 篇文章 1 订阅

订阅专栏

@Data
public class Org {
    private Integer id;
    private Integer pid;
    Org(Integer id, Integer pid) {
        this.id = id;
        this.pid = pid;
    }
}

    public static void main(String[] args) {
        Org root = new Org(1,null);
        Org org1 = new Org(2,1);
        Org org2 = new Org(3,1);
        Org org3 = new Org(4,2);
        Org org4 = new Org(5,2);
        List<Org> list = Arrays.asList(root,org1,org2,org3,org4);
        //若 pid 为 null 会报空指针异常
        Map<Integer, List<Org>>  error = list.stream().collect(Collectors.groupingBy(Org::getPid));
        //正确用法
        Map<Integer, List<Org>>  correct = list.stream().collect(Collectors.groupingBy(org -> Optional.ofNullable(org.getPid()).orElse(0)));
    }