AQS(一) 对CLH队列的增强

2018-01-11 16:52 faunjoe88 阅读(897) 评论(0) 编辑收藏举报

基本概念

AQS(AbstractQueuedSynchronizer),顾名思义，是一个抽象的队列同步器。

它的队列是先进先出(FIFO)的等待队列
基于这个队列，AQS提供了一个实现阻塞锁的机制
最终，基于这个阻塞锁，可以实现多线程的同步

先进先出的等待队列

这个等待队列，是基于CLH锁实现的。
CLH锁是以发明人命名的自旋锁，这个锁是一个基于队列的自旋锁，是对SpinLock，TicketLock的进化。
具体可以参考另外一篇文章，最好是可以自己实现下。

CLH锁的基本原理：

使用FIFO队列保证公平性
有当前节点和前置节点，当前节点不断自旋，查询(监听)前置节点的状态（isLocked)(保证了FIFO)
一系列的前置节点和当前节点构成队列
当前节点运行完成后，更改自己的状态，那监听当前节点状态的线程就会结束自旋

CLH锁，使用了一个属性tail（尾节点）来关联前置节点和当前节点
lock时：

新建节点并与当前线程绑定(使用ThreadLocal)
获取原有的尾节点作为前置节点，设置尾节点为当前节点（这样其他节点就可以挂在当前节点后）

//说明有前置节点
    if(原尾节点不为空){
        //自旋以等待前置节点状态改变
    }

unlock时：

把当前节点的isLock状态置为false.
(下一个节点的)监听当前节点状态的自旋被解除

总结：
当前节点不断自旋访问前置节点的状态，状态改变时结束自旋

AQS对CLH的增强和进化

AQS基于CLH锁的访问前置节点信息的原理实现，并添加了一些功能：

支持阻塞而不是一直自旋，竞争激烈时，阻塞性能更好
支持可重入
支持取消节点
支持中断
支持独占（互斥）和共享两种模式
支持Condition Condition替代对象监听器(Monitor)用来等待，唤醒线程，用于线程间的协作

AQS基于对CLH的改进提供了实现同步器的机制，解决了获取锁失败后的问题，
子类仅仅需要实现获取/释放锁的具体逻辑。该机制包括：

AQS不关注怎么获取锁/释放锁(如何获取和释放锁，由具体子类实现)。

//获取锁的逻辑，成功返回true
    protected boolean tryAcquire(int arg) {
        throw new UnsupportedOperationException();
    }
    //释放锁，完全释放(考虑重入)返回true
    protected boolean tryRelease(int arg) {
        throw new UnsupportedOperationException();
    }
    //共享模式获取锁
    protected int tryAcquireShared(int arg) {
        throw new UnsupportedOperationException();
    }
    //共享模式释放锁
    protected boolean tryReleaseShared(int arg) {
        throw new UnsupportedOperationException();
    }

AQS关注的是获取失败后：
2.1 构建CLH队列
2.2 不断自旋
2.2.1 检查前置节点是否是头节点（头节点是持有锁的节点），如果是头节点，则尝试获取锁，获取成功，线程持有锁，否则走下面
2.2.2 检查前置节点的状态，如果不是等待唤醒（SIGNAL）,就把前置节点设置为SIGNAL，否则把当前节点的线程阻塞，
也就是说，第二次循环时，当前节点就被阻塞了。
2.2.3 线程被唤醒后，会继续执行2.2.1，得到获取锁的机会

/**
                入口：获取锁，失败就阻塞
             * Acquires in exclusive mode, ignoring interrupts.  Implemented
             * by invoking at least once {@link #tryAcquire},
             * returning on success.  Otherwise the thread is queued, possibly
             * repeatedly blocking and unblocking, invoking {@link
             * #tryAcquire} until success.  This method can be used
             * to implement method {@link Lock#lock}.
             *
             * @param arg the acquire argument.  This value is conveyed to
             *        {@link #tryAcquire} but is otherwise uninterpreted and
             *        can represent anything you like.
             */
            public final void acquire(int arg) {
                if (!tryAcquire(arg) &&
                    acquireQueued(addWaiter(Node.EXCLUSIVE), arg))
                    selfInterrupt();
            }

            /**
             * 构建了一个队列 对应2.1

             1.入队：原子性的更新tail
             2.出队：更新head

             *        +------+  prev +-----+       +-----+
                 head |      | <---- |     | <---- |     |  tail
                      +------+       +-----+       +-----+
             * Creates and enqueues node for current thread and given mode.
             *
             * @param mode Node.EXCLUSIVE for exclusive, Node.SHARED for shared
             * @return the new node
             */
            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 pred = tail;
                if (pred != null) {
                    node.prev = pred;
                    if (compareAndSetTail(pred, node)) {
                        pred.next = node;
                        return node;
                    }
                }
                enq(node);
                return node;
            }
            /**
                自旋至阻塞的过程：对应2.2
             * Acquires in exclusive uninterruptible mode for thread already in
             * queue. Used by condition wait methods as well as acquire.
             *
             * @param node the node
             * @param arg the acquire argument
             * @return {@code true} if interrupted while waiting
             */
            final boolean acquireQueued(final Node node, int arg) {
                boolean failed = true;
                try {
                    boolean interrupted = false;
                    for (;;) {
                        //检查头节点，对应2.2.1
                        final Node p = node.predecessor();
                        if (p == head && tryAcquire(arg)) {
                            setHead(node);
                            p.next = null; // help GC
                            failed = false;
                            return interrupted;
                        }
                        //判断前置节点的状态，并阻塞当前节点对应的线程 对应2.2.2
                        if (shouldParkAfterFailedAcquire(p, node) &&
                            parkAndCheckInterrupt())
                            interrupted = true;
                    }
                } finally {
                    if (failed)
                        cancelAcquire(node);
                }
            }

释放锁成功后:
3.1 拿到头节点的下一个节点
3.2 唤醒这个节点
3.3 被唤醒的节点走到2.2.3 然后走2.2.1

/**
    释放锁的入口：成功后唤醒下一个节点
     * Releases in exclusive mode.  Implemented by unblocking one or
     * more threads if {@link #tryRelease} returns true.
     * This method can be used to implement method {@link Lock#unlock}.
     *
     * @param arg the release argument.  This value is conveyed to
     *        {@link #tryRelease} but is otherwise uninterpreted and
     *        can represent anything you like.
     * @return the value returned from {@link #tryRelease}
     */
    public final boolean release(int arg) {
        if (tryRelease(arg)) {
            //对应3.1
            Node h = head;
            if (h != null && h.waitStatus != 0)
                //对应3.2，唤醒之后，下一个节点结束阻塞，自旋获取锁进入2.2.1
                unparkSuccessor(h);
            return true;
        }
        return false;
    }

小结：

从整个锁的获取和释放的流程来看，其基本的原理与CLH锁并没有太大区别：

1.尝试获取锁，失败则进入队列，然后自旋检查前置节点的状态
2.释放锁，同时保证某个后置节点能获取到锁并结束自旋

刷新页面返回顶部

耐心等待