Spring Boot服务无需落地磁盘返回文件的规范实现方法

你当前通过返回ByteArray避免了磁盘写入，但无法控制客户端下载的文件名和扩展名，核心原因是没有在响应头中告知客户端文件的元信息。下面是两种惯用的无磁盘实现方案，都能直接控制文件名：

方案一：返回ResponseEntity<ByteArray>（适合小文件） #

通过ResponseEntity封装字节数据和响应头，明确指定Content-Disposition和Content-Type，让客户端识别文件名和文件类型。

修改后的控制器代码：

@RestController
@RequestMapping("/dataComparison")
@Profile("!prod")
class ComparisonReportController(
    private val reportService: ReportMappingService,
) {
    /*
    解析CSV文件出错时，请确保文件以UTF-8编码保存
    */

    @PostMapping(
        "/requestReport",
        consumes = [MULTIPART_FORM_DATA_VALUE],
        produces = [MediaType.APPLICATION_OCTET_STREAM_VALUE]
    )
    @ResponseStatus(HttpStatus.OK)
    fun getReport(
        @RequestParam("file", required = true) file: MultipartFile,
        @RequestParam(
            "fromDate",
            required = true
        ) @DateTimeFormat(iso = DateTimeFormat.ISO.DATE) fromDate: LocalDate,
        @RequestParam("numDays", required = true) numDays: Long,
        @RequestParam("reportType", required = true) reportType: ReportType,
    ): ResponseEntity<ByteArray> {
        return withLoggingContext(getItemControllerTags()) {
            log.info { "收到大小为[${file.size}b]的CSV文件，开始解析..." }

            if (numDays <= 0) {
                throw IllegalArgumentException("报告周期不能为$numDays天")
            }
            if (file.isEmpty) {
                throw IllegalArgumentException("无法使用空文件${file.name}执行数据对比")
            }

            val config = createReportConfig(file, fromDate, numDays, reportType)
            val report = reportService.getReport(config)
            val lines = report.render()
            val bytes = ReserveGroupReport.getCsvBytes(lines)

            // 构建响应头，指定文件名和文件类型
            val headers = HttpHeaders()
            // attachment表示触发下载，filename指定文件名（可根据reportType动态生成）
            headers.add(HttpHeaders.CONTENT_DISPOSITION, "attachment; filename=数据对比报告_${reportType.name.lowercase()}.csv")
            // 明确设置CSV的Content-Type，让客户端更准确识别
            headers.contentType = MediaType.parseMediaType("text/csv; charset=UTF-8")

            ResponseEntity.ok()
                .headers(headers)
                .body(bytes)
        }
    }
}

关键说明： #

• Content-Disposition头：attachment会触发浏览器下载行为，filename字段指定客户端保存的文件名（可根据业务动态生成，比如结合reportType或日期）；如果希望在浏览器直接打开文件，可替换为inline; filename=xxx.csv。
• Content-Type设置为text/csv; charset=UTF-8，比通用的APPLICATION_OCTET_STREAM更精准，避免客户端识别错误。

方案二：使用StreamingResponseBody（适合大文件） #

如果生成的CSV文件较大，直接返回ByteArray会占用较多内存，可使用StreamingResponseBody流式输出，全程无需将文件写入磁盘，也能控制文件名：

@PostMapping(
    "/requestReport",
    consumes = [MULTIPART_FORM_DATA_VALUE],
    produces = [MediaType.TEXT_CSV_VALUE]
)
fun getReportStream(
    // 参数同前
): ResponseEntity<StreamingResponseBody> {
    return withLoggingContext(getItemControllerTags()) {
        // 前置校验逻辑同前

        val config = createReportConfig(file, fromDate, numDays, reportType)
        val report = reportService.getReport(config)
        val lines = report.render()

        val headers = HttpHeaders()
        headers.add(HttpHeaders.CONTENT_DISPOSITION, "attachment; filename=数据对比报告_${reportType.name.lowercase()}.csv")
        headers.contentType = MediaType.parseMediaType("text/csv; charset=UTF-8")

        val responseBody = StreamingResponseBody { outputStream ->
            // 直接将CSV内容流式写入响应输出流，无需全量加载到内存
            lines.forEach { line ->
                outputStream.write("$line\n".toByteArray(Charsets.UTF_8))
            }
            outputStream.flush()
        }

        ResponseEntity.ok()
            .headers(headers)
            .body(responseBody)
    }
}

关键说明： #

• 流式输出不会将整个文件加载到内存，适合大体积CSV生成场景。
• 同样通过Content-Disposition控制文件名，客户端行为和方案一一致。

内容的提问来源于stack exchange，提问作者Eogcloud