java 线程中使用事务

原创

mob649e816a77bf 2024-10-11 11:03:34 ©著作权

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

在Java线程中使用事务

事务管理是实现高一致性和可靠性的关键，它可以确保一系列操作要么全部成功，要么全部失败。开发者在进行多线程编程时，如何在每个线程中使用数据库事务是一个常见的问题。本文将介绍如何在Java线程中实现事务，步骤和代码示例详细讲解。

流程概述

下面是使用Java线程和事务的步骤概述：

步骤	描述
1	创建线程任务
2	获取数据库连接
3	开始事务
4	执行数据库操作
5	提交/回滚事务
6	释放资源

步骤详解

1. 创建线程任务

首先，我们需要定义一个实现 Runnable 接口的类，用于在线程中执行事务操作。

public class TransactionTask implements Runnable {
    @Override
    public void run() {
        // 在这里执行数据库事务逻辑
    }
}

这里我们实现了 Runnable 接口，并在 run() 方法中定义了线程将要执行的逻辑。

2. 获取数据库连接

在执行事务之前，我们需要获取数据库连接。以下代码展示了如何获取数据库连接：

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

public class DatabaseUtil {
    public static Connection getConnection() throws SQLException {
        String url = "jdbc:mysql://localhost:3306/mydb"; // 数据库URL
        String user = "root"; // 数据库用户名
        String password = "password"; // 数据库密码
        return DriverManager.getConnection(url, user, password); // 返回数据库连接
    }
}

使用 DriverManager 提供的 getConnection 方法获取连接。

3. 开始事务

在 run 方法中，我们使用连接对象来设置事务的开始状态：

Connection connection = null;
try {
    connection = DatabaseUtil.getConnection(); // 获取数据库连接
    connection.setAutoCommit(false); // 开始事务，关闭自动提交
} catch (SQLException e) {
    e.printStackTrace();
}

这里，我们将 auto commit 设置为 false，以启动事务。

4. 执行数据库操作

我们可以通过 Connection 对象执行 SQL 语句。下面是一个插入数据的示例：

import java.sql.PreparedStatement;

try {
    String sql = "INSERT INTO users (name, email) VALUES (?, ?)"; // SQL语句
    PreparedStatement preparedStatement = connection.prepareStatement(sql);
    preparedStatement.setString(1, "John Doe"); // 设置参数
    preparedStatement.setString(2, "john@example.com"); // 设置参数
    preparedStatement.executeUpdate(); // 执行插入操作
} catch (SQLException e) {
    e.printStackTrace();
}

这里我们使用 PreparedStatement 为防止 SQL 注入并执行插入操作。

5. 提交/回滚事务

在执行完数据库操作后，我们可以选择是否提交事务：

try {
    connection.commit(); // 提交事务
} catch (SQLException e) {
    try {
        connection.rollback(); // 回滚事务
    } catch (SQLException rollbackEx) {
        rollbackEx.printStackTrace();
    }
}

根据操作结果，你决定是否提交或回滚事务。

6. 释放资源

最后，不要忘记释放数据库连接资源：

finally {
    if (connection != null) {
        try {
            connection.close();
        } catch (SQLException e) {
            e.printStackTrace();
        }
    }
}

使用 finally 块确保连接总是被关闭。

完整代码示例

以下是完整的代码示例，结合上面的所有步骤：

public class TransactionTask implements Runnable {
    @Override
    public void run() {
        Connection connection = null;
        try {
            connection = DatabaseUtil.getConnection(); // 获取数据库连接
            connection.setAutoCommit(false); // 开始事务

            // 执行数据库操作
            String sql = "INSERT INTO users (name, email) VALUES (?, ?)";
            PreparedStatement preparedStatement = connection.prepareStatement(sql);
            preparedStatement.setString(1, "John Doe");
            preparedStatement.setString(2, "john@example.com");
            preparedStatement.executeUpdate();

            connection.commit(); // 提交事务
        } catch (SQLException e) {
            if (connection != null) {
                try {
                    connection.rollback(); // 回滚事务
                } catch (SQLException rollbackEx) {
                    rollbackEx.printStackTrace();
                }
            }
            e.printStackTrace();
        } finally {
            if (connection != null) {
                try {
                    connection.close(); // 释放资源
                } catch (SQLException e) {
                    e.printStackTrace();
                }
            }
        }
    }
}

序列图展示

以下是描述线程执行事务的序列图：

sequenceDiagram
    participant T as Thread
    participant D as Database
    T->>D: 开始事务
    T->>D: 执行 SQL 语句
    T-->>D: 提交事务
    alt 事务失败
        T-->>D: 回滚事务
    end
    T->>D: 关闭连接