今天我们来学习可重入排他锁,它同样是JUC包下使用AQS同步框架来实现的,因此代码比较简洁,只要了解了AQS的原理那么就会觉得it is so easy。

1 怎么理解可重入锁

排他,说明锁只能被一人占有(就类似信号量(semaphore)为1,每一时刻只能有一人拥有)。可重入,同一个人可以多次使用同一资源,这非常合理。试想一下,当一只保险箱的钥匙归你所属后,难道你不能再去锁匠那再配一把来开箱子吗?在这个法制社会里,钥匙是你的,那你去配钥匙肯定是合法的,至于你为什么要配钥匙呢?一家人嘛,不过我相信你不可能把备份钥匙给别人用的吧,毕竟保险箱的东西可都是你们一家人,如果给了别人一份备用钥匙,那就不可想象喽。(这个和信号量有区别,信号量是每次使用都是需要获取凭证的,它可不管你们是不是一家人。不过其主要原因是两者关注点不同:信号量主要是关注的进入的量多少,而锁是为了保护资源的0安全,而为了安全,需要怎么做呢?当然确保是一家人喽)。

2 可重入锁(ReentrantLock)的condition

在分析源码之前,我们先简单来讲一下condition。Java Lock接口定义中有一个newCondition()方法,返回condition,而这个condition要实现的效果就类似于java中的wait/notify一致。也就是说这个condition的出现其实就是为了替代它(因为使用这个经常出错)。这里我们简单讲一下原理,有兴趣的小伙伴可以自行看一下源码,这里不再赘述:由于它还是使用了AQS的框架(其内部保持着一个同步队列,可以获取锁),那么为了实现等待和唤醒,它又增加了一个等待队列,当调用await方法时,节点加入到这个等待队列中。而当调用signal方法时,只要把节点从等待队列中拿出来再放到同步队列中,让其有机会去获取锁了,condition这个实现逻辑非常的清晰,也非常简洁,可见功底非同一般(T_T 我什么时候能达到这个程度啊)。

顺带提几句:

  1. 虽然这个方法可以很方便的实现生产者/消费者模型,但是java JUC包下其实已经有一个类(就是使用了ReentrantLock的condition)实现其功能了:LinkedBlockingQueue类(嘻嘻,没想到吧。)
  2. AQS-用团购的方式理解闭锁(CountDownLatch)这篇文章中我们有提到过能实现类似的功能的CyclicBarrier,其原理也用到了这个condition。感兴趣的小伙伴们可以去看看其实现哦。

3 可重入锁(ReentrantLock)源码分析

现在我们来看看锁的源码吧。对于锁而言,我们常用的方法有lock和unlock.。我们以开保险箱的情景来描述一下其实现。

构造函数:

// 规定一下,拿钥匙的排队规则:是否需要严格排队。
// 默认是false(吞吐量更大,因为公平锁带来更多上下文切换)。
 public ReentrantLock(boolean fair)

Fair:是否公平。和之前讲的信号量等类似,如果安全,则人必须按照排队的顺序来拿钥匙,否则后面来的人也有机会获取锁。

Lock 获取锁:(也可能是配锁):

 public void lock() {
        sync.lock();
    }
​
// 以非公平为例,公平的逻辑与信号量实现一致 会先判断是否前面有人排队,
final void lock() {
            if (compareAndSetState(0, 1))  // 尝试获取原装锁                setExclusiveOwnerThread(Thread.currentThread());  // 标记是谁获取锁的人
            else
                acquire(1);   // 实际调用tryAcquire方法
        }
​
        protected final boolean tryAcquire(int acquires) {
            return nonfairTryAcquire(acquires);
        }
​
 final boolean nonfairTryAcquire(int acquires) {
            final Thread current = Thread.currentThread();
            int c = getState();
            if (c == 0) {
                if (compareAndSetState(0, acquires)) {  // 尝试获取锁
                    setExclusiveOwnerThread(current);
                    return true;
                }
            }
            else if (current == getExclusiveOwnerThread()) {  // 可重入逻辑
                int nextc = c + acquires;
                if (nextc < 0) // overflow
                    throw new Error("Maximum lock count exceeded"); // 钥匙是会用坏的,次数有点多了。
                setState(nextc);
                return true;
            }
            return false;
        }

当人获取锁后呢,会记录下锁的拥有者。为什么要记录呢?当然是为了方便下一次这个家庭的人要使用保险箱,要拿钥匙需要去锁匠那里再配一把。具体分析一下代码:

if (compareAndSetState(0, 1)): 尝试获取原装锁。

setExclusiveOwnerThread(current): 记录获取钥匙的人,其实就是宝箱现在属于这个家庭了

else if (current == getExclusiveOwnerThread()) : 是不是钥匙的所属人一个家庭的,是的话他可以重新配一把钥匙哦。同时之后要记录保险箱一共有多少把钥匙了(方便后面归还,法制社会要保证保险箱绝对安全,必须所有钥匙都归还)。


锁的释放:(保险箱钥匙要传承给别人了,可是这么多钥匙都要还啊,不然谁知道你会不会后面来偷开别人的宝箱呢)

public void unlock() {
        sync.release(1);
    }
​
protected final boolean tryRelease(int releases) {
            int c = getState() - releases;  // 释放了钥匙,还有几把在外面没有还
            if (Thread.currentThread() != getExclusiveOwnerThread())
                throw new IllegalMonitorStateException();
            boolean free = false;
            if (c == 0) {   // 钥匙全还了
                free = true;
                setExclusiveOwnerThread(null); // 把宝箱所属人清空。
            }
            setState(c);
            return free;
        }

释放的逻辑非常简单,就是把所有外借的钥匙包括备份钥匙都归还了,那宝箱又成了无主之物。后面排队的人就又可以继续来获取钥匙了,同时也不用担心这个保险箱会有其备份钥匙在别人手上了。

  • 推荐阅读 -

用火车购票的方式打开 AQS同步器(一)

用火车购票的方式打开 AQS同步器(二)


小燃儿
110 声望2 粉丝

java developer.