Java 如何执行 SQL 脚本

在开发过程中，我们经常需要执行 SQL 脚本来初始化数据库或者更新数据。Java 提供了多种方式来执行 SQL 脚本，本文将介绍一种常用的方法，并给出代码示例来解决一个具体的问题。

问题描述

假设我们有一个学生信息管理系统的数据库，其中有一个名为 students 的表用于存储学生信息。现在我们需要初始化数据库，并插入一些初始的学生数据。我们已经编写了一个 SQL 脚本文件 init.sql，其中包含了创建 students 表和插入初始数据的 SQL 语句。我们希望通过 Java 代码来执行该脚本文件。

解决方案

我们可以使用 Java 的 JDBC（Java Database Connectivity）来执行 SQL 脚本。JDBC 是 Java 提供的用于与数据库进行交互的标准接口，我们可以使用它来连接数据库、执行 SQL 语句等。

以下是一个示例代码，展示了如何使用 JDBC 执行 SQL 脚本：

import java.io.BufferedReader;
import java.io.FileReader;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.Statement;

public class SqlScriptExecutor {

    public static void main(String[] args) {
        try {
            // 加载数据库驱动
            Class.forName("com.mysql.jdbc.Driver");

            // 连接数据库
            Connection connection = DriverManager.getConnection("jdbc:mysql://localhost:3306/mydb", "username", "password");

            // 创建 Statement 对象
            Statement statement = connection.createStatement();

            // 读取 SQL 脚本文件
            BufferedReader reader = new BufferedReader(new FileReader("init.sql"));
            StringBuilder sb = new StringBuilder();
            String line;
            while ((line = reader.readLine()) != null) {
                sb.append(line);
            }
            reader.close();

            // 执行 SQL 脚本
            String[] queries = sb.toString().split(";");
            for (String query : queries) {
                statement.executeUpdate(query);
            }

            // 关闭连接
            statement.close();
            connection.close();

            System.out.println("SQL 脚本执行成功！");
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
}

上述代码中，我们首先加载数据库驱动，这里以 MySQL 为例，需要加载 com.mysql.jdbc.Driver。然后使用 DriverManager.getConnection() 方法来连接数据库，需要传入数据库的 URL、用户名和密码。接着创建 Statement 对象，用于执行 SQL 语句。

我们通过读取 SQL 脚本文件 init.sql，将其中的 SQL 语句逐行读取并存储在一个 StringBuilder 中。然后使用 split() 方法将 SQL 语句分割成多个独立的查询语句，并使用 executeUpdate() 方法逐个执行这些查询语句。

最后，我们需要关闭数据库连接，释放资源。

代码说明

以下是对上述代码的一些说明：

  1. 在实际的代码中，需要根据数据库类型和连接信息来加载相应的数据库驱动和连接数据库。
  2. SQL 脚本文件 init.sql 应该提前准备好，并放置在代码所在的目录下或者指定的路径下。
  3. SQL 脚本文件中的 SQL 语句应该以分号 ; 结尾，用于分割多个 SQL 语句。
  4. 在实际的代码中，可能需要根据具体的需求对 SQL 脚本的执行过程进行异常处理，比如捕获 SQL 语句执行过程中的异常并进行相应的处理。

流程图

以下是执行 SQL 脚本的流程图：

flowchart TD
    start(开始)
    loadDriver[加载数据库驱动]
    connect[连接数据库]
    createStatement[创建 Statement 对象]
    readScript[读取 SQL 脚本文件]
    executeScript[执行 SQL 脚本]
    closeStatement[关闭 Statement]
    closeConnection[关闭数据库连接]
    end(结束)
    
    start --> loadDriver
    loadDriver --> connect
    connect --> createStatement
    createStatement --> readScript
    readScript --> executeScript
    executeScript --> closeStatement
    closeStatement --> closeConnection
    closeConnection --> end

序列图

以下是执行 SQL 脚本的序列图：

sequenceDiagram
    participant JavaApp as Java 应用程序
    participant JDBC as JDBC
    participant MySQL as MySQL 数据库
    
    Java