Java8中通过Stream使用Collectors.toMap及踩坑记录

1. 说明

        Java8中添加Stream流操作,可以简化部分操作,使得代码更为简洁。本文主要记录部分使用方法及踩坑记录

2. 简单示例

        如果需要将一个List转为Map,key为对象的userId, value为userName。

2.1 非Stream实现

在Java7中通常可以使用以下代码实现:

@Test
public void test01() {
        List list = new ArrayList<>();
        list.add(new User("1", null, null));
        list.add(new User("2", "zhang", "wu"));
        list.add(new User("1", "qiqi", "wu"));
        HashMap map = new HashMap<>();
        for (User user : list) {
            map.put(user.getUserId(), user.getUserName());
        }
        System.out.println(map);
}

2.2 Stream实现

@Test
public void test02() {
        List list = new ArrayList<>();
        list.add(new User("2", "zhang", "wu"));
        list.add(new User("1", "qiqi", "wu"));
        Map collect = list.stream().collect(Collectors.toMap(User::getUserId, User::getUserName));
        System.out.println(collect);
}

2.3 踩坑点

        但是如果按照2.2中在项目中去实现是远远不够的,实际使用中会出现很多bug。例如更换下输入数据便会报错。

List list = new ArrayList<>();
list.add(new User("1", null, null));
list.add(new User("2", "zhang", "wu"));
list.add(new User("1", "qiqi", "wu"));

报错场景及原因

  • 需要自定义key重复处理策略,不然key重复会报错java.lang.IllegalStateException: Duplicate key xxx
  • value为null,会报空指针异常java.lang.NullPointerException

解决方案

  • 自定义重复策略, 例如:(s1, s2) -> s1
  • 处理value为空问题,例如: u.getUserName() == null ? "" : u.getUserName()
@Test
public void test03() {
        List list = new ArrayList<>();
        list.add(new User("1", null, null));
        list.add(new User("2", "zhang", "wu"));
        list.add(new User("1", "qiqi", "wu"));
        Map collect = list.stream().collect(Collectors.toMap(User::getUserId, u ->
                        u.getUserName() == null ? "" : u.getUserName()
                , (s1, s2) -> s1));
        System.out.println(collect);
}

你可能感兴趣的:(Java,java,开发语言)