Java ReentrantLock
同步锁synchronized和重入锁ReentrantLock都是用于并发程序设计必不可少的手段,在JDK 5.0早期版本中,同步锁性能远远低于重入锁,但是在6.0版本之后,jdk对同步锁做了大量的优化,使得同步锁跟重入锁性能差距并不大,并且jdk团队表示,同步锁还有进一步升级优化的空间

一、重入锁的概念

重入的概念是指在同一个线程内部,这种锁是可以反复进入的。

  1. lock.lock();
  2. lock.lock();
  3. // do something
  4. lock.unlock();
  5. lock.unlock();

一个线程可多次获取锁,但同时也要释放相同的次数,否则该线程将持续拥有锁,其他线程将无法进入临界区。

二、重入锁的几个重要方法

2021-10-01-11-42-00-141740.jpg
lock:获取锁,如果锁被其他线程占用,则休眠等待。
lockInterruptibly:获取锁,可以被其他线程所中断
tryLock:尝试获取锁,不等待
tryLock(time, timeUnit):在一定时间内尝试获取锁
unlock: 释放锁

2.1 中断响应

对于synchronized来说,一个线程要么获取到锁开始执行,要么继续等待。但是对于重入锁来说,提供了更灵活的一种机制,那就是在等待锁的过程中,可以取消对锁的请求,这样可以有效避免死锁的可能。

2.2 锁申请等待时间

中断响应是一种通过外部通知中断对锁的请求,从而避免死锁的一种机制。除此之外,还有一种机制,那就是等待限时。

  1. tryLock(long timeout, TimeUnit unit)

2.3 公平锁、非公平锁

重入锁默认是非公平锁

  1. public ReentrantLock() {
  2. sync = new NonfairSync();
  3. }

可以通过构造函数参数实现公平锁

  1. public ReentrantLock(boolean fair) {
  2. sync = fair ? new FairSync() : new NonfairSync();
  3. }

如果是非公平锁,在并发场景下,系统会随机从等待队列中挑选一个线程。如果是公平锁,系统会维护一个有序队列,会按照进入队列的次序有序执行,因此公平锁虽然避免了饥饿现象,但是会需要更高的成本来维护这个有序队列。

2.4 AQS的组成结构

重入锁的加锁和解锁过程主要有AQS完成,AQS维护了一个双向链表,每个节点Node存储一个线程及线程的状态,Head节点代表正在持有锁的线程。
2021-10-01-11-42-00-340745.jpg
节点状态:
CANCELLED: 当前节点超时或者中断被取消
SIGNAL:当前节点的后续节点处于等待状态
CONDITION:当前节点再等待condition
PROPAGATE:状态向后传播

  1. static final class Node {
  2. /** waitStatus状态:当前节点被取消调度*/
  3. static final int CANCELLED = 1;
  4. /** waitStatus状态:后续节点等待被唤醒*/
  5. static final int SIGNAL = -1;
  6. /** waitStatus状态:当前节点等待在condition上,
  7. * 当其他线程调用了Condition的signal()方法后,
  8. * CONDITION状态的结点将从等待队列转移到同步队列中,
  9. * 等待获取同步锁*/
  10. static final int CONDITION = -2;
  11. /** waitStatus状态:共享模式下,会唤醒后续节点及后续的后续节点*/
  12. static final int PROPAGATE = -3;
  13. /** 节点状态*/
  14. volatile int waitStatus;
  15. }

当线程获取锁失败之后,就通过addWaiter加入到同步队列中(加入到尾部),自旋判断自己是否是链表的头节点,如果是头节点,就不断参试获取资源,获取成功后则退出同步队列。
深入理解ReentrantLock - 图3

三、重入锁的实现原理

首先看下重入锁的类图关系
深入理解ReentrantLock - 图4
ReentrantLock定义了内部类SyncSync继承自AbstractQueuedSynchronizer(简称AQS),是一个同步等待队列,本质上是一个带有头尾指针的双向链表。
AQS中有一个很重要的变量, 不同的组件中代表的含义有所不同,而在重入锁组件中,代表被线程重入的次数,为0表示没有线程持有锁。

  1. private volatile int state;

