浏览代码

Fix controller advisor bad request

Daniel Bohry 1 年之前
父节点
当前提交
92b94ac0ae
共有 1 个文件被更改,包括 10 次插入0 次删除
  1. 10 0
      src/main/java/com/danielbohry/authservice/api/ControllerAdvisor.java

+ 10 - 0
src/main/java/com/danielbohry/authservice/api/ControllerAdvisor.java

@@ -1,5 +1,6 @@
 package com.danielbohry.authservice.api;
 
+import com.danielbohry.authservice.exceptions.BadRequestException;
 import com.danielbohry.authservice.exceptions.NotFoundException;
 import io.jsonwebtoken.ExpiredJwtException;
 import org.springframework.http.HttpStatus;
@@ -34,6 +35,15 @@ public class ControllerAdvisor extends ResponseEntityExceptionHandler {
         return new ResponseEntity<>(body, HttpStatus.BAD_REQUEST);
     }
 
+    @ExceptionHandler(BadRequestException.class)
+    protected ResponseEntity<Object> handleBadRequestException(RuntimeException ex) {
+        Map<String, Object> body = new LinkedHashMap<>();
+        body.put("timestamp", LocalDateTime.now());
+        body.put("message", ex.getMessage());
+
+        return new ResponseEntity<>(body, HttpStatus.BAD_REQUEST);
+    }
+
     @ExceptionHandler(ExpiredJwtException.class)
     protected ResponseEntity<Object> handleExpiredJwtException(RuntimeException ex) {
         Map<String, Object> body = new LinkedHashMap<>();