本文轉自測試人社群,原文連結:https://ceshiren.com/t/topic/28202
pom檔案導入
<dependency>
<groupId>com.fasterxml.jackson.dataformat</groupId>
<artifactId>jackson-dataformat-csv</artifactId>
<version>${jackson.version}</version>
</dependency>
<dependency>
<groupId>com.fasterxml.jackson.datatype</groupId>
<artifactId>jackson-datatype-jsr310</artifactId>
<version>${jackson.version}</version>
</dependency>
帶有header的csv
orderLines.csv
item,quantity,unitPrice,orderDate
“No. 9 Sprockets”,12,1.23,2019-04-17
“Widget (10mm)”,4,3.45,2022-01-16
成員變量與header名一緻
- 實體類的成員變量
- csv對應的header名
@Data
public class OrderLine{
private String item;
private int quantity;
private BigDecimal unitPrice;
private LocalDate orderDate;
public OrderLine() {
}
public OrderLine(String item, int quantity,
BigDecimal unitPrice, LocalDate orderDate) {
this.item = item;
this.quantity = quantity;
this.unitPrice = unitPrice;
this.orderDate = orderDate;
}
}
對應測試方法
@Test
void csvReadWithHeader() throws IOException {
//帶着header讀取
CsvSchema orderLineSchema = CsvSchema.emptySchema().withHeader();
//功能上等價的便捷方法: mapper.registerModules(mapper.findModules());
//我們需要使用 findAndRegisterModules方法,以便 Jackson正确處理我們的日期
//Jackson也可以自動搜尋所有子產品,不需要我們手動注冊
csvMapper.findAndRegisterModules();
MappingIterator<OrderLine> orderLines = csvMapper.readerFor(OrderLine.class)
.with(orderLineSchema)
.readValues(new File("src/test/resources/csv/orderLines.csv"));
System.out.println(orderLines.readAll());
}
成員變量與header名不一緻
- 實體類的成員變量
- csv對應的header名
@Data
public class OrderList {
@JsonProperty("item")
private String otherItem;
@JsonProperty("quantity")
private int qua;
@JsonProperty("unitPrice")
private BigDecimal price;
@JsonProperty("orderDate")
private LocalDate date;
}
對應測試方法
@Test
void csvReadWithHeader1() throws IOException {
//帶着header讀取
CsvSchema orderLineSchema = CsvSchema.emptySchema().withHeader();
//功能上等價的便捷方法: mapper.registerModules(mapper.findModules());
//我們需要使用 findAndRegisterModules方法,以便 Jackson正确處理我們的日期
//Jackson也可以自動搜尋所有子產品,不需要我們手動注冊
csvMapper.findAndRegisterModules();
MappingIterator<OrderList> orderLines = csvMapper.readerFor(OrderList.class)
.with(orderLineSchema)
.readValues(new File("src/test/resources/csv/orderLines.csv"));
System.out.println(orderLines.readAll());
}
直接聲明類型解析
- readerForMapOf(String.class)
@Test
void csvReadWithHeaderMap() throws IOException {
//功能上等價的便捷方法: mapper.registerModules(mapper.findModules());
//我們需要使用 findAndRegisterModules方法,以便 Jackson正确處理我們的日期
//Jackson也可以自動搜尋所有子產品,不需要我們手動注冊
csvMapper.findAndRegisterModules();
//.setSkipFirstDataRow(true) 第一行header不解析
CsvSchema schema = CsvSchema.builder().setSkipFirstDataRow(true)
.addColumn("item")
.addColumn("quantity")
.addColumn("unitPrice")
.addColumn("orderDate")
.build();
MappingIterator<Object> objectMappingIterator = csvMapper.readerForMapOf(String.class)
// 注意:不需要包裝
.with(schema)
.readValues(new File("src/test/resources/csv/orderLines.csv"));
System.out.println(objectMappingIterator.readAll());
}
沒有header的csv
orderlist.csv
“No. 9 Sprockets”,12,1.23,2019-04-17
“Widget (10mm)”,4,3.45,2022-01-16
成員變量與header名一緻
- 實體類的成員變量
- csv對應的header名
@Data
public class OrderLine{
private String item;
private int quantity;
private BigDecimal unitPrice;
private LocalDate orderDate;
public OrderLine() {
}
public OrderLine(String item, int quantity,
BigDecimal unitPrice, LocalDate orderDate) {
this.item = item;
this.quantity = quantity;
this.unitPrice = unitPrice;
this.orderDate = orderDate;
}
}
對應測試方法
@Test
void csvReadOutHeader() throws IOException {
//功能上等價的便捷方法: mapper.registerModules(mapper.findModules());
//我們需要使用 findAndRegisterModules方法,以便 Jackson正确處理我們的日期
//Jackson也可以自動搜尋所有子產品,不需要我們手動注冊
csvMapper.findAndRegisterModules();
CsvSchema schema = CsvSchema.builder()
.addColumn("item")
.addColumn("quantity")
.addColumn("unitPrice")
.addColumn("orderDate")
.build();
MappingIterator<Object> objectMappingIterator = csvMapper.readerFor(OrderLine.class)
.with(schema)
.readValues(new File("src/test/resources/csv/orderlist.csv"));
System.out.println(objectMappingIterator.readAll());
}
成員變量與header名不一緻
- 實體類的成員變量
- csv對應的header名
@Data
public class OrderList {
@JsonProperty("item")
private String otherItem;
@JsonProperty("quantity")
private int qua;
@JsonProperty("unitPrice")
private BigDecimal price;
@JsonProperty("orderDate")
private LocalDate date;
}
對應測試方法
@Test
void csvReadOutHeader() throws IOException {
//功能上等價的便捷方法: mapper.registerModules(mapper.findModules());
//我們需要使用 findAndRegisterModules方法,以便 Jackson正确處理我們的日期
//Jackson也可以自動搜尋所有子產品,不需要我們手動注冊
csvMapper.findAndRegisterModules();
CsvSchema schema = CsvSchema.builder()
.addColumn("item")
.addColumn("quantity")
.addColumn("unitPrice")
.addColumn("orderDate")
.build();
MappingIterator<Object> objectMappingIterator = csvMapper.readerFor(OrderList.class)
.with(schema)
.readValues(new File("src/test/resources/csv/orderlist.csv"));
System.out.println(objectMappingIterator.readAll());
}
直接聲明類型解析
- readerForMapOf(String.class)
@Test
void csvReadOutHeaderMap() throws IOException {
//功能上等價的便捷方法: mapper.registerModules(mapper.findModules());
//我們需要使用 findAndRegisterModules方法,以便 Jackson正确處理我們的日期
//Jackson也可以自動搜尋所有子產品,不需要我們手動注冊
csvMapper.findAndRegisterModules();
CsvSchema schema = CsvSchema.builder()
.addColumn("item")
.addColumn("quantity")
.addColumn("unitPrice")
.addColumn("orderDate")
.build();
MappingIterator<Object> objectMappingIterator = csvMapper.readerForMapOf(String.class)
.with(schema)
.readValues(new File("src/test/resources/csv/orderlist.csv"));
System.out.println(objectMappingIterator.readAll());
}