Mysql

关注公众号 jb51net

关闭
首页 > 数据库 > Mysql > MySQL可重入锁实现

MySQL实现可重入锁的实践指南

作者:java修仙传

本文介绍了如何在MySQL中实现可重入锁,包括锁表设计、加锁和解锁逻辑以及为什么必须使用事务来保证锁的互斥性、原子性和生命周期,需要的朋友可以参考下

在分布式系统和并发编程中,锁是保证数据一致性的关键工具。而基于 MySQL 实现的可重入锁,不仅能满足跨进程的互斥需求,还能支持同一个线程多次获取锁而不阻塞。记录一下我对 MySQL 实现可重入锁的思考:如何用 MySQL 实现可重入锁?为什么实现过程中必须依赖事务?希望能解决你的一些疑惑。

一、先搞懂:什么是可重入锁?

可重入锁(也叫递归锁)的核心特性是:同一个线程可以多次获取同一把锁,不会因为自己持有锁而发生死锁。

举个例子:

在 Java 里 ReentrantLock 就是典型的可重入锁,而我们要做的,是用 MySQL 模拟出类似的效果。

二、MySQL 可重入锁的基础:锁表设计

要实现可重入锁,我们需要一张表来记录锁的持有状态、持有者和重入次数:

CREATE TABLE `lock_table` (
  `id` INT AUTO_INCREMENT PRIMARY KEY,
  -- 锁的唯一标识
  `lock_name` VARCHAR(255) NOT NULL,
  -- 持有锁的线程标识
  `holder_thread` VARCHAR(255),
  -- 锁的重入次数,用于实现可重入性
  `reentry_count` INT DEFAULT 0
);

这张表的核心字段:

三、核心实现:加锁与解锁逻辑

1. 加锁流程

1. 开启事务
2. 执行 SQL:
   SELECT holder_thread, reentry_count 
   FROM lock_table 
   WHERE lock_name = ? FOR UPDATE;
   - 若记录不存在:执行 INSERT INTO lock_table (lock_name, holder_thread, reentry_count) VALUES (?, ?, 1)
   - 若记录存在且持有者是当前线程:执行 UPDATE lock_table SET reentry_count = reentry_count + 1 WHERE lock_name = ?
3. 提交事务

2. 解锁流程

1. 开启事务
2. 执行 SQL:
   SELECT holder_thread, reentry_count 
   FROM lock_table 
   WHERE lock_name = ? FOR UPDATE;
   - 若记录存在、持有者是当前线程且重入次数 > 1:执行 UPDATE lock_table SET reentry_count = reentry_count - 1 WHERE lock_name = ?
   - 若记录存在、持有者是当前线程且重入次数 ≤ 1:执行 DELETE FROM lock_table WHERE lock_name = ?
3. 提交事务

四、灵魂拷问:为什么必须加事务?

很多同学会疑惑:“我不加事务,直接执行 SQL 不行吗?” 答案是:不行。事务是 MySQL 可重入锁的灵魂,没有事务,锁的特性会直接失效。我们从三个维度拆解原因:

1. 控制锁的生命周期:避免锁提前释放

MySQL InnoDB 默认 autocommit=1,单条 SQL 执行完毕后会自动提交事务。

如果不加事务:

这会导致:

而事务的作用就是:在事务提交前,锁不会被释放。从查询锁状态到完成写入操作,整个过程中锁都被当前事务持有,保证了互斥性和可重入性。

2. 保证操作原子性:避免并发数据错乱

加锁 / 解锁的核心逻辑是「查询 → 判断 → 写入」,这三步必须是原子操作,否则在高并发场景下会出现数据错乱。

举个并发场景:

事务的原子性可以完美解决这个问题:

3. 确保可重入正确性:重入次数的安全更新

可重入锁的核心是维护 reentry_count 字段:

如果不加事务:

事务的隔离性保证了:

五、完整示例:

我们用伪代码把加锁和解锁流程串起来,更直观地感受事务的作用:

加锁伪代码

