Java锁机制浅析之 AQS

  一、内部原理

  类继承结构

  Lock package相关API继承结构,忽略掉了一些类,以便观察其特点:

  ReentrantLock和 ReentrantReadWriteLock都是借助内部类Sync来实现Lock接口。ReentrantReadWriteLock没有直接实现Lock接口而是内置了读锁-ReadLock和写锁-WriteLock分别实现Lock接口。

  Sync包含两个子类:FairSync(公平锁),NonfairSync(非公平锁);在构建ReentrantLock和 ReentrantReadWriteLock对象时,可通过带有参数的构造函数指定是否采用公平锁;不指定时默认是非公平锁。Sync类是从·AbstractQueuedSynchronizer继承而来。

  抽象队列同步器AQS

  AbstractQueuedSynchronizer简称AQS,Doug Lea大师创作,用来构建锁或者其他同步组件的基础框架类,除了ReentrantLock和ReentrantReadWriteLock还有很多其他的并发工具类的实现都依赖于AQS;CountDownLatch、CyclicBarrier、Semaphore等都是基于AQS实现。(本章主要分析AQS的源码,下一章可以分析下这些类的用法)。

  AQS为同步状态的原子性管理、线程的阻塞和解除阻塞以及排队提供了一种通用机制。基于模板方法模式设计,在使用时需要通过组合一个AQS的子类并重写其同步资源获取和释放逻辑方法tryAcquire()和tryRelease()。

  AQS的使用思路很简单,在有线程获取锁时,调用子类tryAcquire()尝试获取锁,在获取失败时,将线程加入到一个基于CLH锁结构实现的等待队列中,当持有锁的线程执行完操作释放锁时,调用子类的tryRelease()释放锁资源,并唤醒等待队列中的下一个线程开始获取锁。

  主要API

  同步状态管理

