Question

我有很多具有不同列标题的CSV文件。目前我正在阅读那些csv文件，并根据列标题将它们映射到不同的POJO类。因此，一些CSV文件有大约100个列标题，这使得创建POJO类变得很困难。

那么有什么技术我可以使用单个pojo，所以当读取这些csv文件时可以映射到单个POJO类或者我应该逐行读取CSV文件并相应地解析或者我应该在运行时创建POJO（ javaassist）？

Answer 1

如果我正确理解您的问题，您可以使用uniVocity-parsers来处理此问题并在地图中获取数据：

//First create a configuration object - there are many options 
//available and the tutorial has a lot of examples
CsvParserSettings settings = new CsvParserSettings();
settings.setHeaderExtractionEnabled(true);

CsvParser parser = new CsvParser(settings);
parser.beginParsing(new File("/path/to/your.csv"));

// you can also apply some transformations:
// NULL year should become 0000
parser.getRecordMetadata().setDefaultValueOfColumns("0000", "Year");

// decimal separator in prices will be replaced by comma
parser.getRecordMetadata().convertFields(Conversions.replace("\\.00", ",00")).set("Price");

Record record;
while ((record = parser.parseNextRecord()) != null) {
     Map<String, String> map = record.toFieldMap(/*you can pass a list of column names of interest here*/);
     //for performance, you can also reuse the map and call record.fillFieldMap(map);
}

或者您甚至可以解析文件并在一个步骤中获取不同类型的bean。以下是您的工作方式：

CsvParserSettings settings = new CsvParserSettings();

//Create a row processor to process input rows. In this case we want
//multiple instances of different classes:
MultiBeanListProcessor processor = new MultiBeanListProcessor(TestBean.class, AmountBean.class, QuantityBean.class);

// we also need to grab the headers from our input file
settings.setHeaderExtractionEnabled(true);

// configure the parser to use the MultiBeanProcessor
settings.setRowProcessor(processor);

// create the parser and run
CsvParser parser = new CsvParser(settings);

parser.parse(new File("/path/to/your.csv"));

// get the beans:
List<TestBean> testBeans = processor.getBeans(TestBean.class);
List<AmountBean> amountBeans = processor.getBeans(AmountBean.class);
List<QuantityBean> quantityBeans = processor.getBeans(QuantityBean.class);

查看示例here和here

如果您的数据太大并且您无法将所有内容保存在内存中，则可以使用MultiBeanRowProcessor来逐行传输输入。方法rowProcessed(Map<Class<?>, Object> row, ParsingContext context)将为您提供为当前行中的每个类创建的实例的映射。在方法内部，只需调用：

AmountBean amountBean = (AmountBean) row.get(AmountBean.class);
QuantityBean quantityBean = (QuantityBean) row.get(QuantityBean.class);
...

//perform something with the instances parsed in a row.

希望这有帮助。

免责声明：我是这个图书馆的作者。它是开源和免费的（Apache 2.0许可证）

Answer 2

对我来说，在这种情况下创建一个POJO类并不是一个好主意。因为列数和文件数都不变。因此，最好使用更动态的东西，而不必为了支持更多列或文件而在很大程度上更改代码。

我会为给定的csv文件寻找List（或Map）Map List<Map<>>。每个map代表csv文件中的一行，其中key为列名。

您可以轻松地将其扩展为多个csv文件。

将多个CSV映射到单个POJO

2 个答案: