使用OpenCSV只能部分解析CSV文件

Question

我有一个CSV文件，我想用开放CSV的csv阅读器在Java中解析。

为此，我创建了一个Bean对象，信息映射到该对象。我的有点长，所以这是我从教程中得到的一个例子：

package net.viralpatel.java;

public class Country {
    private String countryName;
    private String capital;

    public String getCountryName() {
    return countryName;
    }

    public void setCountryName(String countryName) {
        this.countryName = countryName;
    }

    public String getCapital() {
        return capital;
    }

    public void setCapital(String capital) {
        this.capital = capital;
    }
}

我用来解析CSV文件并将信息映射到bean的代码类似于以下代码：

ColumnPositionMappingStrategy strat = new ColumnPositionMappingStrategy();
strat.setType(Country.class);
String[] columns = new String[] {"countryName", "capital"};
strat.setColumnMapping(columns);

CsvToBean csv = new CsvToBean();

String csvFilename = "C:\\sample.csv";
CSVReader csvReader = new CSVReader(new FileReader(csvFilename));

List list = csv.parse(strat, csvReader);

问题在于，我的CSV不仅包含原始数据，还包含列标题和其他数据。对于专栏标题，我只从某行读取我的文件，解决了这个问题：

CSVReader csvReader = new CSVReader(new FileReader(csvFilename), ';', '\'', 1);

（1 是开始读取的行）

其他数据主要是文件末尾（例如）整数列中的字符串。

例如，我有一个包含整数信息的“最大速度”列，紧挨着一个包含整数信息的“距离”列。但是在“距离”栏的末尾是总距离，所以字符串“总:”就在它旁边的“最大速度”栏中。

我该怎么做才能确保读者忽略最后几行，只读取上面的原始信息？

PS:我看的CSV文件长度不一。所以说“在X行之后停止阅读”不会起作用。另一方面,“附录”行总是相同的。所以说“在文件结束前停止读两行”应该可以。

非常感谢您的帮助。

Answer 1

如果您使用的是较新版本的opencsv，则将CsvToBeanFilter注入到您的CSVtoBean类中。opencsv javadoc提供了一个如何创建过滤器的优秀示例。对于您的示例，您只需创建一个过滤器，如果最大速度为null、空或“total:”，则其allowLine方法将返回false：

Answer 2

您可以始终下降到较低级别并在将其映射到bean之前检查原始字符串数组，如下所示：

ColumnPositionMappingStrategy<Country> strat = new ColumnPositionMappingStrategy<Country>();
    strat.setType(Country.class);
    String[] columns = new String[] {"countryName", "capital"};
    strat.setColumnMapping(columns);

    PublicProcessLineCsvToBean<Country> csv = new PublicProcessLineCsvToBean<Country>();

    String csvFilename = "C:\\sample.csv";
    CSVReader csvReader = new CSVReader(new FileReader(csvFilename));
    List<Country> list = new ArrayList<Country>();

    String [] row = csvReader.readNext(); //skip header
    if(row == null) throw new RuntimeException("File is empty");
    row = csvReader.readNext();
    String [] nextRow = csvReader.readNext();
    while(row != null) {
        if(nextRow == null) break; //check what 'row' is last
        if("Total:".equalsIgnoreCase(row[1])) break; //check column for special strings

        list.add(csv.processLine(strat, row));

        row = nextRow;
        nextRow = csvReader.readNext();
    }

并使工艺线公开：

public static class PublicProcessLineCsvToBean<T> extends CsvToBean<T> {

        @Override
        public T processLine(MappingStrategy<T> mapper, String[] line) throws IllegalAccessException, InvocationTargetException, InstantiationException, IntrospectionException {
            return super.processLine(mapper, line);
        }
    }