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

布满荆棘的人生 2022-05-14 14:46 122阅读 0赞

# 一、前言 #

在同一个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 提交事务，这意味着当前线程释放了获取的锁，这时候被阻塞的线程会竞争获取该锁。

[原文链接][Link 1]

[Link 1]: http://click.aliyun.com/m/1000020101/