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

本文主要是介绍Java8中通过Stream使用Collectors.toMap及踩坑记录，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

1. 说明

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

2. 简单示例

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

2.1 非Stream实现

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

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

2.3 踩坑点

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

List<User> 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<User> 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<String, String> collect = list.stream().collect(Collectors.toMap(User::getUserId, u ->u.getUserName() == null ? "" : u.getUserName(), (s1, s2) -> s1));System.out.println(collect);
}

这篇关于Java8中通过Stream使用Collectors.toMap及踩坑记录的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！

---