by emanjusaka from https://www.emanjusaka.top/archives/8 彼岸花开可奈何
本文欢送分享与聚合,全文转载请留下原文地址。

前言

AbstractQueuedSynchronizer形象同步队列简称AQS,它是实现同步器的根底组件,并发包中锁的底层就是应用AQS实现的。大多数开发者可能永远不会间接应用AQS,然而晓得其原理对于架构设计还是很有帮忙的。AQS是Java中的一个抽象类,全称是AbstractQueuedSynchronizer,即形象队列同步器。它定义了两种资源共享模式:独占式和共享式。独占式每次只能有一个线程持有锁,例如ReentrantLock实现的就是独占式的锁资源;共享式容许多个线程同时获取锁,并发访问共享资源,ReentrantReadWriteLock和CountDownLatch等就是实现的这种模式。AQS保护了一个volatile的state变量和一个FIFO(先进先出)的队列。其中state变量代表的是竞争资源标识,而队列代表的是竞争资源失败的线程排队时寄存的容器 。

一、原理

AQS的核心思想是通过一个FIFO的队列来治理线程的期待和唤醒,同时保护了一个state变量来示意同步状态,能够通过getState、setState、compareAndSetState函数批改其值。当一个线程想要获取锁时,如果state为0,则示意该线程获取锁胜利,否则示意该线程获取锁失败。它将被放入期待队列中,直到满足特定条件能力再次尝试获取。当一个线程开释锁时,如果state为1,则示意该线程开释锁胜利,否则示意该线程开释锁失败。AQS通过CAS操作来实现加锁和解锁。

1.1 CLH队列

AQS中的CLH队列锁是CLH锁的一种变体,将自旋操作改成了阻塞线程操作。AQS 中的对 CLH 锁数据结构的改良次要包含三方面:扩大每个节点的状态、显式的保护前驱节点和后继节点以及诸如出队节点显式设为 null 等辅助 GC 的优化。

在 AQS(AbstractQueuedSynchronizer)中应用的 CLH 队列,head 指针和 tail 指针别离指向 CLH 队列中的两个要害节点。

  1. head 指针:head 指针指向 CLH 队列中的首个节点,该节点示意以后持有锁的线程。当一个线程胜利地获取到锁时,它就成为了持有锁的线程,并且会将该信息记录在 head 指针所指向的节点中。
  2. tail 指针:tail 指针指向 CLH 队列中的最初一个节点,该节点示意队列中最初一个期待获取锁的线程。当一个线程尝试获取锁时,它会生成一个新的节点,并将其插入到 CLH 队列的尾部,而后成为 tail 指针所指向的节点。这样,tail 指针的作用是标记以后 CLH 队列中最初一个期待获取锁的线程。

通过 head 指针和 tail 指针,CLH 队列可能保护一种有序的期待队列构造,保障线程获取锁的程序和互斥拜访的正确性。当一个线程开释锁时,它会批改以后节点的状态,并唤醒后继节点上的线程,让后续的线程可能及时感知锁的开释,并抢夺获取锁的机会。

1.2 线程同步

对于AQS来说,线程同步的要害是对状态值state进行操作。state为0时示意没有线程持有锁,大于0时示意有线程持有锁。依据state是否属于一个线程,操作state的形式分为独占形式和共享形式。

在独占形式下获取和开释资源应用的办法为:

  • void acquire(int arg)
  • void acquireInterruptibly(int arg)
  • boolean release(int arg)

应用独占形式获取的资源是与具体线程绑定的,就是说如果一个线程获取到了资源,就会标记是这个线程获取到了,其余线程再尝试操作state获取资源时会发现以后该资源不是本人持有的,就会在获取失败后被阻塞。

在共享形式下获取和开释资源的办法为:

  • void acquireShared(int arg)
  • voidacquireSharedInterruptibly(int arg)
  • boolean releaseShared(int arg)。

对应共享形式的资源与具体线程是不相干的,当多个线程去申请资源时通过CAS形式竞争获取资源,当一个线程获取到了资源后,另外一个线程再次去获取时如果以后资源还能满足它的须要,则以后线程只须要应用CAS形式进行获取即可。

二、资源获取与开释

2.1 独占式

  1. 当一个线程调用acquire(int arg)办法获取独占资源时,会首先应用tryAcquire办法尝试获取资源,具体是设置状态变量state的值,胜利则间接返回,失败则将以后线程封装为类型为Node.EXCLUSIVE的Node节点后插入到AQS阻塞队列的尾部,并调用LockSupport.park(this)办法挂起本人。

        public final void acquire(int arg) {      if (! tryAcquire(arg) &&          acquireQueued(addWaiter(Node.EXCLUSIVE), arg))          selfInterrupt();   }

<!---->

  1. 当一个线程调用release(int arg)办法时会尝试应用tryRelease操作开释资源,这里是设置状态变量state的值,而后调用LockSupport.unpark(thread)办法激活AQS队列外面被阻塞的一个线程(thread)。被激活的线程则应用tryAcquire尝试,看以后状态变量state的值是否能满足本人的须要,满足则该线程被激活,而后持续向下运行,否则还是会被放入AQS队列并被挂起。

        public final boolean release(int arg) {        if (tryRelease(arg)) {            Node h = head;            if (h ! = null && h.waitStatus ! = 0)                unparkSuccessor(h);            return true;        }        return false;    }

