作者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/cn.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