NOTE
6.26 公平锁
所谓公平锁,遵循先到先得的原则。 即使锁已经被释放了,后到的也不能去抢占锁,得等到前面没人时才能去获取 1. 如何使用 2. 原理分析 2.1. 构造方法 2.1.1. 底层使用AQS实现 2.2. 加锁 - lock 2.2.1. 调用公平锁的lock方法 - FairSync.lock 2.2.
这是历史学习笔记,可能存在过时或不完整的理解。
所谓公平锁,遵循先到先得的原则。 即使锁已经被释放了,后到的也不能去抢占锁,得等到前面没人时才能去获取
1. 如何使用
public class TestReentrantLock
{
private static int val = 0;
private final static Lock lock = new ReentrantLock(true);//公平锁
public static void main(String[] args) throws InterruptedException
{
Thread thread1 = new Thread(() -> {
for (int i = 0; i < 100000; i++)
{
try
{
lock.lock();
val++;
}
finally
{
lock.unlock();
}
}
});
Thread thread2 = new Thread(() -> {
for (int i = 0; i < 100000; i++)
{
try
{
lock.lock();
val--;
}
finally
{
lock.unlock();
}
}
});
thread1.start();
thread2.start();
thread1.join();
thread2.join();
System.out.println(val);
}
}
2. 原理分析
2.1. 构造方法
2.1.1. 底层使用AQS实现
public class ReentrantLock implements Lock, java.io.Serializable {
private final Sync sync;
//默认非公平锁
public ReentrantLock() {
sync = new NonfairSync();
}
public ReentrantLock(boolean fair) {
//true的话,公平锁使用FairSync,否则是NonfairSync
sync = fair ? new FairSync() : new NonfairSync();
}
//Sync是AQS的子类
abstract static class Sync extends AbstractQueuedSynchronizer {}
//FairSync是Sync的子类
static final class FairSync extends Sync {}
}
2.2. 加锁
- lock
public void lock() {
//调用FairSync的lock
sync.lock();
}
2.2.1. 调用公平锁的lock方法
- FairSync.lock
final void lock() {
//调用AQS的acquire
acquire(1);
}
2.2.2. 调用AQS的acquire方法获取锁
- AQS.acquire
public final void acquire(int arg) {
//调用FairSync的tryAcquire获取锁
if (!tryAcquire(arg) &&
//获取锁失败加入AQS队列。并且死循环阻塞当前线程,等待唤醒继续获取锁
acquireQueued(addWaiter(Node.EXCLUSIVE), arg))
//恢复中断标记
selfInterrupt();
}
由于FairSync重写了AQS的tryAcquire方法,因此这里会调用FairSync的tryAcquire 其他的逻辑同5.AQS.md,下面只是简要说一下主要逻辑
2.2.3. 尝试获取锁【只有队头才允许抢占锁–公平锁】
- FairSync.tryAcquire
protected final boolean tryAcquire(int acquires) {
final Thread current = Thread.currentThread();
int c = getState();
//锁尚未被获取
if (c == 0) {
//【公平锁】:队列中我的前面没人等待锁(队列为空或者我就是队列的队头)
if (!hasQueuedPredecessors() &&
//CAS设置state获取锁成功
compareAndSetState(0, acquires)) {
//设置持有锁的线程为当前线程
setExclusiveOwnerThread(current);
return true;
}
}
//锁已经被获取,且是当前线程,那么重入
else if (current == getExclusiveOwnerThread()) {
//增加state量
int nextc = c + acquires;
if (nextc < 0)
throw new Error("Maximum lock count exceeded");
setState(nextc);
return true;
}
//获取锁失败返回false
return false;
}
2.2.4. 尝试获取锁失败加入阻塞队列
- AQS.addWaiter
private Node addWaiter(Node mode) {
//用当前线程、EXCLUSIVE模式构造节点
Node node = new Node(Thread.currentThread(), mode);
// 队列不为空
Node pred = tail;
if (pred != null) {
//插入到队尾
node.prev = pred;
if (compareAndSetTail(pred, node)) {
pred.next = node;
return node;
}
}
//队列为空或者插入到队尾失败
enq(node);
return node;
}
2.2.4.1. 入队的操作
- enq
private Node enq(final Node node) {
//死循环直到入队成功
for (;;) {
Node t = tail;
//队列为空,那么初始化头节点。注意是new Node而不是当前node(即队头是个占位符)
if (t == null) {
if (compareAndSetHead(new Node()))
tail = head;
//队列不为空,插入到队尾
} else {
node.prev = t;
if (compareAndSetTail(t, node)) {
t.next = node;
return t;
}
}
}
}
2.2.5. 阻塞,等待唤醒继续获取锁
- acquireQueued
final boolean acquireQueued(final Node node, int arg) {
boolean failed = true;
try {
boolean interrupted = false;
//死循环直到获取锁成功
for (;;) {
//逻辑1.
//当前节点的前一个节点时头节点的时候(公平锁:即我的前面没有人等待获取锁),尝试获取锁
final Node p = node.predecessor();
if (p == head && tryAcquire(arg)) {
//获取锁成功后设置头节点为当前节点
setHead(node);
p.next = null; // help GC
failed = false;
return interrupted;
}
//逻辑2.
//当前节点的前一个节点状态时SIGNAL(承诺唤醒当前节点)的时候,阻塞当前线程。
//什么时候唤醒?释放锁的时候
//唤醒之后干什么?继续死循环执行上面的逻辑1
if (shouldParkAfterFailedAcquire(p, node) &&
parkAndCheckInterrupt())
interrupted = true;
}
} finally {
//何时执行这段逻辑?发生异常导致获取锁失败的时候
if (failed)
cancelAcquire(node);
}
}
2.2.5.1. 判断是否需要阻塞
- shouldParkAfterFailedAcquire
//根据(前一个节点,当前节点)->是否阻塞当前线程
private static boolean shouldParkAfterFailedAcquire(Node pred, Node node) {
int ws = pred.waitStatus;
//前一个节点的状态时SIGNAL,即释放锁后承诺唤醒当前节点,那么返回true可以阻塞当前线程
if (ws == Node.SIGNAL)
return true;
//前一个节点状态>0,即CANCEL。
//那么往前遍历找到没有取消的前置节点。同时从链表中移除CANCEL状态的节点
if (ws > 0) {
do {
node.prev = pred = pred.prev;
} while (pred.waitStatus > 0);
pred.next = node;
// 前置节点状态>=0,即0或者propagate。
//这里通过CAS把前置节点状态改成signal成功获取锁,失败的话再阻塞。why?
} else {
compareAndSetWaitStatus(pred, ws, Node.SIGNAL);
}
return false;
}
2.2.5.1.1. 阻塞当前线程
- parkAndCheckInterrupt
private final boolean parkAndCheckInterrupt() {
//使用Unsafe阻塞当前线程,这里会清除线程中断的标记,因此需要返回中断的标记
LockSupport.park(this);
return Thread.interrupted();
}
2.3. 解锁
public void unlock() {
//调用AQS的release方法
sync.release(1);
}
2.3.1. 使用AQS释放锁
- release
public final boolean release(int arg) {
//Sync重写了调用Sync释放锁成功
if (tryRelease(arg)) {
Node h = head;
//队头不为空且状态正常,那么唤醒头节点
if (h != null && h.waitStatus != 0)
unparkSuccessor(h);
return true;
}
return false;
}
Sync重写了tryRelease方法,因此这里调用的是Sync.tryRelease 其他的逻辑同5.AQS.md,下面只是简要说一下主要逻辑
2.3.2. 尝试释放锁
- Sync.tryRelease
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);
}
//CAS设置解锁
setState(c);
return free;
}
2.3.3. 释放锁成功后唤醒阻塞队列中的节点
- AQS.unparkSuccessor
private void unparkSuccessor(Node node) {
int ws = node.waitStatus;
//当前节点的状态<0,则把状态改为0
//0是空的状态,因为node这个节点的线程释放了锁后续不需要做任何
if (ws < 0)
compareAndSetWaitStatus(node, ws, 0);
//当前节点的下一个节点为空或者状态>0(即是取消状态)
Node s = node.next;
if (s == null || s.waitStatus > 0) {
s = null;
//那么从队尾开始往前遍历找到离当前节点最近的下一个状态<=0的节点(即非取消状态)
for (Node t = tail; t != null && t != node; t = t.prev)
if (t.waitStatus <= 0)
s = t;
}
//唤醒下一个节点(公平锁)
if (s != null)
LockSupport.unpark(s.thread);
}