3.1 公平锁的加锁过程

sync.lock时调用AQS的acquire方法,这是一种模板设计模式,即AQS中定义了整体的处理流程,但是具体的实现细节会根据锁类型的不同,放到子类方法中执行。

  1. public final void acquire(int arg) {
  2. // 尝试获取锁
  3. if (!tryAcquire(arg) &&
  4. // 获取锁失败的线程安全的加入到等待队列中
  5. acquireQueued(addWaiter(Node.EXCLUSIVE), arg))
  6. // 自行了断
  7. selfInterrupt();
  8. }

下面来看看每一步的实现细节

公平锁的tryAcquire

tryAcquire是一个钩子方法。实现细节放在具体的子类中。
深入理解ReentrantLock - 图5

  1. protected final boolean tryAcquire(int acquires) {
  2. final Thread current = Thread.currentThread();
  3. //获取当前锁的状态
  4. int c = getState();
  5. // 没有线程占用
  6. if (c == 0) {
  7. // 如果当前线程排在队列中的第一个,并且CAS抢占锁成功
  8. if (!hasQueuedPredecessors() &&
  9. compareAndSetState(0, acquires)) {
  10. // 设置占用锁的线程为当前线程
  11. setExclusiveOwnerThread(current);
  12. return true;
  13. }
  14. }
  15. // 判断占有锁的线程是否是当前线程
  16. else if (current == getExclusiveOwnerThread()) {
  17. int nextc = c + acquires;
  18. if (nextc < 0)
  19. throw new Error("Maximum lock count exceeded");
  20. // 重入次数加1,acquires是由acquire(1)传入
  21. setState(nextc);
  22. return true;
  23. }
  24. return false;
  25. }

acquireQueued(addWaiter())

线程在同步队列中会尝试获取锁,失败则被阻塞,被唤醒后会不停的重复这个过程,直到线程真正持有了锁,并将自身结点置于队列头部。对算法感兴趣的同学可以自行深入研究一下。

addWaiter入队过程(队列不为空时)

1、根据当前线程创建一个新的Node节点
2、如果队尾元素不为空,则通过CAS操作插入到队尾
3、如果队列为空,new一个节点并设置为对头head
2021-10-01-11-42-00-762742.jpg

  1. private Node addWaiter(Node mode) {
  2. Node node = new Node(Thread.currentThread(), mode);
  3. // Try the fast path of enq; backup to full enq on failure
  4. Node pred = tail;
  5. if (pred != null) {
  6. node.prev = pred;
  7. if (compareAndSetTail(pred, node)) {
  8. pred.next = node;
  9. return node;
  10. }
  11. }
  12. enq(node);
  13. return node;
  14. }
  15. //入队逻辑
  16. private Node enq(final Node node) {
  17. for (;;) {
  18. Node t = tail;
  19. if (t == null) { // Must initialize
  20. if (compareAndSetHead(new Node()))
  21. tail = head;
  22. } else {
  23. node.prev = t;
  24. if (compareAndSetTail(t, node)) {
  25. t.next = node;
  26. return t;
  27. }
  28. }
  29. }
  30. }

在元素入队之后,又做了哪些操作呢?
1、通过自旋检测是否有资格获取锁,如果获取到了锁,则将当前节点设置为头节点
2、如果不是Head->next节点或获取锁失败,则阻塞当前线程

  1. final boolean acquireQueued(final Node node, int arg) {
  2. boolean failed = true;
  3. try {
  4. boolean interrupted = false;
  5. for (;;) {
  6. // 获取当前节点的prev节点
  7. final Node p = node.predecessor();
  8. // 如果是头节点,则去尝试获取锁
  9. if (p == head && tryAcquire(arg)) {
  10. // 获取锁成功,则将当前节点设置为头节点
  11. setHead(node);
  12. p.next = null; // help GC
  13. failed = false;
  14. return interrupted;
  15. }
  16. // 如果没有获取到锁,先判断是否需要阻塞当前线程
  17. if (shouldParkAfterFailedAcquire(p, node) &&
  18. // 通过LockSupport.park(this)原语阻塞当前线程
  19. parkAndCheckInterrupt())
  20. interrupted = true;
  21. }
  22. } finally {
  23. if (failed)
  24. cancelAcquire(node);
  25. }
  26. }

