使用特定的POJO字段映射csv文件的特定列

内容来源于 Stack Overflow,并遵循CC BY-SA 3.0许可协议进行翻译与使用

  • 回答 (1)
  • 关注 (0)
  • 查看 (22)

我必须根据具有特定POJO属性的索引映射特定的CSV列。映射将基于一个json文件,该文件将包含columnIndex和属性名称,这意味着对于来自csv文件的特定columnIndex,您必须映射Pojo类中的特定属性。下面是一个json文件示例,它显示了具有Pojo属性的列映射策略。

[{ “索引”:0 “列名”: “日期”},{ “索引”:1, “COLUMNNAME”: “deviceAddress”},{ “索引”:7, “COLUMNNAME”: “IPADDRESS”},{ “索引”:3 “列名”: “的userName”},{ “索引”:10 “列名”: “基团”},{ “索引”:5 “列名”: “eventCategoryName”},{“索引“:6,” COLUMNNAME “:” 消息“}]

我已经尝试过使用OpenCSV库,但我遇到的挑战是我无法用它读取部分列。如上面的json,你可以看到我们正在跳过索引2和4来读取CSV文件。下面是openCSV文件的代码。

public static List<BaseDataModel> readCSVFile(String filePath,List<String> columnListBasedOnIndex) {
        List<BaseDataModel> csvDataModels = null;
        File myFile = new File(filePath);
        try (FileInputStream fis = new FileInputStream(myFile)) {
            final ColumnPositionMappingStrategy<BaseDataModel> strategy = new ColumnPositionMappingStrategy<BaseDataModel>();
            strategy.setType(BaseDataModel.class);


            strategy.setColumnMapping(columnListBasedOnIndex.toArray(new String[0]));

            final CsvToBeanBuilder<BaseDataModel> beanBuilder = new CsvToBeanBuilder<>(new InputStreamReader(fis));
            beanBuilder.withMappingStrategy(strategy);

            csvDataModels = beanBuilder.build().parse();

        } catch (Exception e) {
            e.printStackTrace();
        }
}


List<ColumnIndexMapping> columnIndexMappingList = dataSourceModel.getColumnMappingStrategy();
                    List<String> columnNameList = columnIndexMappingList.stream().map(ColumnIndexMapping::getColumnname)
                            .collect(Collectors.toList());

List<BaseDataModel> DataModels = Utility
                                    .readCSVFile(file.getAbsolutePath() + File.separator + fileName, columnNameList);

我也尝试过univocity但是使用这个库我怎么能用特定属性映射csv。以下是代码 -

CsvParserSettings settings = new CsvParserSettings();
        settings.detectFormatAutomatically(); //detects the format 
        settings.getFormat().setLineSeparator("\n");
        //extracts the headers from the input
        settings.setHeaderExtractionEnabled(true);
        settings.selectIndexes(0, 2); //rows will contain only values of columns at position 0 and 2
        CsvRoutines routines = new CsvRoutines(settings); // Can also use TSV and Fixed-width routines
        routines.parseAll(BaseDataModel.class, new File("/path/to/your.csv"));


        List<String[]> rows = new CsvParser(settings).parseAll(new File("/path/to/your.csv"), "UTF-8");

请看看在这种情况下有人可以帮助我。

提问于
用户回答回答于

这里是univocity-parsers的作者。您可以在代码而不是注释中定义类属性的映射。像这样的东西:

public class BaseDataModel {
    private String a;
    private int b;
    private String c;
    private Date d;
}

然后在您的代码上,将属性映射到您需要的任何列名称:

ColumnMapper mapper = routines.getColumnMapper();
mapper.attributeToColumnName("a", "col1");
mapper.attributeToColumnName("b", "col2");
mapper.attributeToColumnName("c", "col3");
mapper.attributeToColumnName("d", "col4");

您还可以使用mapper.attributeToIndex("d", 3);将属性映射到给定的列索引。

希望这可以帮助。

扫码关注云+社区

领取腾讯云代金券