各位好，想請教是否有更好的寫法，主要針對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<>好像也挺浪費資源的
想詢問是否有更好的寫法，還請不吝賜教，感謝!