在《Java Core 「11」AQS-AbstractQueuedSynchronizer》中介绍过 AQS 之后,再回过头来看一下 ReentrantLock 中的实现细节。
ReentrantLock 中有三个内部类,Sync 继承自 AQS,FairSync 实现公平锁机制,NonfairSync 实现非公平锁机制。
01-公平锁 & 非公平锁中的 tryAcquire
非公平锁中获取锁的过程:
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;j
复制代码
公平锁中获取锁的过程:
final Thread current = Thread.currentThread();
int c = getState();
if (c == 0) {
/** 锁空闲时 */
if (!hasQueuedPredecessors() && /** 检查 sync 队列总是否有其他线程在等待 */
compareAndSetState(0, acquires)) {
setExclusiveOwnerThread(current);
return true;
}
}
else if (current == getExclusiveOwnerThread()) {
/** 锁被当前线程占用,重入 */
int nextc = c + acquires;
if (nextc < 0)
throw new Error("Maximum lock count exceeded");
setState(nextc);
return true;
}
return false;
复制代码
通过上面的代码可以发现,在锁空闲时,当前线程能否获得锁的条件是不一样的:
02-Sync 中的 tryRelease
公平锁 & 非公平锁释放锁时使用的 tryRelease 都是同一份,即 Sync 中的 tryRelease 方法:
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;
复制代码
03-公平锁示例分析
我们结合一个实例分析一下多个线程获得公平锁的执行情况,我们创建三个线程 t1/t2/t3,执行同一份代码runnable
,使用同一公平锁lock
。
ReentrantLock lock = new ReentrantLock(true);
Runnable runnable = () -> {
lock.lock();
try {
System.out.println(Thread.currentThread() + " running");
try {
TimeUnit.MILLISECONDS.sleep(500);
} catch (InterruptedException ignored) {}
} finally {
lock.unlock();
}
};
new Thread(runnable, "t1").start();
new Thread(runnable, "t2").start();
new Thread(runnable, "t3").start();
复制代码
不妨假设线程被调度的顺序是 t1 → t2 → t3,所以存在如下一种时序:t1 执行lock.lock();
获得锁,开始执行并 sleep 500ms,此时 t2 执行lock.lock();
被阻塞,t3 执行lock.lock();
同样被阻塞。
注:以下的阅读顺序,可以按照先 Lock 流程,再 Unlock 流程。
03.1-t1 的执行流程
Lock 流程
ReentrantLock#lock → AbstractQueuedSynchronizer#acquire → ReentrantLock.FairSync#tryAcquire (true, state 是 0,sync 队列为空)
Unlock 流程
ReentrantLock#unlock → AbstractQueuedSynchronizer#release
public final boolean release(int arg) {
/** state 由1变0,返回结果 true */
if (tryRelease(arg)) {
Node h = head;
/** h 不为 null,而且 waitStatus 为 SIGNAL(-1),由 t2 自旋时 park 之前设置,具体看下节 03.2 */
if (h != null && h.waitStatus != 0)
/** 会将 h (head) 的 waitStatus 设置为 0,
* h.next != null,会 LockSupport.unpark(h.next.thread)
* t2 会在 LockSupport.park 处被唤醒
*/
unparkSuccessor(h);
return true;
}
return false;
}
复制代码
03.2-t2 的执行流程
Lock 流程
ReentrantLock#lock → AbstractQueuedSynchronizer#acquire → ReentrantLock.FairSync#tryAcquire (false,state 是 1,为 t1 tryAcquire 时设置) → AbstractQueuedSynchronizer#addWaiter (会初始化 sync 队列,并将 t2 包装到 Node 中加入队列尾部)→ AbstractQueuedSynchronizer#acquireQueued
final boolean acquireQueued(final Node node, int arg) {
boolean interrupted = false;
try {
for (;;) {
/** t2 执行时,p 为 head */
final Node p = node.predecessor();
{ /** ReentrantLock.FairSync#tryAcquire 仍会失败*/
if (p == head && tryAcquire(arg))
setHead(node);
p.next = null; // help GC
return interrupted;
}
/**
* 这段代码在无限循环中(线程 t2 在自旋)
* 第1次,p (head) 的 waitStatus 为0,然后被设置为 SIGNAL(-1)
* 第2次,线程 t2 会在 parkAndCheckInterrupt 中 LockSupport.park(this);
* 当 t1 在 unlock 中唤醒 t2 后,
* 根据 t2.isInterrupted(true) 判断是否被中断过更新 interrupted 的值
* 无限循环继续,再次执行到 ttryAcquire 时会获得锁
* 此时,setHead(node) 会把 head 指向 t2 所在节点
*/
if (shouldParkAfterFailedAcquire(p, node))
interrupted |= parkAndCheckInterrupt();
}
} catch (Throwable t) {
cancelAcquire(node);
if (interrupted)
selfInterrupt();
throw t;
}
}
复制代码
Unlock 流程
与 03.1 中 unlock 流程一致
03.3-t3 的执行流程
Lock 流程
ReentrantLock#lock → AbstractQueuedSynchronizer#acquire → ReentrantLock.FairSync#tryAcquire (false,state 是 1,为 t1 tryAcquire 时设置) → AbstractQueuedSynchronizer#addWaiter (将 t3 包装到 Node 中加入 sync 队列尾部) → AbstractQueuedSynchronizer#acquireQueued
final boolean acquireQueued(final Node node, int arg) {
boolean interrupted = false;
try {
for (;;) {
/** t3 执行时,p 为 t2 所在的节点 */
final Node p = node.predecessor();
{ /** 这里不会执行 */
if (p == head && tryAcquire(arg))
setHead(node);
p.next = null; // help GC
return interrupted;
}
/**
* 这段代码在无限循环中(线程 t3 在自旋)
* 第1次,p (t2 节点) 的 waitStatus 为0,然后被设置为 SIGNAL(-1)
* 第2次,线程 t3 会在 parkAndCheckInterrupt 中 LockSupport.park(this);
*/
if (shouldParkAfterFailedAcquire(p, node))
interrupted |= parkAndCheckInterrupt();
}
} catch (Throwable t) {
cancelAcquire(node);
if (interrupted)
selfInterrupt();
throw t;
}
}
复制代码
Unlock 流程
与 03.1 中 unlock 流程一致
历史文章推荐
Java Core 「11」AQS-AbstractQueuedSynchronizer
Java Core 「10」J.U.C 同步工具类 -2
Java Core 「9」J.U.C 同步工具类 -1
Java Core 「8」字节码增强技术
Java Core 「7」各种不同类型的锁
Java Core「6」反射与 SPI 机制
Java Core「5」自定义注解编程
Java Core「4」java.util.concurrent 包简介
Java Core「3」volatile 关键字
Java Core「2」synchronized 关键字
Java Core「1」JUC- 线程基础
评论