public boolean lock(String lockName, String threadName) {
    Connection conn = getConnection();
    try {
        // 1. 开启事务
        conn.setAutoCommit(false);
        // 2. 查询锁状态(加排他锁)
        String querySql = "SELECT holder_thread, reentry_count FROM lock_table WHERE lock_name = ? FOR UPDATE";
        try (PreparedStatement ps = conn.prepareStatement(querySql)) {
            ps.setString(1, lockName);
            ResultSet rs = ps.executeQuery();
            if (rs.next()) {
                String holder = rs.getString("holder_thread");
                int count = rs.getInt("reentry_count");
                if (threadName.equals(holder)) {
                    // 可重入:重入次数+1
                    String updateSql = "UPDATE lock_table SET reentry_count = reentry_count + 1 WHERE lock_name = ?";
                    try (PreparedStatement updatePs = conn.prepareStatement(updateSql)) {
                        updatePs.setString(1, lockName);
                        updatePs.executeUpdate();
                    }
                } else {
                    // 被其他线程持有,获取锁失败
                    conn.rollback();
                    return false;
                }
            } else {
                // 锁不存在,直接加锁
                String insertSql = "INSERT INTO lock_table (lock_name, holder_thread, reentry_count) VALUES (?, ?, 1)";
                try (PreparedStatement insertPs = conn.prepareStatement(insertSql)) {
                    insertPs.setString(1, lockName);
                    insertPs.setString(2, threadName);
                    insertPs.executeUpdate();
                }
            }
        }
        // 3. 提交事务
        conn.commit();
        return true;
    } catch (Exception e) {
        // 异常回滚
        conn.rollback();
        return false;
    } finally {
        conn.setAutoCommit(true);
        closeConnection(conn);
    }
}

解锁伪代码

public boolean unlock(String lockName, String threadName) {
    Connection conn = getConnection();
    try {
        // 1. 开启事务
        conn.setAutoCommit(false);
        // 2. 查询锁状态
        String querySql = "SELECT holder_thread, reentry_count FROM lock_table WHERE lock_name = ? FOR UPDATE";
        try (PreparedStatement ps = conn.prepareStatement(querySql)) {
            ps.setString(1, lockName);
            ResultSet rs = ps.executeQuery();
            if (rs.next()) {
                String holder = rs.getString("holder_thread");
                int count = rs.getInt("reentry_count");
                if (!threadName.equals(holder)) {
                    // 不是锁持有者,解锁失败
                    conn.rollback();
                    return false;
                }
                if (count > 1) {
                    // 重入次数>1,仅减1
                    String updateSql = "UPDATE lock_table SET reentry_count = reentry_count - 1 WHERE lock_name = ?";
                    try (PreparedStatement updatePs = conn.prepareStatement(updateSql)) {
                        updatePs.setString(1, lockName);
                        updatePs.executeUpdate();
                    }
                } else {
                    // 重入次数=1,删除锁记录
                    String deleteSql = "DELETE FROM lock_table WHERE lock_name = ?";
                    try (PreparedStatement deletePs = conn.prepareStatement(deleteSql)) {
                        deletePs.setString(1, lockName);
                        deletePs.executeUpdate();
                    }
                }
            } else {
                // 锁不存在,解锁失败
                conn.rollback();
                return false;
            }
        }
        // 3. 提交事务
        conn.commit();
        return true;
    } catch (Exception e) {
        conn.rollback();
        return false;
    } finally {
        conn.setAutoCommit(true);
        closeConnection(conn);
    }
}

六、总结

用 MySQL 实现可重入锁,本质是用数据库表存储锁状态,用事务保证锁的互斥性、原子性和生命周期。

事务的核心作用可以概括为三点:

  1. 锁生命周期管理:事务提交前,锁不会释放,保证当前线程持续持有锁。
  2. 原子性保障:将「查询 → 判断 → 写入」封装为原子操作,避免并发数据错乱。
  3. 可重入正确性:安全维护重入次数,确保同一个线程可以多次获取 / 释放锁。

这种实现方式虽然不如 Redis 等分布式锁框架高效,但胜在简单可靠,适合对性能要求不高、需要强一致性的场景,也能帮我们更好地理解事务和锁的本质。

以上就是MySQL实现可重入锁的实践指南的详细内容,更多关于MySQL可重入锁实现的资料请关注脚本之家其它相关文章!

您可能感兴趣的文章:
阅读全文