fix : 북마크 예외 처리

tripzzle · Nov 22, 2023 · ac43d11 · ac43d11
1 parent 89fcc3b
commit ac43d11
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/main/java/com/tgd/trip/schedule/service/ScheduleService.java b/src/main/java/com/tgd/trip/schedule/service/ScheduleService.java
@@ -156,6 +156,9 @@ public void createPhoto(Long scheduleId, Long dayId, List<MultipartFile> files)
     public void createBookmark(Long scheduleId, SecurityUser securityUser) {
         User findUser = userService.getVerifyUser(securityUser.getMember().getUserId());
         Schedule findSchedule = getSchedule(scheduleId);
+        if (!findSchedule.getUser().getEmail().equals(findUser.getEmail())) {
+            throw new CustomException(ErrorCode.DIFFERENT_USER);
+        }
         ScheduleBookmark scheduleBookmark = new ScheduleBookmark(findSchedule);
         findUser.addScheduleBookmark(scheduleBookmark);
         scheduleBookmarkRepository.save(scheduleBookmark);
@@ -165,6 +168,9 @@ public void createBookmark(Long scheduleId, SecurityUser securityUser) {
     public void deleteBookmark(Long scheduleId, SecurityUser securityUser) {
         User findUser = userService.getVerifyUser(securityUser.getMember().getUserId());
         Schedule findSchedule = getSchedule(scheduleId);
+        if (!findSchedule.getUser().getEmail().equals(findUser.getEmail())) {
+            throw new CustomException(ErrorCode.DIFFERENT_USER);
+        }
         scheduleBookmarkRepository.deleteByUserAndSchedule(findUser, findSchedule);
     }