Blog Post

从源码入手详解ReentrantLock,一个比synchronized更强大的可重入锁

写在开头随手一翻,发现对于Java中并发多线程的学习已经发布了十几篇博客了,多线程 是Java基础中的重中之重!因此,可能还需要十几篇博客才能大致的讲完这部分的知识点,初学者对于这部分内容一定要多花心思,不可马虎!今天我们继续来学习一个重要知识点:ReentrantLock

ReentrantLock :是一种独占式的可重入锁,位于java.util.concurrent.locks中,是Lock接口的默认实现类,底部的同步特性基于AQS实现,和synchronized关键字类似,但更灵活、功能更强大、也是目前实战中使用频率非常高的同步类。

几种不同锁的定义在学习ReentrantLock之前,我们先来复习一下如下的几类锁的定义,这个其实很早的博文中就已经详细的整理过了,这里为了更好理解ReentrantLock锁,还是简单罗列一下。

独占锁与共享锁独占锁:同一时间,一把锁只能被一个线程获取;共享锁:同意时间,一把锁可以被多个线程获取。公平锁与非公平锁公平锁:按照申请锁的时间先后,进行锁的再分配工作,这种锁往往性能稍差,因为要保证申请时间上的顺序性;非公平锁: 锁被释放后,后续线程获得锁的可能性随机,或者按照设置的优先级进行抢占式获取锁。可重入锁所谓可重入锁就是一个线程在获取到了一个对象锁后,线程内部再次获取该锁,依旧可以获得,即便持有的锁还没释放,仍然可以获得,不可重入锁这种情况下会发生死锁!

可重入锁在使用时需要注意的是:由于锁会被获取 n 次,那么只有锁在被释放同样的 n 次之后,该锁才算是完全释放成功。

可中断锁与不可中断锁可中断锁:在获取锁的过程中可以中断获取,不需要非得等到获取锁后再去执行其他逻辑;不可中断锁:一旦线程申请了锁,就必须等待获取锁后方能执行其他的逻辑处理。ReentrantLock是一种同时拥有独占式、可重入、可中断、公平/非公平特性的同步器!

ReentrantLock根据上面总结出的特点,我们从底层源码出发来验证一下结论的准确性,首先我们通过一个关系图谱来大致梳理一下ReentrantLock的内部构造。

ReentrantLock实现了Lock和Serializable接口:

代码语言:javascript复制public class ReentrantLock implements Lock, java.io.Serializable {}

其内部拥有三个内部类,分别为Sync、FairSync、NonfariSync,其中FairSync、NonfariSync继承父类Sync。Sync又继承了AQS(AbstractQueuedSynchronizer),添加锁和释放锁的大部分操作实际上都是在 Sync 中实现的。

问题1:ReentrantLock内部公平锁与非公平锁如何实现?在内部通过构造器来实现公平锁与非公平锁的设置,默认为非公平锁,同样可以通过传参设置为公平锁。底层实现其实是通过FairSync、NonfariSync这个两个内部类,源码如下:

代码语言:javascript复制//无参构造,默认为非公平锁

public ReentrantLock() {

sync = new NonfairSync();

}

// 传入一个 boolean 值,true 时为公平锁,false 时为非公平锁

public ReentrantLock(boolean fair) {

sync = fair ? new FairSync() : new NonfairSync();

}

问题2:独占锁如何实现?在源码中无论是Sync这个内部类或是其子类,都会调用setExclusiveOwnerThread(current)这个方法,这个方法是AQS的父类AOS(AbstractOwnableSynchronizer)中的方法,用以标记锁的持有者为独占模式。

在这里插入图片描述

问题3:ReentrantLock如何获取和释放锁?由于ReentrantLock是默认非公平锁,所以我们就以非公平模式为例去看一下它底层如何实现锁的获取与释放的。

1️⃣ 锁的获取

核心方法为Sync内部类的nonfairTryAcquire方法,如下为其源码,先获取当前锁的状态,若为0说明没有被任何线程获取,此时直接获取即可;另外一种state不为0时,则需要判断占有线程是否为当前线程,若是则可以获取,并将state值加一返回,否则获取失败。

【注意】:公平模式下获取锁的时会多一步调用hasQueuedPredecessors 的逻辑判断,用以判断当前线程对应的节点在等待队列中是否有前驱节点,毕竟公平锁的竞争严格按照获取锁的时间进行分配的。

代码语言:javascript复制final boolean nonfairTryAcquire(int acquires) {

final Thread current = Thread.currentThread();

int c = getState();

//1. 如果该锁未被任何线程占有,该锁能被当前线程获取

if (c == 0) {

if (compareAndSetState(0, acquires)) {

setExclusiveOwnerThread(current);

return true;

}

}

//2.若被占有,检查占有线程是否是当前线程

else if (current == getExclusiveOwnerThread()) {

// 3. 再次获取,计数加一

int nextc = c + acquires;

if (nextc < 0) // overflow

throw new Error("Maximum lock count exceeded");

setState(nextc);

return true;

}

return false;

}

2️⃣ 锁的释放

对应的以非公平锁中释放为例,通过源码我们可以看到,每调用一次则同步状态减1,直至同步状态为0,锁才被完全的释放完,否则返回false。

代码语言:javascript复制protected final boolean tryRelease(int releases) {

//1. 同步状态减1

int c = getState() - releases;

if (Thread.currentThread() != getExclusiveOwnerThread())

throw new IllegalMonitorStateException();

boolean free = false;

if (c == 0) {

//2. 只有当同步状态为0时,锁成功被释放,返回true

free = true;

setExclusiveOwnerThread(null);

}

// 3. 锁未被完全释放,返回false

setState(c);

return free;

}

3️⃣ 小总结

经过上面源码的学习,我们已经能够确认一点就是:ReentrantLock是一种同时拥有独占式、可重入、可中断、公平/非公平特性的同步器!我们接下来就继续再来学习一下它的使用。

问题4:ReentrantLock的使用我们通过一个小demo,来感受一下基于非公平锁模式下的ReentrantLock的使用哈

代码语言:javascript复制public class Test {

//初始化一个静态lock对象

private static final ReentrantLock lock = new ReentrantLock();

//初始化计算量值

private static int count;

public static void main(String[] args) throws InterruptedException {

Thread thread1 = new Thread(()->{

for (int i = 0; i <1000 ; i++) {

lock.lock();

try {

count++;

} finally {

lock.unlock();

}

}

});

Thread thread2 = new Thread(()->{

for (int i = 0; i < 1000; i++) {

lock.lock();

try {

count++;

} finally {

lock.unlock();

}

}

});

thread1.start();

thread2.start();

thread1.join();

thread2.join();

System.out.println("result:"+count);

}

}

上面这个程序预期输出结果为:2000,thread1和thread2分别做了加1000次的操作,由于ReentrantLock是独占式可重入锁,故最终可以成功打印出预期结果!