AbstractQueuedSynchronizer源码解读

   AbstractQueuedSynchronizer(AQS),是 Java 并发包中,实现各种同步结构和部分其他组成单元(如线程池中的 Worker)的基础。AQS 内部数据和方法,可以简单拆分为:

  • 一个 volatile 的int成员表征状态(state属性),同时提供了 setState 和 getState 方法。这里之所以用int不用boolean是因为有场景并不是只能有一个线程获取锁, 比如CountDownLatch就允许多个线程获取锁。
    1: ReentrantLock: 用state变量记录某个线程获取独占锁的次数,获取锁时+1,释放锁时-1,在获取时会校验线程是否可以获取锁。
    2:Semaphore: 用state变量作为计数器,只有在大于0时允许线程进入。获取锁时-1,释放锁时+1。
    3: CountDownLatch: 用state变量作为计数器,在初始化时指定。只要state还大于0,获取共享锁会因为失败而阻塞,直到计数器的值为0时,共享锁才允许获取,所有等待线程会被逐一唤醒。
  • 一个先入先出(FIFO)的等待线程队列,以实现多线程间竞争和等待,这是 AQS 机制的核心之一。
  • 各种基于 CAS 的基础操作方法,以及各种期望具体同步结构去实现的 tryAcquire/tryRelease 方法。


    AbstractQueuedSynchronizer关键属性

下面来看看acquire获取锁的部分源码:

public abstract class AbstractQueuedSynchronizer
    extends AbstractOwnableSynchronizer
    implements java.io.Serializable {
//定义为final说明子类不能覆盖这个实现过程, 之类只需要实现tryAcquire方法
    public final void acquire(int arg) {
//这里分别会调用tryAcquire方法用于获取锁,如果失败,会调用addWaiter方法,将当前线的Node加入到FIFO队列里,通过acquireQueued判断当前节点是否为头节点,如果是则试图获取锁, 获取不成功会加入等待队列, 并把线程挂起, 不是头结点的话直接加入等待队列并挂起线程。
        if (!tryAcquire(arg) &&
            acquireQueued(addWaiter(Node.EXCLUSIVE), arg))
            selfInterrupt();
    }

//这个方法在AbstractQueuedSynchronizer里留给子类来实现,
    protected boolean tryAcquire(int arg) {
        throw new UnsupportedOperationException();
    }

}

这里分别会调用tryAcquire方法用于获取锁,如果失败,会调用addWaiter方法,将当前线的Node加入到FIFO队列里,通过acquireQueued判断当前节点是否为头节点,如果是则试图获取锁, 获取不成功会加入等待队列, 并把线程挂起, 不是头结点的话直接加入等待队列并挂起线程。在AbstractQueuedSynchronizer里tryAcquire方法是留给子类来实现的,下面来看看ReentrantLock里的非公平锁是如何实现这里的逻辑的。源码如下:

abstract static class Sync extends AbstractQueuedSynchronizer {
      //非公平锁最终最调用到当前这个方法,传入的acquires为1
    final boolean nonfairTryAcquire(int acquires) {
        final Thread current = Thread.currentThread();
         int c = getState();// 获取当前 AQS 内部状态量
          if (c == 0) { // 0 表示无人占有,则直接用 CAS 修改状态位,
                if (compareAndSetState(0, acquires)) {// 不检查排队情况,直接争抢
                      setExclusiveOwnerThread(current);  // 并设置当前线程独占锁
                    return true;
                }
          } else if (current == getExclusiveOwnerThread()) { // 即使状态不是 0,也可能当前线程是锁持有者,因为这是重入锁
                int nextc = c + acquires;
                  if (nextc < 0) // overflow
                throw new Error("Maximum lock count exceeded");
                setState(nextc);
                return true;
    }
    return false;
}


}

ReentrantLock的公平锁的实现逻辑如下:

   static final class FairSync extends Sync {
//公平锁的实现逻辑
        protected final boolean tryAcquire(int acquires) {
            final Thread current = Thread.currentThread();
            int c = getState();// 获取当前 AQS 内部状态量
            if (c == 0) {// 0 表示无人占有,
                if (!hasQueuedPredecessors() && //这里需要判断队列头节点的的下一个节点为空。也就是说不允许线程进行插队。
                    compareAndSetState(0, acquires)) { //为空才会尝试获取锁
                    setExclusiveOwnerThread(current);
                    return true;
                }
            }
            else if (current == getExclusiveOwnerThread()) { // 即使状态不是 0,也可能当前线程是锁持有者,因为这是重入锁
                int nextc = c + acquires;
                if (nextc < 0)
                    throw new Error("Maximum lock count exceeded");
                setState(nextc);
                return true;
            }
            return false;
        }
    }

接下来再分析 acquireQueued,如果前面的 tryAcquire 失败,代表着锁争抢失败,进入排队竞争阶段。这里就是利用 FIFO 队列,实现线程间对锁的竞争的部分,算是AQS 的核心逻辑。源码如下:

