[问题] 多对多关系如何优雅合并?

楼主: sssyoyo (柚子)   2022-04-09 15:25:35
各位好,想请教是否有更好的写法,主要针对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<>好像也挺浪费资源的
想询问是否有更好的写法,还请不吝赐教,感谢!
作者: b9502056 (壞魯)   2022-04-09 22:00:00
用两个Map<Integer, Set<Integer>> 维护多对多关系

Links booklink

Contact Us: admin [ a t ] ucptt.com