深入解析AbstractQueuedSynchronizer源码2-独占模式

AbstractQueuedSynchronizer是juc包下面解决资源竞争的基础,功能主要包括三部分:
第一部分Condition监视器,已在Condition源码解析文章中做了分析。
第二部分独占模式,ReentrantLock,ReentrantReadWriteLock的写锁都是基于AbstractQueuedSynchronizer的独占锁实现的。
第三部分共享模式,ReentrantReadWriteLock的读锁,CountDownLatch,Semaphore都是基于AbstractQueuedSynchronizer的共享锁实现的。
AbstractQueuedSynchronizer实现由两个队列实现,一个Sync queue。一个Condition Queue,Condition Queue再第一篇文章中已介绍。
Sync queue是实现共享锁和独占锁的基础。
本文分析独占锁实现。

AbstractQueuedSynchronizer 类变量

    // 记录Sync Queue 的头
    private transient volatile Node head;

    /**
     * 记录Sync Queue 的尾部
     * method enq to add new wait node.
     */
    private transient volatile Node tail;

    // 同步状态,lock,计数器,信号箱的资源都是维护这个状态,类似于资源总数
    private volatile int state;

Node类

该类是两个队列实现的关键,Sync queue,Condition Queue里的节点都是Node节点

static final class Node {

       /**
       SHARED 和EXCLUSIVE 主要区分共享模式和独占模式
       **/
        static final Node SHARED = new Node();

        static final Node EXCLUSIVE = null;
        // 取消状态
        static final int CANCELLED =  1;
       // 该状态说明后继节点需要被唤醒,独占模式
        static final int SIGNAL    = -1;
       // 该状态说明节点是Condition下的节点,参考Condition解析
        static final int CONDITION = -2;
        // 该状态用于共享模式,无条件传播唤醒
        static final int PROPAGATE = -3;
        // waitStatus 表示node节点的状态,0初始结束状态
        volatile int waitStatus;
       // Sync Queue是双向链表
        volatile Node prev;
        volatile Node next;
        // node线程
        volatile Thread thread;
        // Condition queue 单向链表
        Node nextWaiter;

        final boolean isShared() {
            return nextWaiter == SHARED;
        }

        final Node predecessor() throws NullPointerException {
            Node p = prev;
            if (p == null)
                throw new NullPointerException();
            else
                return p;
        }

        Node() {    // Used to establish initial head or SHARED marker
        }

        Node(Thread thread, Node mode) {     // Used by addWaiter
            this.nextWaiter = mode;
            this.thread = thread;
        }

        Node(Thread thread, int waitStatus) { // Used by Condition
            this.waitStatus = waitStatus;
            this.thread = thread;
        }
    }

独占获取资源方法

独占获取资源的入口如下:

   // 忽略中断异常
    public final void acquire(int arg) {
        if (!tryAcquire(arg) &&
            acquireQueued(addWaiter(Node.EXCLUSIVE), arg))
            selfInterrupt();
    }
   // 抛出中断异常
    public final void acquireInterruptibly(int arg)
            throws InterruptedException {
        if (Thread.interrupted())
            throw new InterruptedException();
        if (!tryAcquire(arg))
            doAcquireInterruptibly(arg);
    }
    // 超时获取,并抛出中断异常
    public final boolean tryAcquireNanos(int arg, long nanosTimeout)
            throws InterruptedException {
        if (Thread.interrupted())
            throw new InterruptedException();
        return tryAcquire(arg) ||
            doAcquireNanos(arg, nanosTimeout);
    }

tryAcquire留给工具自己去实现,用于判断是否满足获取资源要求
本文重点分析acquireQueued,doAcquireInterruptibly,doAcquireNanos实现。

acquireQueued方法

final boolean acquireQueued(final Node node, int arg) {
        boolean failed = true;
        try {
            boolean interrupted = false;
            for (;;) {
                // 独占模式一次只能唤醒一个节点,即唤醒head的后继节点,这里判断节点是否是head的后继节点
                final Node p = node.predecessor();
                if (p == head && tryAcquire(arg)) {
                    // 如果是head的后继节点,说明没有竞争,直接把改节点设置成head节点,然后返回中断状态
                    setHead(node);
                    p.next = null; // help GC
                    failed = false;
                    return interrupted;
                }
                // 如果不是head的后继节点,说明存在竞争,需要封装改节点,然后等待被唤醒
                if (shouldParkAfterFailedAcquire(p, node) &&
                        parkAndCheckInterrupt())
                    interrupted = true;
            }
        } finally {
            if (failed)
                cancelAcquire(node);
        }
    }

