使用数据库悲观锁实现不可重入的分布式锁

最新推荐文章于 2024-07-31 09:48:24 发布

煊琰

最新推荐文章于 2024-07-31 09:48:24 发布

阅读量150

点赞数

本文介绍如何使用数据库悲观锁机制实现不可重入的分布式锁。通过具体代码示例，展示如何利用for update命令锁定资源，确保在多线程或多JVM环境下资源访问的线程安全性。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

一、前言

在同一个jvm进程中时，可以使用JUC提供的一些锁来解决多个线程竞争同一个共享资源时候的线程安全问题，但是当多个不同机器上的不同jvm进程共同竞争同一个共享资源时候，juc包的锁就无能无力了，这时候就需要分布式锁了。常见的有使用zk的最小版本，redis的set函数，数据库锁来实现，本节我们谈谈使用数据库悲观锁机制来实现一个分布式锁。

二、使用数据库悲观锁实现不可重入的分布式锁

这个比较简单，先来看代码：

public class DBdistributedLock {

    private DataSource dataSource;

    private static final String cmd = "select * from lock where uid = 1 for update";

    public DBdistributedLock(DataSource ds) {
        this.dataSource = ds;
    }

    public static interface CallBack{
        public void doAction();
    }
    public void lock(CallBack callBack)  {
        Connection conn = null;
        PreparedStatement stmt = null;
        ResultSet rs = null;

        try {
            //try get lock
            System.out.println(Thread.currentThread().getName() + " begin try lock");
            conn = dataSource.getConnection();
            conn.setAutoCommit(false);
            stmt = conn.prepareStatement(cmd);
            rs = stmt.executeQuery();
          
            //do business thing
            callBack.doAction();
            
            //release lock
            conn.commit();
            System.out.println(Thread.currentThread().getName() + " release lock");

        } catch (SQLException e) {
            e.printStackTrace();

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

        }
    }

使用数据库悲观锁实现分布式锁主要用了数据库的for update命令，执行改命令后，对应行记录会被锁住，其它线程会被阻塞主，直到获取到这行记录的线程提交了事务。这里需要注意要把自动提交设置为false。

多个线程执行 rs = stmt.executeQuery();时候，只有一个线程会获取到行锁，其它线程被阻塞挂起，获取锁的线程执行传递的callback 的业务逻辑，执行完毕后执行commit 提交事务，这意味着当前线程释放了获取的锁，这时候被阻塞的线程会竞争获取该锁。

原文链接