AQS-用配钥匙和保险箱理解可重入锁（ReentrantLock）

tech2022-08-09 142

今天我们来学习可重入排他锁，它同样是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 我什么时候能达到这个程度啊）。

顺带提几句：

虽然这个方法可以很方便的实现生产者/消费者模型，但是java JUC包下其实已经有一个类（就是使用了ReentrantLock的condition）实现其功能了：LinkedBlockingQueue类（嘻嘻，没想到吧。）

在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; }

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