java修改数据库某条数据如何加锁

原创

mob64ca12d9081f 2024-08-24 06:56:15 ©著作权

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

Java修改数据库某条数据时的加锁处理

在现代的企业应用中，数据的并发处理是一个常见而复杂的问题。在多用户同时访问数据库的场景下，如何准确且安全地修改数据库中的数据是一个极具挑战性的任务。加锁机制可以有效地解决这个问题。本文将探讨如何在Java中对数据库某条数据进行加锁，并提供具体的示例和流程图。

问题背景

假设我们有一个用户账户管理系统，用户的账户余额存储在数据库中。当多个用户同时对同一账户进行操作（如转账、提现等）时，若没有适当的加锁机制，可能会导致数据的不一致或错误。因此，我们需要在对用户余额进行修改时加锁，以确保操作的原子性和一致性。

加锁的基本概念

在数据库中，加锁可以分为多种类型，主要包括：

共享锁（Shared Lock）：多个事务可以同时读取同一数据，但无法修改。
排他锁（Exclusive Lock）：只有一个事务可以读写该数据，其他事务必须等待，直至释放锁。

在Java中，我们可以使用JDBC与数据库结合，通过SQL语句来实现这些锁定机制。

使用示例

下面是一个使用Java和MySQL数据库对账户余额进行修改时加锁的示例。

数据库表结构

首先，假设我们有一个用户账户表 accounts，其结构如下：

CREATE TABLE accounts (
    id INT PRIMARY KEY AUTO_INCREMENT,
    username VARCHAR(50) NOT NULL,
    balance DECIMAL(10, 2) NOT NULL
);

Java代码示例

以下是Java代码示例，展示了如何在对账户余额进行修改时加锁。

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

public class AccountService {
    private final String url = "jdbc:mysql://localhost:3306/mydatabase";
    private final String user = "username";
    private final String password = "password";

    public void transferMoney(int fromAccountId, int toAccountId, double amount) {
        Connection connection = null;
        PreparedStatement lockStmt = null;
        PreparedStatement updateStmt = null;

        try {
            connection = DriverManager.getConnection(url, user, password);
            connection.setAutoCommit(false); // 开启事务

            // 对转出账户加锁
            String lockSQL = "SELECT balance FROM accounts WHERE id = ? FOR UPDATE";
            lockStmt = connection.prepareStatement(lockSQL);
            lockStmt.setInt(1, fromAccountId);
            lockStmt.executeQuery();

            // 对转入账户加锁
            lockStmt.setInt(1, toAccountId);
            lockStmt.executeQuery();

            // 更新转出账户余额
            String updateSQL = "UPDATE accounts SET balance = balance - ? WHERE id = ?";
            updateStmt = connection.prepareStatement(updateSQL);
            updateStmt.setDouble(1, amount);
            updateStmt.setInt(2, fromAccountId);
            updateStmt.executeUpdate();

            // 更新转入账户余额
            updateStmt.setDouble(1, -amount); // 余额增加
            updateStmt.setInt(2, toAccountId);
            updateStmt.executeUpdate();

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

流程图

以下是对上述示例的流程图描述，使用Mermaid语法：

flowchart TD
    A[开始] --> B{获取数据库连接}
    B --> C[开启事务]
    C --> D[对转出账户加锁]
    D --> E[对转入账户加锁]
    E --> F{是否失败？}
    F -- 是 --> G[回滚事务]
    G --> Z[结束]
    F -- 否 --> H[转出账户余额更新]
    H --> I[转入账户余额更新]
    I --> J[提交事务]
    J --> Z