diff --git a/src/main/java/ecureuill/milhasapi/infra/expection/ControllerAdvice.java b/src/main/java/ecureuill/milhasapi/infra/expection/ControllerAdvice.java new file mode 100644 index 0000000..9693d71 --- /dev/null +++ b/src/main/java/ecureuill/milhasapi/infra/expection/ControllerAdvice.java @@ -0,0 +1,27 @@ +package ecureuill.milhasapi.infra.expection; + +import org.springframework.http.ResponseEntity; +import org.springframework.web.bind.annotation.ExceptionHandler; +import org.springframework.web.bind.annotation.RestControllerAdvice; +import org.springframework.web.server.ResponseStatusException; + +import jakarta.persistence.EntityNotFoundException; + +@RestControllerAdvice +public class ControllerAdvice { + + @ExceptionHandler(value = EntityNotFoundException.class) + public ResponseEntity handleEntityNotFoundException(EntityNotFoundException e) { + return ResponseEntity.notFound().build(); + } + + @ExceptionHandler(value = ResponseStatusException.class) + public ResponseEntity handleResponseStatusException(ResponseStatusException ex) { + return ResponseEntity.status(ex.getStatusCode()).body(ex.getReason()); + } + + @ExceptionHandler(value = Exception.class) + public ResponseEntity handleException(Exception e) { + return ResponseEntity.badRequest().body(e.getMessage()); + } +}