使用ByteArrayOutputStream构建CSV：数据流方式及相关技术疑问

云原生阿岚

2026-6-14

两段Kotlin代码

第一段（Controller接口）

@RequestMapping("/test")
fun getData(): ResponseEntity<ByteArray> {
    val items = repository.getItems()
    val outputStream = ByteArrayOutputStream()
    
    myService.writeData(items, outputStream)
    val byteArrayData = outputStream.toByteArray()
    
    outputStream.close() // ---------> 是否也需要在此处关闭？

    ResponseEntity.ok()
        .contentType(MediaType.APPLICATION_OCTET_STREAM)
        .header(
            HttpHeaders.CONTENT_DISPOSITION,
            "attachment; filename=file.csv"
        )
       .body(byteArrayData)
}

第二段（Service写CSV逻辑）

fun writeData(items: List<Item>, outputStream: ByteArrayOutputStream) {
    CSVPrinter(
        outputStream.writer(),
        CSVFormat.DEFAULT
    ).use { csvPrinter ->
        items.forEachIndexed { index, it ->
            csvPrinter.printRecord(…)
        }
    }
}

技术疑问解答

1. 在CSVPrinter上使用.use函数时，是否还需要关闭outputStream？

.use会自动关闭它包裹的CSVPrinter实例，但不会关闭传入的outputStream。IO流的常规设计是“谁创建谁关闭”，CSVPrinter的close()方法仅会关闭自身创建的Writer，不会处理外部传入的底层流。不过这里用的是ByteArrayOutputStream——它是基于内存的流，不占用磁盘、网络这类外部资源，所以即使不调用close()也不会有资源泄漏问题。如果要保持代码规范，调用close()也无妨，或者把ByteArrayOutputStream也用.use包裹起来。

2. 何时调用outputStream.toByteArray？在流关闭后调用是否可行？

ByteArrayOutputStream的toByteArray()是把内部缓存的字节数组拷贝一份返回，无论流是否关闭都可以调用。关闭流只是标记它不可再写入，不会清空或销毁已缓存的内存数据。不过建议在关闭流之前调用toByteArray()，更符合“先读取数据、再关闭资源”的常规流操作逻辑。

3. 此前用StringBuilder构建CSV出现OutOfMemory错误，切换为数据流方式后堆内存优化不明显，问题出在哪？

核心问题是当前写法仍然是把所有数据加载到内存中：

repository.getItems()一次性将所有Item查询出来存入内存列表，已经占用大量堆内存；
ByteArrayOutputStream会把整个CSV内容缓存到内存字节数组，调用toByteArray()还会再拷贝一份，相当于内存里存了两份完整的CSV数据。

要真正优化内存，需要做两点调整：

避免一次性加载所有Item：改用分页查询或流式查询（比如JPA的Stream<Item>），实现边查边写；
不用ByteArrayOutputStream缓存内容：直接把响应的输出流（HttpServletResponse的OutputStream）传给writeData，边写边把数据输出到客户端，内存中只会保留当前处理的少量数据，不会缓存整个CSV文件。

示例优化代码：

@RequestMapping("/test")
fun getData(response: HttpServletResponse) {
    response.contentType = MediaType.APPLICATION_OCTET_STREAM_VALUE
    response.setHeader(HttpHeaders.CONTENT_DISPOSITION, "attachment; filename=file.csv")
    
    repository.getItemsStream().use { itemsStream -> // 假设返回流式查询结果
        CSVPrinter(response.writer, CSVFormat.DEFAULT).use { csvPrinter ->
            itemsStream.forEach { item ->
                csvPrinter.printRecord(…)
            }
        }
    }
}

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