shouldParkAfterFailedAcquire

    /**
     * 该方法检查是否满足节点被封装的要求
     * 即pred的状态是否是SIGNAL,SIGNAL代表后继节点需要被唤醒
     * 不满足要求,修改满足要求,返回false,再来一次
     */
    private static boolean shouldParkAfterFailedAcquire(Node pred, Node node) {
        int ws = pred.waitStatus;
        // 满足要求,返回true
        if (ws == Node.SIGNAL)
            return true;
        // 大于0表示节点的前驱节点被取消,需要跳过已被取消的所有节点
        if (ws > 0) {

            do {
                node.prev = pred = pred.prev;
            } while (pred.waitStatus > 0);
            pred.next = node;
        } else {

            // 否则把前驱节点设置成SIGNAL状态
            compareAndSetWaitStatus(pred, ws, Node.SIGNAL);
        }
        return false;
    }

parkAndCheckInterrupt

    // 该方法封装阻塞线程,runnable->waiting,被唤醒之后检查是否是中断唤醒
    private final boolean parkAndCheckInterrupt() {
        // 这里阻塞,线程状态由runnable变成waiting状态,直到被唤醒
        LockSupport.park(this);
        // 被唤醒之后,检查是否是中断唤醒
        return Thread.interrupted();
    }

cancelAcquire

//因异常取消获取节点
private void cancelAcquire(Node node) {
        // Ignore if node doesn't exist
        if (node == null)
            return;
        node.thread = null;                 // 1. 线程引用清空
        Node pred = node.prev;
        while (pred.waitStatus > 0)       // 2.  若前继节点是 CANCELLED 的, 则也一并清除
            node.prev = pred = pred.prev;
        Node predNext = pred.next;         // 3. 这里的 predNext也是需要清除的(只不过在清除时的 CAS 操作需要 它)
        node.waitStatus = Node.CANCELLED; // 4. 设置成清除状态
        if (node == tail && compareAndSetTail(node, pred)) { // 5. 若需要清除额节点是尾节点, 则直接 设置pred为尾节点,并删除predNext 
            compareAndSetNext(pred, predNext, null);   
        } else {
            int ws;
            if (pred != head &&
                    ((ws = pred.waitStatus) == Node.SIGNAL || // 6. 如果pred没被取消,设置pred的waitStatus==SIGNAL 表示后继节点需要唤醒
                            (ws <= 0 && compareAndSetWaitStatus(pred, ws, Node.SIGNAL))) && 
                    pred.thread != null) {
                Node next = node.next;
                if (next != null && next.waitStatus <= 0) // 7. next.waitStatus <= 0 表示 next 是个一个想要获取lock的节点
                    compareAndSetNext(pred, predNext, next);
            } else {
                unparkSuccessor(node); // 若 pred 是头节点, 直接唤醒下node的next节点。
            }
            node.next = node; // help GC
        }
    }

以上是独占方式获取资源的过程,其中doAcquireInterruptibly,doAcquireNanos两方法实现和acquireQueued实现区别不是很大,支持抛出中断异常,和超时等待,不做具体分析。

release

    public final boolean release(int arg) {
        if (tryRelease(arg)) {
            Node h = head;
            // 如果head不为空,并列有后继节点需要被唤醒,直接唤醒后继节点
            if (h != null && h.waitStatus != 0)
                unparkSuccessor(h);
            return true;
        }
        return false;
    }

unparkSuccessor

private void unparkSuccessor(Node node) {

    int ws = node.waitStatus;
    // 唤醒后继节点之前,先把node设置成最终状态0
    if (ws < 0)
        compareAndSetWaitStatus(node, ws, 0);


    // 拿到node的后继节点,如果被取消,设置成null,遍历,直到拿到需要被唤醒的节点
    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;
    }
    //不为空,唤醒
    if (s != null)
        LockSupport.unpark(s.thread);
}

以上就是独占模式下获取资源和释放资源的过程,之后会继续分析共享模式实现。

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

推荐阅读更多精彩内容