在HTTP異常返回自定義消息

Question

這是我的ExceptionHandler類

@ResponseStatus(HttpStatus.NOT_FOUND) 
public class NotFoundException extends Exception { 

    private static final long serialVersionUID = -1964839652066902559L; 

    public NotFoundException(String message) { 
     super(message); 
    } 
} 

在我的服務，我有

@Service 
public class myService { 
public String getName() throws NotFoundException{ 
      logger.error("Name not found"); 
      throw new NotFoundException("Name not found"); 
     } 
} 

我得到的迴應是：

{ 
    "timestamp": 1486967853916, 
    "status": 404, 
    "error": "Not Found", 
    "exception": "com.exceptions.NotFoundException", 
    "message": "Not Found", 
    "path": "/names" 
} 

我只是想我的消息從服務中傳遞並返回異常。但是這沒有發生任何人可以告訴我該怎麼做。

Answer 1

使用@ControllerAdvice和@ExceptionHandler的組合，可能這會幫助你。

@ControllerAdvice 
public class GlobalValidationResponse extends ResponseEntityExceptionHandler { 

    @ExceptionHandler(NotFoundException.class) 
     public ResponseEntity<?> handleEntityNotFoundException(NotFoundException ex, HttpServletRequest request, HttpServletResponse response) { 

      return ResponseEntity.status(HttpStatus.NOT_FOUND).body(new ValidationError(ex.getMessage())); 

     } 
} 

的ValidationError實體

public class ValidationError { 

    //@JsonInclude(JsonInclude.Include.NON_EMPTY) 
    private Map<String, String> errors = new HashMap<>(); 

    private String errorMessage; 

    public ValidationError() { 
     super(); 
    } 

    public ValidationError(String errorMessage) { 
     this.errorMessage = errorMessage; 
    } 

    public ValidationError(Map<String, String> errors) { 
     this.errors = errors; 
    } 

    public Map<String, String> getErrors() { 
     return errors; 
    } 

    public void setErrors(Map<String, String> errors) { 
     this.errors = errors; 
    } 

    public String getErrorMessage() { 
     return errorMessage; 
    } 
}