//这个方法是不允许子类重写的
    final boolean acquireQueued(final Node node, int arg) {
        boolean failed = true;
        try {
            boolean interrupted = false;
            for (;;) {
                final Node p = node.predecessor(); //获取前一个节点
                if (p == head && tryAcquire(arg)) { / 如果前一个节点是头结点,则当前节点是FIFO队列里的第一个节点,则会再次调用tryAcquire方法尝试获取锁。
                    setHead(node); //成功,则设置当前节点为头节点
                    p.next = null; // help GC 清空前面节点对当前节点的引用
                    failed = false;
                    return interrupted;
                }
                if (shouldParkAfterFailedAcquire(p, node) && //获取锁失败后, 需要判断当前线程是否不要挂起, 正常情况下是都会需要挂起的
                    parkAndCheckInterrupt())
                    interrupted = true; //这里需要设置一个标识位, 是因为park方法无法感知到其他线程调用了当前线程的interrupt方法改变了现场的interrupt状态
            }
        } finally {
            if (failed) // 获取锁失败,调用cancelAcquire方法
                cancelAcquire(node);
        }
    }

acquireQueued 的逻辑,简要来说,就是如果当前节点的前面是头节点(AQS里的头结点是哨兵结点),则试图获取锁,一切顺利则成为新的头节点;否则,会挂起线程。上面的部分分析了获取锁的逻辑。下面再来看看释放锁的逻辑AbstractQueuedSynchronizer的relase源码如下:

//释放锁需要调用release方法
    public final boolean release(int arg) {
        if (tryRelease(arg)) {
//释放成功后,调用unparkSuccessor方法,让下个节点得到锁
            Node h = head;
            if (h != null && h.waitStatus != 0)
                unparkSuccessor(h);
            return true;
        }
        
//留给具体的子类来实现
    protected boolean tryRelease(int arg) {
        throw new UnsupportedOperationException();
    }

//当前节点释放锁,下个节点得到锁状态 
    private void unparkSuccessor(Node node) {
        int ws = node.waitStatus;
//设置当前节点的waitStatus
        if (ws < 0)
            compareAndSetWaitStatus(node, ws, 0);
//当前节点的下一个节点
        Node s = node.next;
//下一个节点为空情况的处理
        if (s == null || s.waitStatus > 0) {
            s = null;
            for (Node t = tail; t != null && t != node; t = t.prev)
                if (t.waitStatus <= 0)
                    s = t;
        }
//下一个节点不为空,则调用LockSupport的unpark方法唤醒线程
        if (s != null)
            LockSupport.unpark(s.thread);
    }

ReentrantLock里tryRelease的实现如下所示:

        protected final boolean tryRelease(int releases) {
            int c = getState() - releases; //用state减去release, 这个状态是新的状态 
            if (Thread.currentThread() != getExclusiveOwnerThread())
                throw new IllegalMonitorStateException();
            boolean free = false; //释放的结果用free来表示
            if (c == 0) { //为零,表求当前没有钱程点用了,
                free = true;
                setExclusiveOwnerThread(null);
            }
  //设置状态 
            setState(c);
            return free;
        }

Node的状态

java.util.concurrent.locks.AbstractQueuedSynchronizer.Node

    /** 表示线程已取消 */
    static final int CANCELLED =  1;
    /** 表示线程等待唤醒 */
    static final int SIGNAL    = -1;
    /** 表示线程等待获取同步锁 */
    static final int CONDITION = -2;
    /** 表示共享模式下无条件传播 */
    static final int PROPAGATE = -3;

CANCELLED(1):表示当前结点已取消调度。当timeout或被中断(响应中断的情况下),会触发变更为此状态,进入该状态后的结点将不会再变化。
SIGNAL(-1):表示后继结点在等待当前结点唤醒。后继结点入队时,会将前继结点的状态更新为SIGNAL。
CONDITION(-2):表示结点等待在Condition上,当其他线程调用了Condition的signal()方法后,CONDITION状态的结点将从等待队列转移到同步队列中,等待获取同步锁。
PROPAGATE(-3):共享模式下,前继结点不仅会唤醒其后继结点,同时也可能会唤醒后继的后继结点。
0:新结点入队时的默认状态。
注意,负值表示结点处于有效等待状态,而正值表示结点已被取消。所以源码中很多地方用>0、<0来判断结点的状态是否正常。

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 158,233评论 4 360
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 67,013评论 1 291
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 108,030评论 0 241
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 43,827评论 0 204
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 52,221评论 3 286
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 40,542评论 1 216
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 31,814评论 2 312
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 30,513评论 0 198
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 34,225评论 1 241
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 30,497评论 2 244
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 31,998评论 1 258
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 28,342评论 2 253
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 32,986评论 3 235
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 26,055评论 0 8
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 26,812评论 0 194
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 35,560评论 2 271
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 35,461评论 2 266

推荐阅读更多精彩内容