feat : 3차 세미나 실습 과제_CustomValidateException GlobalExceptionHandler에 추가 …

…#7
NOW-SOPT-SERVER · May 1, 2024 · beda91d · beda91d
1 parent e1438ea
commit beda91d
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/week02/seminar/demo/src/main/java/com/example/demo/common/GlobalExceptionHandler.java b/week02/seminar/demo/src/main/java/com/example/demo/common/GlobalExceptionHandler.java
@@ -1,9 +1,8 @@
 package com.example.demo.common;
 
-import com.example.demo.common.dto.ErrorMessage;
 import com.example.demo.common.dto.ErrorResponse;
+import com.example.demo.exception.CustomValidateException;
 import com.example.demo.exception.NotFoundException;
-import jakarta.persistence.EntityNotFoundException;
 import java.util.Objects;
 import org.springframework.http.HttpStatus;
 import org.springframework.http.ResponseEntity;
@@ -22,4 +21,9 @@ protected ResponseEntity<ErrorResponse> handleNotFoundException(NotFoundExceptio
     return ResponseEntity.status(HttpStatus.NOT_FOUND).body(ErrorResponse.of(e.getErrorMessage()));
   }
 
+  @ExceptionHandler(CustomValidateException.class)
+  protected ResponseEntity<ErrorResponse> handleNotFoundException(CustomValidateException e) {
+    return ResponseEntity.status(HttpStatus.NOT_FOUND).body(ErrorResponse.of(e.getErrorMessage()));
+  }
+
 }