作者sssyoyo (柚子)
看板java
標題[問題] 多對多關係如何優雅合併?
時間Sat Apr 9 15:25:35 2022
各位好，想請教是否有更好的寫法，主要針對stream的用法
標題可能描述的不太精確，問題簡化後模擬是這樣:
例如有"使用者"與"群組"，他們是多對多關係，存在一張DB的中間表
(一個使用者可以有多個群組身分，一個群組中也能存在多個使用者)
今天我查詢以"使用者"出發，要列出所屬的群組有哪些，返回JSON像這樣
[
{"userId":1,"groups":[1,2,3]},
{"userId":2,"groups":[2,3]},
{"userId":3,"groups":[3]}
]
我寫的方法
https://i.imgur.com/iqnKqPE.png
=======以下是文字版=======
// 模擬資料
List<User> list = new ArrayList<>();
list.add(new User(1, 1));
list.add(new User(1, 2));
list.add(new User(1, 3));
list.add(new User(2, 2));
list.add(new User(3, 3));
list.add(new User(2, 3));
List<UserVo> result = new ArrayList<>();
for (User user : list) {
UserVo userVo = null;
// 如果Id已經存在
Optional<UserVo> matched = result.stream().filter(e -> user.getUserId().
equals(e.getUserId())).findFirst();
if (matched.isPresent()) {
userVo = matched.get();
List<Integer> groups = new ArrayList<>(userVo.getGroups());
groups.add(user.getGroup());
userVo.setGroups(groups);
} else {
userVo = UserVo.builder()
.userId(user.getUserId())
.groups(Arrays.asList(user.getGroup())).build();
result.add(userVo);
}
}
================
痛點在於我還需要多定義一個UserVo，感覺應該能直接拼出JSONObject
以及我總感覺Stream應該可以更優雅的做到，像是groupingBy、歸約之類
我這個寫的只要key(就是userId)已經存在，就需要clone一份既存的groups資料
一直new ArrayList<>好像也挺浪費資源的
想詢問是否有更好的寫法，還請不吝賜教，感謝!
--
法爺晚歸，體中藍盡，止有數百，途中二德，綴行甚遠。
新德欲不敬，老德曰：“噫！不可！夫親子者，萬人之上也。
欺牧薩於水道，戲賊獵於刀鋒，傷屌戰於八碼之外，斬殘術於須臾之間。
倘其怒，將我等一波A之，何如？人為刀俎，我為魚肉，不可犯上，止宜尾隨，以假其威。
新德不從，貓跳裂傷。法爺怒，Biu斃新德。老德急化鵪鶉起舞，以萌爺心苟全性命。
爺視之良久，大悅，開門而走。
--
※ 發信站: 批踢踢實業坊(ptt.cc), 來自: 220.134.136.97 (臺灣)
※ 文章網址: https://webptt.com/m.aspx?n=bbs/java/M.1649489153.A.CFE.html
1F：推 b9502056: 用兩個Map<Integer, Set<Integer>> 維護多對多關係 04/09 22:00
感謝，才發現map有一個computeIfPresent方法特別適合這種狀況
※ 編輯: sssyoyo (220.134.136.97 臺灣), 04/09/2022 23:57:29