Fix: 메뉴 추천 api 수정 (#41)

Digital-Hana-Starbucks · May 8, 2024 · 03e0aa7 · 03e0aa7
1 parent 89c5bab
commit 03e0aa7
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/src/main/java/com/hanaro/starbucks/service/MenuService.java b/src/main/java/com/hanaro/starbucks/service/MenuService.java
@@ -88,8 +88,15 @@ public List<MenuResDto> getRecommendationList(){
 
         Random random = new Random();
 
-        while (idxSet.size() < 9) {
+        while (idxSet.size() < 8) {
             int idx = random.nextInt(menus.size());
+
+            Menu menu = menus.get(idx);
+
+            if(idxSet.contains(idx)) continue;
+
+            if(menu.getCategory().getCategoryIdx() == 4 || menu.getCategory().getCategoryIdx() == 5) continue;
+
             idxSet.add(idx);
             menuList.add(menus.get(idx));
         }