这里最难理解的是这句,为什么当前节点的前节点是头节点时尝试获取锁呢?因为头节点代表当前正在持有锁的线程,该线程执行结束释放锁时,会唤醒队列中阻塞的线程,被唤醒的线程会尝试去获取锁。

  1. if (p == head && tryAcquire(arg)) {
  2. ...
  3. }

3.2 公平锁的解锁过程

  1. public void unlock() {
  2. sync.release(1);
  3. }
  4. public final boolean release(int arg) {
  5. // 释放锁
  6. if (tryRelease(arg)) {
  7. Node h = head;
  8. // 队列不为空
  9. if (h != null && h.waitStatus != 0)
  10. // 唤醒队列中阻塞的线程
  11. unparkSuccessor(h);
  12. return true;
  13. }
  14. return false;
  15. }

释放锁的具体逻辑如下

  1. protected final boolean tryRelease(int releases) {
  2. // 次数-1
  3. int c = getState() - releases;
  4. // 判断是否是当前线程
  5. if (Thread.currentThread() != getExclusiveOwnerThread())
  6. throw new IllegalMonitorStateException();
  7. boolean free = false;
  8. // 释放后没有线程占用
  9. if (c == 0) {
  10. free = true;
  11. // 将占有锁线程置为空
  12. setExclusiveOwnerThread(null);
  13. }
  14. // 更新状态
  15. setState(c);
  16. return free;
  17. }

释放完成后唤醒后续节点

  1. private void unparkSuccessor(Node node) {
  2. //node是当前线程所在节点
  3. int ws = node.waitStatus;
  4. // 状态置为0
  5. if (ws < 0)
  6. compareAndSetWaitStatus(node, ws, 0);
  7. // 找到下个需要被唤醒的节点
  8. Node s = node.next;
  9. if (s == null || s.waitStatus > 0) {
  10. s = null;
  11. for (Node t = tail; t != null && t != node; t = t.prev)
  12. if (t.waitStatus <= 0)
  13. s = t;
  14. }
  15. if (s != null)
  16. //唤醒线程
  17. LockSupport.unpark(s.thread);
  18. }

3.3 非公平锁和公平锁的实现原理基本类似

非公平锁在尝试获取锁时不会判断是否是队列中的第一个元素,而是直接进行CAS比较。 释放锁的流程是一样的。

  1. final boolean nonfairTryAcquire(int acquires) {
  2. final Thread current = Thread.currentThread();
  3. int c = getState();
  4. if (c == 0) {
  5. if (compareAndSetState(0, acquires)) {
  6. setExclusiveOwnerThread(current);
  7. return true;
  8. }
  9. }
  10. else if (current == getExclusiveOwnerThread()) {
  11. int nextc = c + acquires;
  12. if (nextc < 0) // overflow
  13. throw new Error("Maximum lock count exceeded");
  14. setState(nextc);
  15. return true;
  16. }
  17. return false;
  18. }

四、总结

重入锁的实现主要有三个关键的知识点
1、同步等待队列,线程在尝试请求锁失败后,会进入到等待队列中等待被唤醒。如果有线程释放了锁,系统会唤醒等待队列中的一个线程。
2、原子性:原子状态是通过CAS操作来存储当前锁的状态(即state:重入次数)
3、通过LockSupport.parkLockSupport.unpark来挂起和恢复线程,未获取到锁的线程会被挂起。