1

上篇文章 AQS系列一:源码分析非公平ReentrantLock 中,我们分析了ReentrantLock的非公平实现,本篇会承接上文,继续分析ReentrantLock的公平锁实现(以及Condition的实现)。

在此之前我们要先弄明白,“不公平”体现在哪里。

为何“不公”

好吧,我也不清楚。
于是我对比了ReentrantLock的非公平和公平实现,即NonfairSync vs FairSync,发现差别主要体现在加锁,更确切的说是获取锁环节

## 非公平获取锁
final boolean nonfairTryAcquire(int acquires) {
    ...
    if (c == 0) {
        if (compareAndSetState(0, acquires)) {
            setExclusiveOwnerThread(current);
            return true;
        }
    }
    ...
}
## 公平获取锁
protected final boolean tryAcquire(int acquires) {
    ...
    if (!hasQueuedPredecessors() //### 差别体现在此处
        && compareAndSetState(0, acquires)) {
        setExclusiveOwnerThread(current);
        return true;
    }
    ...
}

显然,公平锁多执行了!hasQueuedPredecessors(),看看此方法的逻辑。

public final boolean hasQueuedPredecessors() {
    Node h, s;
    if ((h = head) != null) { ## h头结点
        if ((s = h.next) == null ... ) { ## s二号节点
            ...
        }
        if (s != null && s.thread != Thread.currentThread()) ##检查2号节点绑定线程,是否当前线程
            return true;
    }
    return false;
}

hasQueuedPredecessors方法只有在 2号节点不为空,且绑定线程非当前线程的前提下,会返回true
返回ture意味着!hasQueuedPredecessors() = false,没有资格获取锁(就是没机会执行compareAndSetState——尝试修改state)

反过来讲,没有队列(无线程正在执行),或者没有2号节点(取消或者临时状态),再或者2号节点的绑定线程就是当前线程时,才会尝试获取锁

clipboard.png

我们分析下最后这种情况,2号节点绑定的线程是第1个等待的线程(第1个获取锁失败的线程),第1个等待的线程在hasQueuedPredecessors()的运作下,成为了第1个有资格尝试获取锁的线程。而这,就是公平

那么没有hasQueuedPredecessors方法的非公平锁,到底“不公平”在哪儿呢?
我们回想一下,在加 / 解锁的过程中,nonfairTryAcquire方法被调用的位置就能得到答案了。

public final void acquire(int arg) {
    if (!tryAcquire(arg) ### 位置1,尝试获取
        && acquireQueued(addWaiter(Node.EXCLUSIVE), arg))
        selfInterrupt();
}

final boolean acquireQueued(final Node node, int arg) {
    boolean interrupted = false;
    ...
        for (;;) {
            final Node p = node.predecessor();
            if (p == head && tryAcquire(arg)) { ### 位置2,尝试获取
                setHead(node);
                p.next = null; // help GC
                return interrupted;
            }
            if (shouldParkAfterFailedAcquire(p, node))
                interrupted |= parkAndCheckInterrupt();
        }
    ...
}

在上述代码中,tryAcquire(非公平实现会调用nonfairTryAcquire)会在位置1、2两处触发。试想如下场景:

  • 线程T-3执行完毕,调用了unlock;随着线程T-2被唤醒,位置2处代码可能会被执行。
  • 与此同时,随着新的线程T-1的介入,位置1处的代码也有可能被执行。

因此线程T-2T-1谁能在并发中抢到锁,存在不确定性

  • 如果线程T-2先执行了,T-1失败于位置1处,后续会阻塞于队列尾部;
  • 如果线程T-1先执行了,T-2失败于位置2处,面临又一轮阻塞,这种情况就不怎么“公平”——新来的线程T-1抢先了!

原理说完了,那具体怎么构建公平的ReentrantLock呢?构造函数传参即可:

public ReentrantLock(boolean fair) {
    sync = fair ? new FairSync() : new NonfairSync();  ## 入参fair传入true,构建公平锁
}

Condition解析

使用

ReentrantLock的加解锁过程已详细分析了一遍,如果你经常使用这个工具,肯定对衍生出另一个大咖condition有所了解。
二话不说,先甩出demo:

static Lock lock = new ReentrantLock();
Condition condition = lock.newCondition();;

