Spring MVC + StreamingResponseBody - 异常导致HTTP OK

Question

我在 Spring MVC 控制器中遇到了一个关于 StreamingReponseBody 的奇怪问题。简单的控制器代码：

    @RestController
    public class ServerController {

    @GetMapping("test")
    StreamingResponseBody test(HttpServletResponse response) {
        response.setContentType("text/csv");
        response.setHeader(HttpHeaders.CONTENT_DISPOSITION, "attachment; filename=\"test.csv\"");


        return outputStream -> {
            for (int i = 0; i < 5000; i++) {
                outputStream.write("Hello".getBytes());
                outputStream.write("\n".getBytes());
            }

            try {
                Thread.sleep(2000L);
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
            if (true) {
                throw new RuntimeException("wrong");
            }
            for (int i = 0; i < 5000; i++) {
                outputStream.write("Hello2".getBytes());
                outputStream.write("\n".getBytes());
            }
        };
    }
}

加上一个控制器建议：

@ControllerAdvice
public class CustomControllerAdvice {

    @ExceptionHandler(Exception.class)
    public ResponseEntity<String> handleInternalServerError(Exception exception) {
        System.out.println("Unexpected error " + exception);
        return new ResponseEntity<>(exception.getMessage(), HttpStatus.INTERNAL_SERVER_ERROR);
    }

}

当我删除 throw 子句时，它就像一个魅力。当我在浏览器中输入 URL */test 时，我可以看到我可以下载文件 test.csv。它被正确传输了一段时间，我得到一个 200 HTTP 代码，我可以查看该文件。

但是抛出异常会导致代码做一些奇怪的事情。我没有收到 500 HTTP 代码，而是 200 HTTP 代码。在流的末尾，我得到一个“错误”的文本（异常文本 o_O）......这里发生了什么？为什么我没有收到 500 HTTP 代码。端点发送的数据不是完整的，但它说尽管有例外情况都可以。或者我应该使用其他一些流媒体机制，因为这个机制有问题？

感谢您的帮助！

Answer 1

设置一个 HTTP 预告片告诉客户端一切正常！或者如果你可以切换到 gRPC - gRPC 对流的错误处理感觉更自然，客户端连接并为 onError、onNext 和 onComplete 事件提供回调。

Answer 2

纠正我的理解。您开始使用流发送数据，所以您（我猜）不能在某个时刻说“等待，删除我已经发送的数据并更改 http 状态 [可能在开始发送一些数据之前设置为 200]”。因此，当发生异常时，Spring 会重用 OutputStream 并从以下位置放入数据：

return new ResponseEntity<>(exception.getMessage()

进入同一个输出流。

一个建议可能是包装可能在 try-catch 中引发异常的代码并使用 IOUtils.closeQuietly 但效果是您将获得 5000 倍的 Hello 和流结束（所以仍然是 200 http 状态，您发送的数据正确的是在客户端，但最后没有wrong）

例子：

if (true) {
    try {
        throw new RuntimeException("wrong");
     } catch (RuntimeException e) {
        IOUtils.closeQuietly(outputStream); #apache
        throw e;
     }
}