[問題] 多對多關係如何優雅合併?

看板java作者 (柚子)時間2年前 (2022/04/09 15:25), 2年前編輯推噓1(100)
留言1則, 1人參與, 2年前最新討論串1/2 (看更多)
各位好,想請教是否有更好的寫法,主要針對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://www.ptt.cc/bbs/java/M.1649489153.A.CFE.html

04/09 22:00, 2年前 , 1F
用兩個Map<Integer, Set<Integer>> 維護多對多關係
04/09 22:00, 1F
感謝,才發現map有一個computeIfPresent方法特別適合這種狀況 ※ 編輯: sssyoyo (220.134.136.97 臺灣), 04/09/2022 23:57:29
文章代碼(AID): #1YKJK1p- (java)
文章代碼(AID): #1YKJK1p- (java)