public void doSomething(){
    lock.lock();
    System.out.println(String.format("%s线程,获取到锁了",Thread.currentThread().getName()));
    try {
        System.out.println(String.format("%s线程,await",Thread.currentThread().getName()));
        TimeUnit.SECONDS.sleep(2L); //模拟耗时业务逻辑执行
        condition.await();  //await
        System.out.println(String.format("%s线程,await被唤醒",Thread.currentThread().getName()));
    } catch (InterruptedException e) {
        e.printStackTrace();
    }
    System.out.println(String.format("%s线程,业务执行完毕",Thread.currentThread().getName()));
    lock.unlock();
}

public static void main(String[] args) throws InterruptedException {
    ReentrantLockTest test = new ReentrantLockTest();
    int total = 1;
    while (total>0){
        Thread t = new Thread(()->{
            test.doSomething();
        },"T-"+total);
        t.start();

        TimeUnit.MILLISECONDS.sleep(200L);  //让子线程T-1率先获取到锁
        lock.lock();
        System.out.println(String.format("%s线程,获取到锁了",Thread.currentThread().getName()));
        test.condition.signal();
        System.out.println(String.format("%s线程,signal",Thread.currentThread().getName()));
        lock.unlock();
        total--;
    }
}

结合已掌握的加解锁原理,分析demo执行过程:

  • 人为控制让子线程T-1先获取到锁,200ms后main线程也会尝试获取锁,当然main线程获取不到——由于耗时达2s的业务逻辑疯狂执行中。(sleep处,此时main线程应该构建了同步队列,main线程作为2号节点的绑定线程被无情阻塞,下图)

clipboard.png

  • 2s后,线程T-1搞定了难缠的业务逻辑,却又遭遇condition.await()的伏击
  • 此时,线程main发现自己神奇的不被阻塞了,又神奇的获取到了锁。于是投桃报李,condition.signal()unlock二连招唤醒了线程T-1
  • 线程T-1觉醒于await处,执行完剩余逻辑

demo的执行结果,能初步证明上述分析:

T-1线程,获取到锁了
T-1线程,await
main线程,获取到锁了
main线程,signal
T-1线程,await被唤醒
T-1线程,业务执行完毕

原理

构造器

从构造函数出发:

public Condition newCondition() {
    return sync.newCondition();
}

## Sync类创建ConditionObject
final ConditionObject newCondition() {
    return new ConditionObject();
}

ConditionObject是AQS中的另一内部类,看看它的属性:

## ConditionObject类
private transient Node firstWaiter;
private transient Node lastWaiter;

感觉上和AQS的设定上有些像?

## AQS类
private transient volatile Node head;
private transient volatile Node tail;

先大胆猜测一下,condition中很可能会再次构建同步队列。

await()

接下来就是验证我们的猜测的过程:

public final void await() throws InterruptedException {
    if (Thread.interrupted())
        throw new InterruptedException();
    Node node = addConditionWaiter();    ## 创建等待队列,node是尾节点。    详情参看[addConditionWaiter详情]
    int savedState = fullyRelease(node);    ## 重置state,返回重置前的state值。    详情参看[fullyRelease详情]
    int interruptMode = 0;
    while (!isOnSyncQueue(node)) {    ## 是否在AQS同步队列中
        LockSupport.park(this);    ## 不在AQS同步队列的节点,阻塞当前线程
        if ((interruptMode = checkInterruptWhileWaiting(node)) != 0)
            break;
    }
    if (acquireQueued(node, savedState) && interruptMode != THROW_IE)
        interruptMode = REINTERRUPT;
    if (node.nextWaiter != null) // clean up if cancelled
        unlinkCancelledWaiters();
    if (interruptMode != 0)
        reportInterruptAfterWait(interruptMode);
}
  • addConditionWaiter详情
private Node addConditionWaiter() {
    if (!isHeldExclusively())    ## 当前线程是否owner线程,如果不是,抛异常——这儿决定了await必须用在lock()方法之后
        throw new IllegalMonitorStateException();
    Node t = lastWaiter;
    // If lastWaiter is cancelled, clean out.
    if (t != null && t.waitStatus != Node.CONDITION) {
        unlinkCancelledWaiters();
        t = lastWaiter;
    }

    Node node = new Node(Node.CONDITION); ## 创建新节点,原子形赋值waitStatus=CONDITION=-2,并绑定当前线程到node节点

    ## node会作为尾节点,置于队列最后
    if (t == null)
        firstWaiter = node;
    else
        t.nextWaiter = node;
    lastWaiter = node;
    return node; 
}
  • fullyRelease详情
final int fullyRelease(Node node) {
    try {
        int savedState = getState();    ## 获取当前state
        if (release(savedState))
            return savedState;
        throw new IllegalMonitorStateException();
    } catch (Throwable t) {
        node.waitStatus = Node.CANCELLED;
        throw t;
    }
}

