zoukankan      html  css  js  c++  java
  • 一行一行源码分析清楚AbstractQueuedSynchronizer(一)

    在分析 Java 并发包 java.util.concurrent 源码的时候,少不了需要了解 AbstractQueuedSynchronizer(以下简写AQS)这个抽象类,因为它是 Java 并发包的基础工具类,是实现 ReentrantLock、CountDownLatch、Semaphore、FutureTask 等类的基础。

    本文将从 ReentrantLock 的公平锁源码出发,分析下 AbstractQueuedSynchronizer 这个类是怎么工作的,希望能给大家提供一些简单的帮助。

    申明以下几点:

    1. 本文有点长,但还是挺简单,主要面向读者对象为并发编程的初学者,或者想要阅读 Java 并发包源码的开发者。对于新手来说,可能需要花好几个小时才能完全看懂,但是这时间肯定是值得的。
    2. 源码环境 JDK1.7(1.8没啥变化),看到不懂或有疑惑的部分,最好能自己打开源码看看。Doug Lea 大神的代码写得真心不错
    3. 本文不分析共享模式,这样可以给读者减少很多负担,第三篇对共享模式进行了分析。而且也不分析 condition 部分,所以应该说很容易就可以看懂了。
    4. 本文大量使用我们平时用得最多的 ReentrantLock 的概念,本质上来说是不正确的,读者应该清楚,AQS 不仅仅用来实现可重入锁,只是希望读者可以用锁来联想 AQS 的使用场景,降低阅读压力。
    5. ReentrantLock 的公平锁和非公平锁只有一点点区别,第二篇将做了介绍。

    一、AQS的属性

    先看一下AQS有哪些重要的属性,搞清楚这些就知道AQS是什么套路了。

    //等待队列的头结点,你可以认为当前持有锁的线程节点
    private transient volatile Node head;
    //d队列的尾结点,每个新进来的节点进来,插入到最后,就形成链表
    private transient volatile Node tail;
    // 这个是最重要的,代表当前锁的状态,0代表没有被占用,大于 0 代表有线程持有当前。这个值是可以大于1的,是因为锁可以重入,每次重入都加1
    private volatile int state;
    //代表当前持有独占锁的线程。举个使用例子,因为锁可以重入reentrantLock.lock()可以嵌套调用多次,所以每次用这个来判断当前线程是否已经拥有了锁
    private transient Thread exclusiveOwnerThread; //继承自AbstractOwnableSynchronizer
    

    本章先介绍这四个重要的属性,看上去很简单吧!

    AQS的等待队列结构如下所示,注意了,之后分析过程中所说的 queue,也就是阻塞队列不包含 head,不包含 head,不包含 head

    等待队列中每个线程被包装成一个 Node 实例,数据结构是链表,一起看看源码吧:

    static final class Node {
        // 标识节点当前在共享模式下
        static final Node SHARED = new Node();
        // 标识节点当前在独占模式下
        static final Node EXCLUSIVE = null;
    
        // ======== 下面的几个int常量是给waitStatus用的 ===========
        /** waitStatus value to indicate thread has cancelled */
        // 代表此线程取消了争抢这个锁
        static final int CANCELLED =  1;
        /** waitStatus value to indicate successor's thread needs unparking */
        // 官方的描述是,其表示当前node的后继节点对应的线程需要被唤醒
        static final int SIGNAL    = -1;
        /** waitStatus value to indicate thread is waiting on condition */
        // 本文不分析condition,所以略过吧,下一篇文章会介绍这个
        static final int CONDITION = -2;
        /**
         * waitStatus value to indicate the next acquireShared should
         * unconditionally propagate
         */
        // 同样的不分析,略过吧
        static final int PROPAGATE = -3;
        // =====================================================
    
    
        // 取值为上面的1、-1、-2、-3,或者0(以后会讲到)
        // 这么理解,暂时只需要知道如果这个值 大于0 代表此线程取消了等待,
        //    ps: 半天抢不到锁,不抢了,ReentrantLock是可以指定timeouot的。。。
        volatile int waitStatus;
        // 前驱节点的引用
        volatile Node prev;
        // 后继节点的引用
        volatile Node next;
        // 这个就是线程本尊
        volatile Thread thread;
    
    }
    

    Node 的数据结构其实也挺简单的,就是 thread 、 waitStatus 、 prev 、next 四个属性而已。

    下面我们开始说 ReentrantLock 的公平锁。

    二、ReentrantLock 的公平锁

    我们先来看一个ReetrantLock的使用方式:

    public class ReetreetLockTest {
    
        private static ReentrantLock reentrantLock = new ReentrantLock();
    
        public static void main(String[] args) {
            reentrantLock.lock();
            try {
                Thread.sleep(2000);
            } catch (InterruptedException e) {
                e.printStackTrace();
            } finally {
                reentrantLock.unlock();
            }
    
        }
    }
    

    ReentrantLock 在内部用了内部类 Sync 来管理锁,所以真正的获取锁和释放锁是由 Sync 的实现类来控制的。

    abstract static class Sync extends AbstractQueuedSynchronizer {
    }
    

    Sync 有两个实现,分别为 NonfairSync(非公平锁)和 FairSync(公平锁),我们看 FairSync 部分。

    public ReentrantLock(boolean fair) {
        sync = fair ? new FairSync() : new NonfairSync();
    }
    

    三、线程抢锁

    介绍完上面的基础,我们大致对ReetrantLock有了一些了解,接下来就跟代码走:首先调用lock方法抢锁:

    static final class FairSync extends Sync {
        private static final long serialVersionUID = -3000897897090466540L;
    
        //抢占锁
        final void lock() {
            acquire(1);
        }
    
        //来自AQS类中的方法,直接贴到这里来,便于分析。
        //这个方法中首先会调用tryAcquire方法,如果返回true,方法就结束了。否则acquireQueued方法会把当前线程放到队列进行排队等待
        public final void acquire(int arg) {
            //首先调用tryAcquire,尝试获取一下,因为有可能直接成功就不需要排队了。对公平锁来说:本来没有人持有锁,根本没必要进入队列等待(又是挂起,又是等待唤醒)。
            if (!tryAcquire(arg) &&
                acquireQueued(addWaiter(Node.EXCLUSIVE), arg))
                selfInterrupt();
        }
    
        /**
             * Fair version of tryAcquire.  Don't grant access unless
             * recursive call or no waiters or is first.
             */
        //尝试获取锁的逻辑,boolean代表是否获取到锁
        //返回true: 1、没有线程在等待获取到锁,2、重入锁,线程本身就持有锁,也就可以理所当然的能够获取到锁。
        protected final boolean tryAcquire(int acquires) {
            final Thread current = Thread.currentThread();
            int c = getState();
            //state==0 此时此刻没有线程持有锁
            if (c == 0) {
                //此时此刻锁是可用的,但是这是公平锁,讲究先来后到,先看看是不是有人在等着呢。hasQueuedPredecessors方法看看队列中是否有其他线程
                if (!hasQueuedPredecessors() &&
                    //如果没有人等,则用CAS试一下,如果成功则获取到锁,不成功则表示几乎同时有另外的线程抢先获取到了锁,因为刚刚判断过没人获取到呢
                    compareAndSetState(0, acquires)) {
                    //到这里已经获取到锁了,标记一下自己
                    setExclusiveOwnerThread(current);
                    返回结果
                    return true;
                }
            }
            //state != 0 先判断一下是不是自己持有了锁,如果进入这个分支,则表示是重入,状态值需要加1
            else if (current == getExclusiveOwnerThread()) {
                int nextc = c + acquires;
                if (nextc < 0)
                    throw new Error("Maximum lock count exceeded");
                setState(nextc);
                return true;
            }
            //到了这里说明前面都没有返回true,说明没有获取到锁,则返回到前面的方法看看
            //if (!tryAcquire(arg) &&
            //    acquireQueued(addWaiter(Node.EXCLUSIVE), arg))
            //    selfInterrupt();
            return false;
        }
        
        //假如tryAcquire方法返回了false,接着要执行acquireQueued(addWaiter(Node.EXCLUSIVE), arg),首先要执行addWaiter(Node.EXCLUSIVE)方法。
        
        
        /**
         * Creates and enqueues node for current thread and given mode.
         *
         * @param mode Node.EXCLUSIVE for exclusive, Node.SHARED for shared
         * @return the new node
         */
        //此方法把线程包装成Node,并放入到队列之中
        // 参数mode此时是Node.EXCLUSIVE,代表独占模式
        private Node addWaiter(Node mode) {
            Node node = new Node(Thread.currentThread(), mode);
            // Try the fast path of enq; backup to full enq on failure
            // 以下几行代码想把当前node加到链表的最后面去,也就是进到阻塞队列的最后
            Node pred = tail;
            // tail!=null => 队列不为空(tail==head的时候,其实队列是空的,不过不管这个吧)
            if (pred != null) {
                // 将当前的队尾节点,设置为自己的前驱 
                node.prev = pred;
                // 用CAS把自己设置为队尾, 如果成功后,tail == node 了,这个节点成为阻塞队列新的尾巴
                if (compareAndSetTail(pred, node)) {
                    pred.next = node;
                    // 线程入队了,可以返回了
                    return node;
                }
            }
            // 仔细看看上面的代码,如果会到这里,
            // 说明 pred==null(队列是空的) 或者 CAS失败(有线程在竞争入队)
            enq(node);
            return node;
        }
        
        /**
         * Inserts node into queue, initializing if necessary. See picture above.
         * @param node the node to insert
         * @return node's predecessor
         */
        //采用自旋的方式入队
        //之前说过,到这个方法只有两种可能:等待队列为空,或者有线程竞争入队,
        // 自旋在这边的语义是:CAS设置tail过程中,竞争一次竞争不到,我就多次竞争,总会排到的
        private Node enq(final Node node) {
            for (;;) {
                Node t = tail;
                // 之前说过,队列为空也会进来这里
                if (t == null) { // Must initialize
                    // 初始化head节点
                    // 细心的读者会知道原来 head 和 tail 初始化的时候都是 null 的
                    // 还是一步CAS,你懂的,现在可能是很多线程同时进来呢
                    if (compareAndSetHead(new Node()))
                         // 给后面用:这个时候head节点的waitStatus==0, 看new Node()构造方法就知道了
    
                        // 这个时候有了head,但是tail还是null,设置一下,
                        // 把tail指向head,放心,马上就有线程要来了,到时候tail就要被抢了
                        // 注意:这里只是设置了tail=head,这里可没return哦,没有return,没有return
                        // 所以,设置完了以后,继续for循环,下次就到下面的else分支了
                        tail = head;
                } else {
                    // 下面几行,和上一个方法 addWaiter 是一样的,
                    // 只是这个套在无限循环里,反正就是将当前线程排到队尾,有线程竞争的话排不上重复排
                    node.prev = t;
                    if (compareAndSetTail(t, node)) {
                        t.next = node;
                        //返回的是原来的尾结点
                        return t;
                    }
                }
            }
        }
        
        
        // 现在,又回到这段代码了
        // if (!tryAcquire(arg) 
        //        && acquireQueued(addWaiter(Node.EXCLUSIVE), arg)) 
        //     selfInterrupt();
        
        // 下面这个方法,参数node,经过addWaiter(Node.EXCLUSIVE),此时已经进入阻塞队列
        // 注意一下:如果acquireQueued(addWaiter(Node.EXCLUSIVE), arg))返回true的话,
        // 意味着上面这段代码将进入selfInterrupt(),所以正常情况下,下面应该返回false
        // 这个方法非常重要,应该说真正的线程挂起,然后被唤醒后去获取锁,都在这个方法里
        final boolean acquireQueued(final Node node, int arg) {
            boolean failed = true;
            try {
                boolean interrupted = false;
                for (;;) {
                    final Node p = node.predecessor();
                    //如果前置是head ,说明当前节点虽然进到了阻塞队列,但是是阻塞队列的第一个,因为它的前驱是head
                    // 注意,阻塞队列不包含head节点,head一般指的是占有锁的线程,head后面的才称为阻塞队列
                    // 所以当前节点可以去试抢一下锁
                    // 这里我们说一下,为什么可以去试试:
                    // 首先,它是队头,这个是第一个条件,其次,当前的head有可能是刚刚初始化的node,
                    // enq(node) 方法里面有提到,head是延时初始化的,而且new Node()的时候没有设置任何线程
                    // 也就是说,当前的head不属于任何一个线程,所以作为队头,可以去试一试,
                    // tryAcquire已经分析过了, 忘记了请往前看一下,就是简单用CAS试操作一下state
                    if (p == head && tryAcquire(arg)) {
                        setHead(node);
                        p.next = null; // help GC
                        failed = false;
                        return interrupted;
                    }
                    //上面的if分支没有成功,要么当前node本来就不是队头, 要么就是tryAcquire(arg)没有抢到,继续往下看
                    if (shouldParkAfterFailedAcquire(p, node) &&
                        parkAndCheckInterrupt())
                        interrupted = true;
                }
            } finally {
                if (failed)
                    cancelAcquire(node);
            }
        }
        
        
        
        /**
         * Checks and updates status for a node that failed to acquire.
         * Returns true if thread should block. This is the main signal
         * control in all acquire loops.  Requires that pred == node.prev.
         *
         * @param pred node's predecessor holding status
         * @param node the node
         * @return {@code true} if thread should block
         */
        // 刚刚说过,会到这里就是没有抢到锁呗,这个方法说的是:"当前线程没有抢到锁,是否需要挂起当前线程?"
        // 第一个参数是前驱节点,第二个参数才是代表当前线程的节点
        private static boolean shouldParkAfterFailedAcquire(Node pred, Node node) {
            int ws = pred.waitStatus;
            // 前驱节点的 waitStatus == -1 ,说明前驱节点状态正常,当前线程需要挂起,直接可以返回true
            if (ws == Node.SIGNAL)
                /*
                 * This node has already set status asking a release
                 * to signal it, so it can safely park.
                 */
                return true;
            if (ws > 0) {
                 // 前驱节点 waitStatus大于0 ,之前说过,大于0 说明前驱节点取消了排队。
            // 这里需要知道这点:进入阻塞队列排队的线程会被挂起,而唤醒的操作是由前驱节点完成的,也就是前驱节点的waitstatus需要等于-1
            // 所以下面这块代码说的是将当前节点的prev指向waitStatus<=0的节点,
            // 简单说,就是为了找个好爹,因为你还得依赖它来唤醒呢,如果前驱节点取消了排队,
            // 找前驱节点的前驱节点做爹,往前遍历总能找到一个好爹的
                /*
                 * Predecessor was cancelled. Skip over predecessors and
                 * indicate retry.
                 */
                do {
                    node.prev = pred = pred.prev;
                } while (pred.waitStatus > 0);
                pred.next = node;
            } else {
                // 仔细想想,如果进入到这个分支意味着什么
                // 前驱节点的waitStatus不等于-1和1,那也就是只可能是0,-2,-3
                // 在我们前面的源码中,都没有看到有设置waitStatus的,所以每个新的node入队时,waitStatu都是0
                // 正常情况下,前驱节点是之前的 tail,那么它的 waitStatus 应该是 0
                // 用CAS将前驱节点的waitStatus设置为Node.SIGNAL(也就是-1)
                /*
                 * waitStatus must be 0 or PROPAGATE.  Indicate that we
                 * need a signal, but don't park yet.  Caller will need to
                 * retry to make sure it cannot acquire before parking.
                 */
                compareAndSetWaitStatus(pred, ws, Node.SIGNAL);
            }
            // 这个方法返回 false,那么会再走一次 for 循序,然后再次进来此方法,此时会从第一个分支返回 true
            return false;
        }
        
        
        // private static boolean shouldParkAfterFailedAcquire(Node pred, Node node)
        // 这个方法结束根据返回值我们简单分析下:
        // 如果返回true, 说明前驱节点的waitStatus==-1,是正常情况,那么当前线程需要被挂起,等待以后被唤醒
        //        我们也说过,以后是被前驱节点唤醒,就等着前驱节点拿到锁,然后释放锁的时候叫你好了
        // 如果返回false, 说明当前不需要被挂起,为什么呢?往后看
    
        // 跳回到前面是这个方法
        // if (shouldParkAfterFailedAcquire(p, node) &&
        //                parkAndCheckInterrupt())
        //                interrupted = true;
    
        // 1. 如果shouldParkAfterFailedAcquire(p, node)返回true,
        // 那么需要执行parkAndCheckInterrupt():
    
        // 这个方法很简单,因为前面返回true,所以需要挂起线程,这个方法就是负责挂起线程的
        // 这里用了LockSupport.park(this)来挂起线程,然后就停在这里了,等待被唤醒=======
        private final boolean parkAndCheckInterrupt() {
            LockSupport.park(this);
            return Thread.interrupted();
        }
        
        
        // 2. 接下来说说如果shouldParkAfterFailedAcquire(p, node)返回false的情况
    
       // 仔细看shouldParkAfterFailedAcquire(p, node),我们可以发现,其实第一次进来的时候,一般都不会返回true的,原因很简单,前驱节点的waitStatus=-1是依赖于后继节点设置的。也就是说,我都还没给前驱设置-1呢,怎么可能是true呢,但是要看到,这个方法是套在循环里的,所以第二次进来的时候状态就是-1了。
    
        // 解释下为什么shouldParkAfterFailedAcquire(p, node)返回false的时候不直接挂起线程:
        
        
    }
    

    四、解锁操作

    最后,就是还需要介绍下唤醒的动作了。我们知道,正常情况下,如果线程没获取到锁,线程会被 LockSupport.park(this); 挂起停止,等待被唤醒。

    // 唤醒的代码还是比较简单的,你如果上面加锁的都看懂了,下面都不需要看就知道怎么回事了
    public void unlock() {
        sync.release(1);
    }
    
    public final boolean release(int arg) {
        // 往后看吧
        if (tryRelease(arg)) {
            Node h = head;
            if (h != null && h.waitStatus != 0)
                unparkSuccessor(h);
            return true;
        }
        return false;
    }
    
    // 回到ReentrantLock看tryRelease方法
    protected final boolean tryRelease(int releases) {
        int c = getState() - releases;
        if (Thread.currentThread() != getExclusiveOwnerThread())
            throw new IllegalMonitorStateException();
        // 是否完全释放锁
        boolean free = false;
        // 其实就是重入的问题,如果c==0,也就是说没有嵌套锁了,可以释放了,否则还不能释放掉
        if (c == 0) {
            free = true;
            setExclusiveOwnerThread(null);
        }
        setState(c);
        return free;
    }
    
    // 唤醒后继节点
    // 从上面调用处知道,参数node是head头结点
    private void unparkSuccessor(Node node) {
        /*
         * If status is negative (i.e., possibly needing signal) try
         * to clear in anticipation of signalling.  It is OK if this
         * fails or if status is changed by waiting thread.
         */
        int ws = node.waitStatus;
        // 如果head节点当前waitStatus<0, 将其修改为0
        if (ws < 0)
            compareAndSetWaitStatus(node, ws, 0);
        /*
         * Thread to unpark is held in successor, which is normally
         * just the next node.  But if cancelled or apparently null,
         * traverse backwards from tail to find the actual
         * non-cancelled successor.
         */
        // 下面的代码就是唤醒后继节点,但是有可能后继节点取消了等待(waitStatus==1)
        // 从队尾往前找,找到waitStatus<=0的所有节点中排在最前面的
        Node s = node.next;
        if (s == null || s.waitStatus > 0) {
            s = null;
            // 从后往前找,仔细看代码,不必担心中间有节点取消(waitStatus==1)的情况
            for (Node t = tail; t != null && t != node; t = t.prev)
                if (t.waitStatus <= 0)
                    s = t;
        }
        if (s != null)
            // 唤醒线程
            LockSupport.unpark(s.thread);
    }
    

    唤醒线程以后,被唤醒的线程将从以下代码中继续往前走:

    private final boolean parkAndCheckInterrupt() {
        LockSupport.park(this); // 刚刚线程被挂起在这里了
        return Thread.interrupted();
    }
    // 又回到这个方法了:acquireQueued(final Node node, int arg),这个时候,node的前驱是head了
    

    好了,后面就不分析源码了,剩下的还有问题自己去仔细看看代码吧。

    五、总结

    在并发环境下,加锁和解锁需要以下三个部件的协调:

    • 锁状态。我们要知道锁是不是被别的线程占有了,这个就是 state 的作用,它为 0 的时候代表没有线程占有锁,可以去争抢这个锁,用 CAS 将 state 设为 1,如果 CAS 成功,说明抢到了锁,这样其他线程就抢不到了,如果锁重入的话,state进行 +1 就可以,解锁就是减 1,直到 state 又变为 0,代表释放锁,所以 lock() 和 unlock() 必须要配对啊。然后唤醒等待队列中的第一个线程,让其来占有锁。
    • 线程的阻塞和解除阻塞。AQS 中采用了 LockSupport.park(thread) 来挂起线程,用 unpark 来唤醒线程。
    • 阻塞队列。因为争抢锁的线程可能很多,但是只能有一个线程拿到锁,其他的线程都必须等待,这个时候就需要一个 queue 来管理这些线程,AQS 用的是一个 FIFO 的队列,就是一个链表,每个 node 都持有后继节点的引用。AQS 采用了 CLH 锁的变体来实现,感兴趣的读者可以参考这篇文章关于CLH的介绍,写得简单明了。

    参考文章:https://javadoop.com/post/AbstractQueuedSynchronizer

  • 相关阅读:
    Android&Handler2
    ACM&数字阶梯求和(大数问题)
    Android&MyThread
    【转】 《基于MFC的OpenGL编程》Part 15 Selection
    【转】 《基于MFC的OpenGL编程》Part 12 Creating and Using Display Lists
    【转】《基于MFC的OpenGL编程》Part 16 Reflection
    【转】 《基于MFC的OpenGL编程》Part 8 Colors
    【转】 《基于MFC的OpenGL编程》Part 14 Quadrics
    开发不同语言版本的C#程序
    基于ADS40的线阵摄影测量及数据处理
  • 原文地址:https://www.cnblogs.com/ChenBingJie123/p/15431247.html
Copyright © 2011-2022 走看看