java - Java InputStream(或 CsvMapper)中的总行数

Question

如何在不循环计算行数的情况下从 InputStream 或 CsvMapper 获取行数？

下面我有一个从 CSV 文件创建的 InputStream。

InputStream content = (... from a resource ...);
CsvMapper mapper = new CsvMapper();
mapper.enable(CsvParser.Feature.WRAP_AS_ARRAY);
MappingIterator<Object[]> it = mapper
        .reader(Object[].class)
        .readValues(content);

是否可以做类似的事情

int totalRows = mapper.getTotalRows();

我想在循环中使用这个数字来更新进度。

while (it.hasNextValue()){
    //do stuff here

    updateProgressHere(currentRow, totalRows);
}

显然，我可以遍历并计算一次。然后再次循环处理，同时更新进度。这是低效且缓慢的，因为其中一些 InputStreams 很大。

Answer 1

除非您提前知道行数，否则不循环是不可能的。您必须完整地读取该文件才能知道其中有多少行，InputStream 和CsvMapper 都没有提前读取的方法并为您抽象(它们都是面向流的接口(interface))。

ObjectReader 可以操作的接口(interface)都不支持查询底层文件大小(如果它是一个文件)或到目前为止读取的字节数。

一个可能的选择是创建您自己的自定义 InputStream，它还提供获取总大小和到目前为止读取的字节数的方法，例如如果它正在从文件中读取，它可以公开底层 File.length() 并跟踪读取的字节数。这可能不完全准确，特别是如果 jackson 在前面缓冲得很远，但它至少可以让你得到一些东西。