2.2 共享式

  1. 当线程调用acquireShared(int arg)获取共享资源时,会首先应用tryAcquireShared尝试获取资源,具体是设置状态变量state的值,胜利则间接返回,失败则将以后线程封装为类型为Node.SHARED的Node节点后插入到AQS阻塞队列的尾部,并应用LockSupport.park(this)办法挂起本人。

        public final void acquireShared(int arg) {        if (tryAcquireShared(arg) < 0)            doAcquireShared(arg);    }

<!---->

  1. 当一个线程调用releaseShared(int arg)时会尝试应用tryReleaseShared操作开释资源,这里是设置状态变量state的值,而后应用LockSupport.unpark(thread)激活AQS队列外面被阻塞的一个线程(thread)。被激活的线程则应用tryReleaseShared查看以后状态变量state的值是否能满足本人的须要,满足则该线程被激活,而后持续向下运行,否则还是会被放入AQS队列并被挂起。

        public final boolean releaseShared(int arg) {          if (tryReleaseShared(arg)) {              doReleaseShared();              return true;          }          return false;      }

三、基于AQS实现自定义同步器

基于AQS实现一个不可重入的独占锁,自定义AQS须要重写一系列函数,还须要定义原子变量state的含意。这里定义,state为0示意目前锁没有被线程持有,state为1示意锁曾经被某一个线程持有,因为是不可重入锁,所以不须要记录持有锁的线程获取锁的次数。

package top.emanjusaka;import java.io.Serializable;import java.util.concurrent.TimeUnit;import java.util.concurrent.locks.AbstractQueuedSynchronizer;import java.util.concurrent.locks.Condition;import java.util.concurrent.locks.Lock;public class UnReentrantLock implements Lock, Serializable {    // 借助 AbstractQueuedSynchronizer 实现    private static class Sync extends AbstractQueuedSynchronizer {        // 查看是否有线程持有锁        protected boolean isHeldExclusively() {            return getState() == 1;        }        // 尝试获取锁        public boolean tryAcquire(int acquires) {            assert acquires == 1;            // 应用CAS 设置state            if (compareAndSetState(0, 1)) {                // 如果 CAS 操作胜利,示意胜利取得了锁。这时,通过 setExclusiveOwnerThread 办法将以后线程设置为独占锁的拥有者                setExclusiveOwnerThread(Thread.currentThread());                return true;            }            // 如果 CAS 操作失败,即无奈将 state 的值从 0 设置为 1,示意锁已被其余线程占用,无奈获取锁,于是返回 false。            return false;        }        // 尝试开释锁        protected boolean tryRelease(int releases) {            assert releases == 1;            if (getState() == 0)                throw new IllegalMonitorStateException();            // 开释胜利,将独占锁的拥有者设为null            setExclusiveOwnerThread(null);            // 将state的值设为0            setState(0);            return true;        }        Condition newCondition() {            return new ConditionObject();        }    }    private final Sync sync = new Sync();    @Override    public void lock() {        sync.acquire(1);    }    @Override    public boolean tryLock() {        return sync.tryAcquire(1);    }    @Override    public void unlock() {        sync.release(1);    }    @Override    public Condition newCondition() {        return sync.newCondition();    }    public boolean isLocked() {        return sync.isHeldExclusively();    }    public void lockInterruptibly() throws InterruptedException {        sync.acquireInterruptibly(1);    }    public boolean tryLock(long timeout, TimeUnit unit) throws InterruptedException {        return sync.tryAcquireSharedNanos(1, unit.toNanos(timeout));    }}

在开释锁时并没有应用 CAS(Compare and Swap)操作,而是间接应用了 setState 办法来将 state 的值设置为 0。

这是因为在开释锁的过程中,并不需要波及到多线程并发的问题。只有持有锁的线程才可能开释锁,其余线程无奈对锁进行操作。因而,不须要应用 CAS 来进行原子性的状态更新。

在这种状况下,能够间接应用一般的办法来设置 state 的值为 0,将独占锁的拥有者设为 null。因为只有一个线程能够操作这个锁,不存在并发竞争的状况,也就不须要应用 CAS 来保障原子性。

须要留神的是,当调用 tryRelease 办法时,应该保障以后线程是持有锁的线程,否则会抛出 IllegalMonitorStateException 异样。这是为了确保只有领有锁的线程能力开释锁,避免误开释其余线程的锁。

四、参考资料

  1. 《并发编程之美》
  2. AbstractQueuedSynchronizer抽象类的源码
本文原创,满腹经纶,如有纰漏,欢送斧正。尊贵的敌人,如果本文对您有所帮忙,欢送点赞,并期待您的反馈,以便于一直优化。
原文地址: https://www.emanjusaka.top/archives/8
微信公众号:emanjusaka的编程栈