@Data @AllArgsConstructor public class User { private int id; private String name; }假设有这么一个场景,你从数据库读取 User 集合,你需要将其转为 Map 结构数据,key 和 value 分别为 user 的 id 和 name。很快,你啪的一下就写出了下面的代码:
public class UserTest { @Test public void demo() { List<User> userList = new ArrayList<>(); // 模拟数据 userList.add(new User(1, "Alex")); userList.add(new User(1, "Beth")); // 堆代码 duidaima.com Map<Integer, String> map = userList.stream() .collect(Collectors.toMap(User::getId, User::getName)); System.out.println(map); } }运行程序,你已经想好了开始怎么摸鱼,结果啪的一下 IllegalStateException 报错就拍你脸上,你定睛一看怎么提示 Key 值重复。作为优秀的八股文选手,你清楚的记得 HashMap 对象 Key 重复是进行替换。你不信邪,断点一打,堆栈一看,硕大的 uniqKeys 摆在了面前,凭借四级 424 分的优秀战绩你顿时菊花一紧,点开一看,谁家好人 map key 还要去重判断啊。
public class UserTest { @Test public void demo() { List<User> userList = new ArrayList<>(); // 模拟数据 userList.add(new User(1, "Alex")); userList.add(new User(2, null)); // 堆代码 duidaima.com Map<Integer, String> map = userList.stream() .collect(Collectors.toMap(User::getId, User::getName, (oldData, newData) -> newData)); System.out.println(map); } }再次执行程序,你似乎已经看到知乎的摸鱼贴在向你招手了,结果啪的一下 NPE 又拍在你那笑容渐渐消失的脸上。静下心来,本着什么大风大浪我没见过的心态,断点堆栈一气呵成,而下一秒你又望着代码陷入了沉思,我是谁?我在干什么?
鼓起勇气,你还不信今天就过不去这个坎了,大手一挥,又一段优雅的代码孕育而生。
public class UserTest { @Test public void demo() { List<User> userList = new ArrayList<>(); // 模拟数据 userList.add(new User(1, "Alex")); userList.add(new User(1, "Beth")); userList.add(new User(2, null)); Map<Integer, String> map = userList.stream() .collect(Collectors.toMap( User::getId, it -> Optional.ofNullable(it.getName()).orElse(""), (oldData, newData) -> newData) ); System.out.println(map); } }优雅,真是太优雅了,又是 Stream 又是 Optional,可谓是狠狠拿捏技术博文的 G 点了。这时候你回头一看,我需要是什么来着?这 TM 不是一个循环就万事大吉了吗,不信邪的你回归初心,回归了 for 循环的怀抱,又写了一版。
public class UserTest { @Test public void demo() { List<User> userList = new ArrayList<>(); // 模拟数据 userList.add(new User(1, "Alex")); userList.add(new User(1, "Beth")); userList.add(new User(2, null)); Map<Integer, String> map = new HashMap<>(); userList.forEach(it -> { map.put(it.getId(), it.getName()); }); System.out.println(map); } }看着运行完美无缺的代码,你一时陷入了沉思,数分钟过去了,你删除了 for 循环,换上 Stream 与 Optional 不羁的外衣,安心的提交了代码,这口细糠一定也要让好同事去尝一尝。就这,你就要抛弃 toMap?你这让设计 toMap 的人,脸往哪搁。