public final boolean release(int arg) {
    if (tryRelease(arg)) {    ## 尝试“清0”state
        Node h = head;    ## 此处head不为空,unpark线程main,return true
        if (h != null && h.waitStatus != 0)
            unparkSuccessor(h);
        return true;
    }
    return false;
}

protected final boolean tryRelease(int releases) {
    int c = getState() - releases;
    if (Thread.currentThread() != getExclusiveOwnerThread())    ## 当前线程验证,如果当前线程!=owner,抛异常
        throw new IllegalMonitorStateException();
    boolean free = false;
    if (c == 0) {    ## 如果state清0,同时清空owner线程,return true
        free = true;
        setExclusiveOwnerThread(null);
    }
    setState(c);
    return free;
}

有了之前分析ReentrantLock的经验,与之非常相像的condition代码应该不难拿下。
这里画出await方法中fullyRelease(node)执行前后的节点和关键属性的变化:

clipboard.png

图右侧(await方法执行到了LockSupport.park(this)时),线程T-1已经阻塞,线程main则解除阻塞状态。

通过上图很容易看出,我们之前的猜测是正确的:await方法又构建了一个同步队列,不过这次的头、尾指针在ConditionObject类中

signal()

再来看看signal方法作了什么:

public final void signal() {
    if (!isHeldExclusively())    ## 和await()方法中的一样,先验证owner线程
        throw new IllegalMonitorStateException();
    Node first = firstWaiter;
    if (first != null)
        doSignal(first);
}

private void doSignal(Node first) {
    do {
        if ( (firstWaiter = first.nextWaiter) == null)
            lastWaiter = null;
        first.nextWaiter = null;
    } while (!transferForSignal(first)    ## condition头结点传递
             && (first = firstWaiter) != null);
}

final boolean transferForSignal(Node node) {
    if (!node.compareAndSetWaitStatus(Node.CONDITION, 0))
        return false;
    Node p = enq(node);    ## 将ConditionObject头结点移动到AQS队列尾部。    详情参看[enq详情]
    int ws = p.waitStatus;
    if (ws > 0 || !p.compareAndSetWaitStatus(ws, Node.SIGNAL))    ## 取消或修改waitStatus失败才作unpark操作,此处unpark不会触发
        LockSupport.unpark(node.thread);
    return true;
}
  • enq详情
private Node enq(Node node) {
    for (;;) {
        Node oldTail = tail;
        if (oldTail != null) {
            node.setPrevRelaxed(oldTail);    ## 入参node,成为了AQS队列新的尾节点
            if (compareAndSetTail(oldTail, node)) {
                oldTail.next = node;
                return oldTail;
            }
        } else {
            initializeSyncQueue();    ## 初始化AQS队列
        }
    }
}

signal中最神奇的莫过于enq(node)方法,它完成了节点的转移,condition队列头结点 -> AQS队列尾节点。

通过下图观察整个signal方法产生的各对象结构和属性变化:

image.png

观察可知,signal执行后节点转移已经完成,线程T-1依然阻塞,此时ConditionObject已经完成了它的历史使命。

线程T-1什么时候解除阻塞呢?其实这部分上篇文章已经分析过了,就是我们的老朋友unlock()。
区别在于线程T-1被唤醒后,执行的是await后续的逻辑:

public final void await() throws InterruptedException {
    if (Thread.interrupted())
        throw new InterruptedException();
    Node node = addConditionWaiter();
    int savedState = fullyRelease(node);
    int interruptMode = 0;
    while (!isOnSyncQueue(node)) {    ## 2.下次循环,node已经在AQS队列中,返回true,跳出循环
        LockSupport.park(this);    ## 1.线程T-1觉醒于此
        if ((interruptMode = checkInterruptWhileWaiting(node)) != 0)
            break;
    }
    if (acquireQueued(node, savedState) ## 3.再度获取到锁
            && interruptMode != THROW_IE)    
        interruptMode = REINTERRUPT;
    if (node.nextWaiter != null) // clean up if cancelled
        unlinkCancelledWaiters();
    if (interruptMode != 0)
        reportInterruptAfterWait(interruptMode);
}

后记

至此,我们已经了解了ReentrantLock的主逻辑的源码实现(公平、非公平、condition),本系列的下篇文章将进入下一副本——CountDownLatch,敬请期待!


青鱼
268 声望25 粉丝

山就在那里,每走一步就近一些