StreamingResponseBody中的异常处理

时间:2019-06-18 14:54:52

标签: java spring spring-mvc asynchronous exception

我正在尝试捕获在我的StreamingResponseBody实现中引发的异常,我可以看到该异常在类内部引发,但是该引发的异常对于方法主体或我的Controller Advice不可见。因此,我的任何处理似乎都不起作用,只是想知道在这种情况下哪种是处理异常的正确方法。

@GetMapping(path = "/test", produces = "application/json")
public StreamingResponseBody test(@RequestParam(value = "var1") final String test)
        throws IOException{

    return new StreamingResponseBody() {

        @Override
        public void writeTo(final OutputStream outputStream) throws IOException{
            try {
                  // Some operations..
            } catch (final SomeCustomException e) {
                throw new IOException(e);
            }
        }
    };
}

我希望我的ControllerAdvice返回Http状态为500的ResponseEntity。

1 个答案:

答案 0 :(得分:0)

我发现在Web环境中处理错误/异常的最好方法是使用禁用的堆栈跟踪创建自定义异常,并使用@ControllerAdvice处理它。

import lombok.Getter;
import org.springframework.http.HttpStatus;

public class MyException extends RuntimeException {

    @Getter private HttpStatus httpStatus;

    public MyException(String message) {
        this(message, HttpStatus.INTERNAL_SERVER_ERROR);
    }

    public MyException(String message, HttpStatus status) {
        super(message, null, false, false);
        this.httpStatus = status;
    }
}

然后像下面这样在@ControllerAdvice中进行处理:

@ExceptionHandler(MyException.class)
public ResponseEntity handleMyException(MyException exception) {
    return ResponseEntity.status(exception.getHttpStatus()).body(
            ErrorDTO.builder()
                    .message(exception.getMessage())
                    .description(exception.getHttpStatus().getReasonPhrase())
                    .build());
}

其中ErrorDTO只是具有两个字段的简单DTO:

@Value
@Builder
public class ErrorDTO {

    private final String message;
    private final String description;

}