Apache Commons CSV 不会忽略缺失的列

Question

使用 Apache Commons CSV 进行解析，但不会忽略缺失的列并抛出异常。

使用此示例数据：

name age
Ali 35
John 25
Vahid 75

下面的代码record.get(DataColumns.surname)抛出java.lang.IllegalArgumentException: Mapping for surname not found, expected one of [name, surname, age] 。 我需要它返回空值、可选值或默认值。 有什么选择吗？ 我知道record.toMap().get(DataColumns.surname.name())是可能的，但它的性能不会很好：

...
enum DataColumns { name, surname, age }
...
Reader in = new BufferedReader(new FileReader(fileName));

try (CSVParser records = CSVFormat.TDF
                .withDelimiter(' ')
                .withIgnoreSurroundingSpaces()
                .withAllowDuplicateHeaderNames(false)
                .withIgnoreHeaderCase()
                .withTrim()
                .withHeader(DataColumns.class)
                .withFirstRecordAsHeader()
                .withSkipHeaderRecord()
                .withAllowMissingColumnNames(false)
                .withIgnoreEmptyLines()
                .parse(in)) {

   for (CSVRecord record : records) {
       String name = record.get(DataColumns.name);
       String surname = record.get(DataColumns.surname);
       Short age = Short.valueOf(record.get(DataColumns.age)); 
   }
}

...

Answer 1

您可以尝试使用record.isMapped(columnName)来检查该列是否存在，记录到一个变量中，这样您就不必再次检查每一行。

另一种选择是使用records.getHeaderNames()并在循环之前将其存储到变量中一次，甚至可能使用Set<String>来提高存在性检查性能： Set<String> headerNames = new HashSet<>(records.getHeaderNames()) 。

然后，您可以通过调用headerNames.contains(columnName)在循环内使用结果变量来检查该列是否存在。

请参阅： https ://javadoc.io/doc/org.apache.commons/commons-csv/latest/org/apache/commons/csv/CSVRecord.html

Answer 2

有方法：record.get(String) 而你给了枚举。

试试 record.get(DataColumns.name.name())