Stream 與 Map：toMap() 的使用需謹慎
在 JDK 8 中 Java 引入了讓人欲罷不能的 stream 流處理，可以說已經成為了我日常開發中不可或缺的一部分。
當完成一次流處理之后需要返回一個集成對象時，已經肌肉記憶的敲下 collect(Collectors.toList()) 或者 collect(Collectors.toSet())。你可能會想，toList 和 toSet 都這么便捷順手了，又怎么能少得了 toMap() 呢。
答應我，一定打消你的這個想法，否則這將成為你噩夢的開端。
什么？你不信，沒有什么比代碼讓人更痛徹心扉，讓我們直接上代碼。
讓我們先準備一個用戶實體類。
@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"));
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));
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 的人，臉往哪擱。
解決的辦法有很多，其中最簡單的之一就是，給它第三個合并參數，解決沖突。因為Collectors.toMap這個方法其實是有三個參數的，第一個是key，第二個是value，第三個是發生沖突的合并規則。默認采用的就是沖突之后拋出異常的處理。
