java读取csv文件存储于数据库

原创

mob649e815adb02 2024-08-04 06:42:30 ©著作权

©著作权归作者所有：来自51CTO博客作者mob649e815adb02的原创作品，请联系作者获取转载授权，否则将追究法律责任

Java读取CSV文件并存储于数据库的完整指南

对于刚入行的小白来说，理解如何在Java中读取CSV文件并将数据存储到数据库中，虽然看起来困难，但只要明确步骤，代码实现起来并不复杂。本文将帮助你理解整个过程，以及每一个环节需要做什么。

流程概述

我们将整个流程分为几个主要步骤，下面以表格展示：

步骤	描述	代码示例
1. 引入依赖	使用必要的库	-
2. 读取CSV	使用Java读取CSV文件数据	`readCsvFile(String filePath)`
3. 连接数据库	设置数据库连接	`connectToDatabase()`
4. 存储数据	将读取的数据存储到数据库	`insertDataIntoDatabase()`
5. 关闭连接	关闭数据库连接，释放资源	`closeConnection()`

详细步骤说明

1. 引入依赖

在Java中，我们需要使用一些库来处理CSV文件和数据库连接。可以在pom.xml文件中增加以下依赖（如果是用Maven管理项目的话）：

<dependency>
    <groupId>org.apache.commons</groupId>
    <artifactId>commons-csv</artifactId>
    <version>1.8</version>
</dependency>
<dependency>
    <groupId>mysql</groupId>
    <artifactId>mysql-connector-java</artifactId>
    <version>8.0.27</version>
</dependency>

2. 读取CSV文件

下面的代码展示了如何读取CSV文件：

import org.apache.commons.csv.CSVFormat;
import org.apache.commons.csv.CSVRecord;
import java.io.FileReader;
import java.io.Reader;

public void readCsvFile(String filePath) {
    try {
        Reader in = new FileReader(filePath);
        Iterable<CSVRecord> records = CSVFormat.DEFAULT.withFirstRecordAsHeader().parse(in);
        for (CSVRecord record : records) {
            String column1 = record.get("Column1");
            String column2 = record.get("Column2");
            // 这里可以打印数据，确认了文件格式
            System.out.println(column1 + ", " + column2);
            // 继续处理，比如将数据插入数据库
            insertDataIntoDatabase(column1, column2);
        }
    } catch (Exception e) {
        e.printStackTrace();
    }
}

这段代码通过 Apache Commons CSV 库读取CSV文件，并逐行解析数据。需要注意，withFirstRecordAsHeader() 表示使用第一行作为表头。

3. 连接数据库

接下来，我们需要连接到数据库。以下是数据库连接的示例代码：

import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.SQLException;

public Connection connectToDatabase() {
    String url = "jdbc:mysql://localhost:3306/your_database";
    String user = "your_username";
    String password = "your_password";
    Connection connection = null;
    
    try {
        connection = DriverManager.getConnection(url, user, password);
        System.out.println("Database connected!");
    } catch (SQLException e) {
        e.printStackTrace();
    }
    return connection;
}

在这里，记得用你实际的数据库名、用户名和密码替换相应的字段。

4. 存储数据到数据库

将数据插入到数据库的过程如下：

import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.SQLException;

public void insertDataIntoDatabase(String column1, String column2) {
    String query = "INSERT INTO your_table (column1, column2) VALUES (?, ?)";
    try (Connection connection = connectToDatabase(); 
         PreparedStatement preparedStatement = connection.prepareStatement(query)) {
        preparedStatement.setString(1, column1);
        preparedStatement.setString(2, column2);
        preparedStatement.executeUpdate();
        System.out.println("Data inserted successfully");
    } catch (SQLException e) {
        e.printStackTrace();
    }
}

使用 PreparedStatement 预处理SQL语句可以更安全有效地插入数据，避免SQL注入。

5. 关闭数据库连接

完成所有操作后，及时关闭数据库连接是良好的编程习惯：

public void closeConnection(Connection connection) {
    if (connection != null) {
        try {
            connection.close();
            System.out.println("Connection closed");
        } catch (SQLException e) {
            e.printStackTrace();
        }
    }
}

使用 try-catch 处理关闭连接的例外情况，确保稳定性。

流程图与序列图

下面是流程图展示整个操作过程。

journey
    title 读取CSV文件并存储到数据库
    section 步骤
      引入依赖: 5: 楼
      读取CSV: 3: 楼
      连接数据库: 4: 楼
      存储数据: 5: 楼
      关闭连接: 3: 楼

接下来是序列图，展示了各个方法之间的调用关系。

sequenceDiagram
    participant User
    participant CSVReader
    participant Database
    User->>CSVReader: readCsvFile(filePath)
    CSVReader->>Database: connectToDatabase()
    Database-->>CSVReader: Connection
    CSVReader->>Database: insertDataIntoDatabase(column1, column2)
    Database-->>CSVReader: Success
    CSVReader->>Database: closeConnection()