[fix] 카테고리 수정 시 변경사항이 VectorDB에는 저장되지 않던 문제 수정

studio-recoding · Jul 14, 2024 · 51a4070 · 51a4070
1 parent 584a77c
commit 51a4070
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/src/main/java/Ness/Backend/domain/category/CategoryService.java b/src/main/java/Ness/Backend/domain/category/CategoryService.java
@@ -76,6 +76,22 @@ public void putUserCategory(Long memberId, PutCategoryDto putCategoryDto){
             //중복되지 않은 카테고리일 경우는 변경사항 저장 가능
             log.info(putCategoryDto.getId() + "번 카테고리 " + changeCategory.getName() + " 수정");
             changeCategory.changeCategory(putCategoryDto.getName(), putCategoryDto.getColor());
+
+            //카테고리 이름 또는 컬러 변경 사항을 VectorDB에도 저장
+            List<Schedule> changeScheduleList = scheduleRepository.findSchedulesByCategory_Id(putCategoryDto.getId());
+            changeScheduleList.forEach(schedule -> {
+                scheduleService.putAiSchedule(
+                        schedule.getInfo(),
+                        schedule.getLocation(),
+                        schedule.getPerson(),
+                        schedule.getStartTime(),
+                        schedule.getEndTime(),
+                        putCategoryDto.getName(),
+                        schedule.getCategory().getId(),
+                        putCategoryDto.getColor(),
+                        schedule.getMember().getId(),
+                        schedule.getId());
+            });
         }
         else {
             throw new DuplicateCategoryException();