refactor: modifyDiary 예외처리 플로우 추가 #24

My-Music-Note · Dec 8, 2024 · 603548b · 603548b
1 parent 16086f9
commit 603548b
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/main/java/my/music/note/back/service/DiaryService.java b/src/main/java/my/music/note/back/service/DiaryService.java
@@ -41,12 +41,11 @@ public void deleteDiary(Long diaryId, Long userId) {
 
     public void modifyDiary(Long diaryId, Long userId, DiaryModifyRequest diaryModifyRequest) {
 
-
         if (!diaryRepository.existsByIdAndUserId(userId, diaryId)) {
             throw new IllegalArgumentException();
         }
 
-        Diary diary = diaryRepository.findById(diaryId).get();
+        Diary diary = diaryRepository.findById(diaryId).orElseThrow(IllegalArgumentException::new);
         diary.modifyDiaryContent(diaryModifyRequest);
     }