Python教程: Springboot读取和写入CSV文件数据的最佳指南

2023-09-19 12:14:00

在 Spring Boot 中轻松读取和写入 CSV 文件数据

引言

在数据驱动的世界中，轻松从各种来源读取和写入数据变得越来越重要。Spring Boot 作为一个流行的 Java 框架，提供了开箱即用的强大功能，包括对 CSV 文件读写的支持。本教程将深入探讨如何在 Spring Boot 应用程序中高效地管理 CSV 文件数据。

CSV 文件格式：基础知识

CSV（逗号分隔值）是一种简单易用的文本文件格式，用于存储表格数据。每行通常代表一条记录，各个字段由逗号分隔。CSV 文件广泛用于数据交换和存储，因为它易于理解和处理。

Spring Boot 对 CSV 文件的支持

Spring Boot 提供了一个专门的 CSV 模块，简化了 CSV 文件读写的过程。此模块包含以下主要组件：

CsvSchema： 定义 CSV 文件的列结构和数据类型。
CsvMarshaller： 将对象转换为 CSV 格式。
CsvUnmarshaller： 将 CSV 数据解析为对象。

配置 Spring Boot CSV 支持

要在 Spring Boot 应用程序中启用 CSV 支持，请遵循以下步骤：

1. 添加 Spring Boot CSV 库依赖：

```xml
<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-csv</artifactId>
</dependency>

配置 CSV Schema 和解析器/转换器：

@SpringBootApplication
public class SpringbootCsvApplication {

    public static void main(String[] args) {
        SpringApplication.run(SpringbootCsvApplication.class, args);
    }

    @Bean
    public CsvSchema csvSchema() {
        CsvSchemaBuilder csvSchemaBuilder = new CsvSchemaBuilder();
        csvSchemaBuilder.setName("csv_schema");
        csvSchemaBuilder.addIgnoredLine(1); // 忽略第一行（标题）
        csvSchemaBuilder.addColumn("name", String.class);
        csvSchemaBuilder.addColumn("age", Integer.class);
        csvSchemaBuilder.addColumn("city", String.class);
        return csvSchemaBuilder.build();
    }

    @Bean
    public CsvMarshaller csvMarshaller() {
        CsvMarshaller csvMarshaller = new CsvMarshaller();
        csvMarshaller.setSchema(csvSchema());
        return csvMarshaller;
    }

    @Bean
    public CsvUnmarshaller csvUnmarshaller() {
        CsvUnmarshaller csvUnmarshaller = new CsvUnmarshaller();
        csvUnmarshaller.setSchema(csvSchema());
        return csvUnmarshaller;
    }
}

读取 CSV 文件

使用 CsvUnmarshaller 可以轻松地从 CSV 文件读取数据：

@RestController
@RequestMapping("/csv")
public class CsvController {

    @Autowired
    private CsvUnmarshaller csvUnmarshaller;

    @GetMapping("/read")
    public List<Person> readCsv() {
        List<Person> people = new ArrayList<>();
        try {
            InputStream inputStream = new FileInputStream("people.csv");
            people = csvUnmarshaller.unmarshal(inputStream);
        } catch (IOException e) {
            e.printStackTrace();
        }
        return people;
    }
}

写入 CSV 文件

使用 CsvMarshaller 可以将对象转换为 CSV 格式并写入文件：

@RestController
@RequestMapping("/csv")
public class CsvController {

    @Autowired
    private CsvMarshaller csvMarshaller;

    @PostMapping("/write")
    public void writeCsv(@RequestBody List<Person> people) {
        try {
            OutputStream outputStream = new FileOutputStream("people.csv");
            csvMarshaller.marshal(people, outputStream);
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
}