扫描二维码关注公众号,回复: 8365162 查看本文章

  int getState(): 获取同步状态

  void setState(): 设置同步状态

  boolean compareAndSetState(int expect, int update):基于CAS,原子设置当前状态

  两组用于实现独占模式和共享模式的抽象方法。

  独占模式锁操作

  acquire(int) :尝试获取争用资源,如果未获取到,当前线程将进入等待队列。

  tryAcquire():争用资源逻辑的具体实现,必须由具体的AQS使用者去实现。

  acquireInterruptibly():支持中断的获取acquire操作

  release():当前线程释放争用资源,并唤醒等待队列中的其他线程进入争抢。

  tryRelease():释放争用资源的具体实现,必须由具体的AQS使用者去实现

  共享模式锁操作

  acquireShared():acquire的共享模式版本。

  tryAcquireShared():tryAcquire的共享模式版本。

  acquireSharedInterruptibly():acquireInterruptibly的共享模式版本

  releaseShared():release的共享模式版本

  tryReleaseShared():tryRelease的共享模式版本

  源码

  内部等待队列Node

  static final class Node {

  static final Node SHARED = new Node(); //共享模式

  static final Node EXCLUSIVE = null; //独占模式

  Node nextWaiter; //独占或共享模式标记

  //等待状态常量定义

  static final int CANCELLED = 1; //取消

  static final int SIGNAL = -1; //通知

  static final int CONDITION = -2; //条件等待

  static final int PROPAGATE = -3; //传播

  volatile int waitStatus; //等待状态属性

  volatile Node prev; //前驱节点

  volatile Node next; //后继节点

  volatile Thread thread; //节点所封装的线程

  //判断节点是否处于共享模式

  final boolean isShared() {

  return nextWaiter == SHARED;

  }

  //获取前驱节点,为空时抛出异常

  final Node predecessor() throws NullPointerException {

  Node p = prev;

  if (p == null) {

  throw new NullPointerException();

  } else {

  return p;

  }

  }

  //以独占或共享模式构造一个线程节点,由AQS的addWaiter调用

  Node(Thread thread, Node mode) {

  this.nextWaiter = mode;

  this.thread = thread;

  }

  //Condition模式下使用此构造函数

  Node(Thread thread, int waitStatus) {

  this.waitStatus = waitStatus;

  this.thread = thread;

  }

  }

  独占锁的获取acquire()

  /**

  * !tryAcquire(arg):

  * 首先调用其子类(实际的AQS使用者)尝试获取锁,获取成功函数就直接返回了。

  * acquireQueued(addWaiter(Node.EXCLUSIVE), arg)): 子类未获取到锁时:

  * 通过addWaiter() 以独占模式将当前线程封装成为Node节点并执行必要的初始化。

  * 通过acquireQueued()根据节点在队列中的位置,执行获取锁或者是阻塞线程操作

  */

  public final void acquire(int arg) {

  if (!tryAcquire(arg) && acquireQueued(addWaiter(Node.EXCLUSIVE), arg))

  selfInterrupt();

  }

  /**

  * 1. 将当前线程封装为node节点。

  * 2. 如队列未初始化,则初始化队列,并把当前线程节点加入队列末尾。

  * 3. 如果队列已经初始化(已经存在等待锁的其他线程节点),则将当前线程节点直接加入到队里末尾

  * 4. 返回代表当前线程的Node节点。

  */

  private Node addWaiter(Node mode) {

  Node node = new Node(Thread.currentThread(), mode); // 将当前线程封装成为Node节点。

  Node pred = tail; // pred指向队列末尾

  if (pred != null) { // 如果队列末尾元素不为空

  node.prev = pred;

  // CAS更新队列末尾指向当前线程节点。更新成功后,将旧队列末尾的后驱指向当前线程节点,返回当前线程节点。

  if (compareAndSetTail(pred, node)) {

  pred.next = node;

  return node;

  }

  }

  enq(node); //如果队列为空,则初始化队列,然后将当前线程节点加入队列

  return node;

  }

  /**

  * 如果队列未初始化,则使用一个无指向的Node作为队列头,将当前线程节点链接到head作为队列尾。

  * 此处使用轻量级的自循环+Cas更新的方式,来处理enq函数被多线程并发执行时的问题,

  * 想一想如果没有循环那么enq函数被多线程并发时可能带来什么问题?

  */

  private Node enq(final Node node) {

  for (;;) {

  Node t = tail;

  if (t == null) { // Must initialize

  if (compareAndSetHead(new Node()))

  tail = head;

  } else {

  node.prev = t;

  if (compareAndSetTail(t, node)) {

  t.next = node;

  return t;

  }

  }

  }

  }

  /**

  * 如果节点前驱是队列头,则尝试获取锁。

  * 此时:node就是队列首个有效节点,head在初始化时是一个无指向的Node,

  * 而node在获取锁后通过setHead(node)被设置成为head,但同时置空了node指向的Thread和前驱节点,head仍然是个无指向的节点)

  * 如果节点前驱不是队列头,则将前驱状态更改为SIGNAL后通过LockSupper.park()阻塞当前线程。

  * 此时:在node之前至少有两个有效节点(node.prev和node.prev.prev)node.prev.prev此时可能已经获取锁,也可能在等待。但是node.prev

  * 一定在等待,那么当前节点应该进入阻塞等待,而前驱节点已经处于等待状态可以被设置设置成待通知状态。

  */

  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)) {

  setHead(node);

  p.next = null; // help GC

  failed = false;

  return interrupted;

  }

  // shouldParkAfterFailedAcquire(): 保证当前节点前驱节点状态为Node.SIGNAL后

  // parkAndCheckInterrupt(): 通过LockSupport.park() 阻塞当前线程。

  if (shouldParkAfterFailedAcquire(p, node) && parkAndCheckInterrupt())

  interrupted = true;

  }

  } finally {

  if (failed)

  //获取成功后,

  cancelAcquire(node);

  }

  }

  /**

  * 如果当前节点前驱节点状态为Node.SIGNAL,则返回true,否则,将前驱节点状态设置为Node.SIGNAL后

  * 如果遇到前驱节点状态为取消的情况,则从前驱节点向前遍历,找到最后一个未被取消的节点,将当前节点连接在该节点后面。

  */郑州哪家医院人流好 http://www.120csfkyy.com/

  private static boolean shouldParkAfterFailedAcquire(Node pred, Node node) {

  int ws = pred.waitStatus;

  // 已经设置了状态,可以安全

  if (ws == Node.SIGNAL) return true;

  if (ws > 0) {

  //从当前线程节点向前遍历,把当前线程节点放到队列最靠后的一个有效等待的节点。

  do {

  node.prev = pred = pred.prev;

  } while (pred.waitStatus > 0);

  pred.next = node;

  } else {

  compareAndSetWaitStatus(pred, ws, Node.SIGNAL);

  }

  return false;

  }

  独占锁的释放release()

  /**

  * 首先调tryRelease()对子类资源进行释放。

  * 如果 head!=null 或者 head的等待状态不为零,说明仍然有后续节点在等待。此时需要唤醒后续节点。

  */

  public final boolean release(int arg) {

  if (tryRelease(arg)) {

  Node h = head;

  if (h != null && h.waitStatus != 0)

  unparkSuccessor(h);

  return true;

  }

  return false;

  }

  /**

  * 将当前节点状态重置为零,并从队列尾部向前找到第一个有效节点,通过unpark通知该节点线程。,

  * 为何不是从当前节点向后查找?

  */

  private void unparkSuccessor(Node node) {

  int ws = node.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;

  }

  if (s != null)

  LockSupport.unpark(s.thread);

  }

  共享锁的获取 acquireShare

  有了独占模式的分析后,共享模式就简单多了。

  /**

  * 调用子类tryAcquireShare尝试获取锁,tryAcquireShare()函数返回值小于零时获取失败。

  * 获取失败时,调用doAcquireShared方法处理

  */

  public final void acquireShared(int arg) {

  if (tryAcquireShared(arg) < 0)

  doAcquireShared(arg);

  }

  /**

  * 与独占模式流程相似,但是封装节点时采用的是SHARED模式。

  */

  private void doAcquireShared(int arg) {

  final Node node = addWaiter(Node.SHARED);

  boolean failed = true;

  try {

  boolean interrupted = false;

  for (;;) {

  final Node p = node.predecessor();

  if (p == head) {

  int r = tryAcquireShared(arg);

  if (r >= 0) {

  setHeadAndPropagate(node, r);

  p.next = null; // help GC

  if (interrupted)

  selfInterrupt();

  failed = false;

  return;

  }

  }

  if (shouldParkAfterFailedAcquire(p, node) &&

  parkAndCheckInterrupt())

  interrupted = true;

  }

  } finally {

  if (failed)

  cancelAcquire(node);

  }

  }

  /**

  * 调用子类释放锁,然后将当前节点从队列

  */

  public final boolean releaseShared(int arg) {

  if (tryReleaseShared(arg)) {

  doReleaseShared();

  return true;

  }

  return false;

  }

  共享锁的释放doReleaseShared

  /**

  * 实际执行释放锁的操作,

  * 由于是共享模式,在释放锁时也存在多个线程并发释放的情况,因此采用循环+CAS的机制

  * 来使并发释放的多个线程顺序unpark队列中的后续节点。

  */

  private void doReleaseShared() {

  for (;;) {

  Node h = head;

  if (h != null && h != tail) {

  int ws = h.waitStatus;

  if (ws == Node.SIGNAL) {

  if (!compareAndSetWaitStatus(h, Node.SIGNAL, 0))

  continue; // loop to recheck cases

  unparkSuccessor(h);

  }

  else if (ws == 0 &&

  !compareAndSetWaitStatus(h, 0, Node.PROPAGATE))

  continue; // loop on failed CAS

  }

  if (h == head) // loop if head changed

  break;

  }

  }

  可中断模式 acquireSharedInterruptibly

  中断模式也获取的也是独占锁,只是会在尝试获取锁前先检查线程的中断状态,如果线程被中断,那么会抛出InterruptedException而不会继续尝试获取锁。中断模式获取的锁也通过release()进行释放。

  /**

  * 在获取锁时会首先判断线程是否被中断,如果中断则直接抛出异常。

  * 未被中断,则调用子类tryAcquire()尝试获取锁。获取失败时,通过doAcquireInterruptibly()进行处理。

  */

  public final void acquireInterruptibly(int arg) throws InterruptedException {

  if (Thread.interrupted())

  throw new InterruptedException();

  if (!tryAcquire(arg))

  doAcquireInterruptibly(arg);

  }

  /**

  * 与doAcquire()同样的流程,区别只在于,线程进入等待队列,而后被其他节点唤醒后,会根据线程中断状态决定后续操作:

  * 如果线程被中断,则直接抛出InterruptedException而不会继续获取锁。

  */

  private void doAcquireInterruptibly(int arg) throws InterruptedException {

  final Node node = addWaiter(Node.EXCLUSIVE);

  boolean failed = true;

  try {

  for (;;) {

  final Node p = node.predecessor();

  if (p == head && tryAcquire(arg)) {

  setHead(node);

  p.next = null; // help GC

  failed = false;

  return;

  }

  if (shouldParkAfterFailedAcquire(p, node) &&

  parkAndCheckInterrupt())

  throw new InterruptedException();

  }

  } finally {

  if (failed)

  cancelAcquire(node);

  }

  }

  到此AQS的源码基本就完了。

猜你喜欢

转载自www.cnblogs.com/gnz49/p/12120156.html