如何将应用程序设置为只返回ResponseStatusException的消息？

Question

默认情况下，spring boot不会为任何异常返回消息，包括ResponseStatusException，这意味着下面关于Bar的消息不会返回给客户端：

    @GetMapping("/foo")
    @ResponseBody
    public Foo getFoo(@RequestParam(name = "bar", defaultValue = "0") int bar) {
        if (bar <= 0) {
            throw new ResponseStatusException(HttpStatus.BAD_REQUEST, "bar must always be positive");
        }
        return example.getFoo(bar);
    }

这可以通过在application.properties中设置server.error.include-message=allays来更改，但是这会导致所有异常消息返回到客户端，包括以下消息：

    @GetMapping("/baz")
    @ResponseBody
    public Baz getBaz() {
        if (!security.checkSecurity()) {
            throw new RuntimeException("Security breach! Hope no one finds out!");
        }
        return example.getBaz();
    }

我知道这是一个微不足道的示例，解决方案只是“不要从控制器抛出服务器异常”，但异常实际上可能来自应用程序中深埋的其他代码，甚至可能是NullPointerException或其他任何代码。

如何使应用程序只显示来自ResponseStatusException的消息，而不显示其他类型的异常？（我想除了在每个控制器方法中添加try-catch子句之外。）

Answer 1

您可以向任何控制器添加额外的(@ExceptionHandler)方法，专门处理同一控制器中的请求处理(@RequestMapping)方法引发的异常。这些方法可以：

• 处理没有@ResponseStatus批注的异常（通常是您没有编写的预定义异常）
• 将用户重定向到专用错误视图
• 生成完全自定义的错误响应

控制器建议允许您使用完全相同的异常处理技术，但将它们应用于整个应用程序，而不仅仅是单个控制器。您可以将它们看作是一个注释驱动的拦截器。

任何带有@ControllerAdvice注释的类都将成为controller-advice,并且支持三种类型的方法：

• 用@ExceptionHandler注释的异常处理方法。
• 用@modelAttribute注释的模型增强方法（用于向模型添加附加数据）。请注意，这些属性对异常处理视图不可用。
• 用@initbinder注释的绑定器初始化方法（用于配置表单处理）。

解决方案：

@ControllerAdvice
public class RestControllerAdvice {

  @ExceptionHandler(ResponseStatusException.class)
  public ResponseEntity<String> handleStatusException(ResponseStatusException exception) {
    return new ResponseEntity<>(exception.getMessage(), HttpStatus.BAD_REQUEST);
  }

  @ExceptionHandler(Exception.class)
  public ResponseEntity<String> handleException(Exception exception) {
    return new ResponseEntity<>("Exception", HttpStatus.INTERNAL_SERVER_ERROR);
  }
}