深入并发原理和大厂面试(一):AQS解析

AQS(AbstractQueuedSynchronizer)是Java众多锁以及并发工具的基础类,底层采用乐观锁,大量采用CAS操作保证其原子性,并且在并发冲突时,采用自旋方法重试。实现了轻量高效的获取锁。

1. AQS的关注点

ReentrantLock中使用到了AQS高并发组件,用它来维护锁的状态,这样就不需要利用操作系统来维护,减少了上下文切换。AQS中使用了CAS、自旋操作来提高性能。但是在线程过多的时候,还是会和操作系统打交道,挂起线程和唤醒线程两个上下文操作。

AQS在线程交替运行时,只需要借助CAS和自旋就可以完成加锁。

而synchronized在JDK1.6版本前是重量级锁,加锁的操作是涉及到操作系统进行互斥操作,就是会把当前线程挂起,然后操作系统进行互斥操作修改,由mutexLock来完成,之后才唤醒。操作系统来判断线程是否加锁,所以它是一个重量级操作。挂起、唤醒这两个操作进行了两次上下文切换,消耗CPU,降低性能。
总之一句话,重量级锁是需要依靠操作系统来实现互斥锁的,这导致大量上下文切换,消耗大量CPU,影响性能。

1.1 信号量

在AQS中,状态是由volatile state来表示。

private volatile int state;

该属性值表示锁的状态。state为0表示锁未被占用,state为1表示锁被线程持有,而state大于1表示锁被重入。

而本文分析的是独占锁,那么同一时刻,锁只能被一个线程持有。

不仅需要记录锁的状态,还需要记录当前获取锁的线程,实现重入。可以通过来记录。

private transient Thread exclusiveOwnerThread; 

1.2. 等待队列

等待队列采用悲观锁的思想,表示当前所等待的资源,状态或条件短时间内可能无法满足,而调用park方法(借助操作系统)来完成线程的阻塞。

在AQS中,队列时一个双端链表,将当前线程包装成某种类型的数据结构扔到等待队列中。

static final class Node {  
// 节点所代表的线程  
volatile Thread thread;    
// 双向链表,每个节点需要保存自己的前驱节点和后继节点的引用  
volatile Node prev;  
volatile Node next;  
// 线程所处的等待锁的状态,初始化时,该值为0。  
volatile int waitStatus;  
//队列中节点线程被取消
static final int CANCELLED =  1;
//节点将其前驱节点设置为-1,当前驱节点释放锁后,会自动唤醒该节点。  
static final int SIGNAL    = -1;  
//线程被重新包装为Node节点,并存入Condition队列中。
static final int CONDITION = -2;  
//共享锁唤醒风暴时,将0->PROPAGATE,表示被传播唤醒
static final int PROPAGATE = -3;  
// 该属性用于条件队列或者共享锁 。在Condition队列中,使用其作为指针。
Node nextWaiter;  
}  

一般在独占锁下,我们需要关注的就是下面几个参数:

  • thread:当前Node所代表的线程;
  • waitStatus:表示节点所处的等待状态;
  • prev next:节点的前驱和后继

1.3. CAS操作

CAS采用乐观锁机制,保证操作的原子性。一般是改变状态或改变指针(引用)指向。

深入并发原理和大厂面试(一):AQS解析_第1张图片

1.4 总结

在AQS源码中:

  1. 锁属性
 //锁的状态
private volatile int state;
//当前持有锁的线程
private transient Thread exclusiveOwnerThread;

  1. sync queue相关的属性
//thread属性为null
private transient volatile Node head; 
private transient volatile Node tail; // 队尾,新入队的节点

  1. Node相关属性
// 节点所代表的线程
volatile Thread thread;

// 双向链表,每个节点需要保存自己的前驱节点和后继节点的引用
volatile Node prev;
volatile Node next;

// 线程所处的等待锁的状态,初始化时,该值为0
volatile int waitStatus;
static final int CANCELLED =  1;
static final int SIGNAL    = -1;

2. 源码解析

ReentrantLock有公平锁和非公平锁两种实现,默认实现非公平锁。但是可配置为公平锁:

ReentrantLock lock=new ReentrantLock(true);

调用公平锁加锁逻辑:

final void lock() {  
    //开始加锁,将state修改为1
    acquire(1); 
}  

真正的加锁方法:

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

2.1 加锁的逻辑方法

只执行上述方法便可完成整个的加锁逻辑。而该方法中又包含下列四个方法的调用:

1. tryAcquire(arg)
该方法由继承AQS的子类实现,为获取锁的具体逻辑;

2. addWaiter(Node.EXCLUSIVE)
该方法由AQS实现,负责在获取锁失败后调用,将当前请求锁的线程包装成Node并且放到等待队列中,并返回该Node。

3. acquireQueued(addWaiter(Node.EXCLUSIVE), arg)
该方法由AQS实现。针对上面加入到队列的Node不断尝试两种操作之一:

  • 若前驱节点是head节点的时候,尝试获取锁;
  • 调用park将当前线程挂起,线程阻塞。

4. selfInterrupt
该方法由AQS实现。恢复用户行为。

  1. 用户在外界调用t1.interrupt()进行中断。

  2. 线程在parkAndCheckInterrupt方法被唤醒之后。会调用Thread.interrupted();判断线程的中断标识,而该方法调用完毕会清除中断标识位。

  3. 而AQS为了不改变用户标识。再次调用selfInterrupt恢复用户行为。

2.2 如何构建等待队列——addWaiter

我们使用ReentrantLock独占锁时,等待队列是延迟加载的。也就是说若是线程交替执行,那么借助信号量(状态)来保证。若是线程并发执行,就需要将阻塞线程放入到队列中。

//注意这个方法可能存在并发问题,mode为null(独占锁)。
private Node addWaiter(Node mode) {  
    Node node = new Node(Thread.currentThread(), mode);  
    Node pred = tail;  
    //队列已经存在
    if (pred != null) {  
       //新节点的前驱指针指向尾节点(可能造成尾分叉)
        node.prev = pred;  
       //保证原子性,只有一个才能成功
        if (compareAndSetTail(pred, node)) {  
            pred.next = node;  
            return node;  
        }  
    }  
    //队列不存在&&上面CAS失败的线程会进入enq方法自旋
    enq(node);  
    return node;  
}  

队列不存在的情况

image

注意,该方法处理CAS操作是原子性的,其他操作都存在并发冲突问题。

private Node enq(final Node node) {  
     for (;;) {  
         Node t = tail;  
          //初始化阻塞队列
         if (t == null) { // Must initialize  
             if (compareAndSetHead(new Node()))  
                 tail = head;  
         } else {  
            //自旋处理addWaiter中CAS加锁失败的线程
             node.prev = t;  
             if (compareAndSetTail(t, node)) {  
                 t.next = node;  
                 return t;  
             }  
         }  
     }  
 }  

该方法采用自旋+CAS。CAS是保证同一时刻只有一个线程能成功改变引用的指向。

深入并发原理和大厂面试(一):AQS解析_第2张图片

根据上面的流程图,sync queue的创建过程。head节点是new Node()产生的,即其中的属性为默认值。也就是thread属性为null。也就是说正在执行的线程也会在sync queue中占据头节点,但是节点中不会保存线程信息。

深入并发原理和大厂面试(一):AQS解析_第3张图片

尾分叉问题:

上面已经说了,该方法是线程不安全的。

 //步骤1:可能多个节点的prev指针都指向尾结点,导致尾分叉
 node.prev = t;  
 //步骤2:但同一时刻,tail引用只会执行一个node。
 if (compareAndSetTail(t, node)) {  
    //步骤3:现在环境是线程安全,旧尾结点的后继指针指向新尾结点。
    t.next = node;  
    return t;  
  }  

深入并发原理和大厂面试(一):AQS解析_第4张图片

执行完步骤2,但步骤3还未执行时,恰好有线程从头节点开始往后遍历。**此时(旧)尾结点中的next域还为null。**它是遍历不到新加进来的尾结点的。这显然是不合理的。

但此时步骤1是执行成功的,所以若是tail节点往前遍历,实际上是可以遍历到所有节点的,这也是为什么在AQS源码中,有时候常常会出现从尾结点开始逆向遍历链表的情况

那些“分叉”的节点,肯定会入队失败。那么继续自旋,等待所有的线程节点全部入队成功。

2.3 尝试获取锁——tryAcquire

根据标志位state,来判断锁是否被占用。此时可能锁未被占用,由于是公平锁,于是会去判断sync queue中是否有人在排队。

protected final boolean tryAcquire(int acquires) {  
    //获取当前线程
    final Thread current = Thread.currentThread();  
    //获取Lock对象的上锁情况,0-表示无线程持有;1-表示被线程持有;大于1-表示锁被重入
    int c = getState();  
    //若此刻无人占有锁
    if (c == 0) {  
        if (!hasQueuedPredecessors() &&    //判断队列中是否有前辈。若返回false代表没有,开始尝试加锁
            compareAndSetState(0, acquires)) {   //此刻队列中没有存在前辈,尝试加锁
            setExclusiveOwnerThread(current);   //将当前线程修改为持有锁的线程(后续判断可重入)
            return true;  
        }  
    }  
    //若是当前线程是持有锁的线程
    else if (current == getExclusiveOwnerThread()) {  
        //当前状态+1
        int nextc = c + acquires;  
        if (nextc < 0)  
            throw new Error("Maximum lock count exceeded");  
        setState(nextc);  
        return true;  
    }  
    //否则,代表加锁失败
    return false;  
}  

下面的方法返回false才会尝试加锁(该方法不具有原子性,可能会放行多个线程)。

//该方法不具有原子性,可能多个线程都觉得自己不需要排队,最终还是依靠外面
//条件上的CAS来保持其原子性。
public final boolean hasQueuedPredecessors() {  
    Node t = tail;   //尾节点
    Node h = head;   //头节点
    Node s;  
    return h != t &&  
        ((s = h.next) == null || s.thread != Thread.currentThread());  
} 

上述方法是判断队列中是否存在元素。可能存在以下几种情况:

  • 此时未维护队列【h和t指向null】,h!=t返回false,即无人排队;
  • 此时队列只有头节点(哑结点)【h和t都指向哑结点】,h!=t返回false,即无人排队;
  • 此时队列中存在2个以上的节点。若线程是头结点的后继节点线程(即处理正在办理业务的线程,进来的线程是第一个排队的线程)。那么s.thread != Thread.currentThread()返回false,即可是尝试加锁。
  • 队列存在2个以上节点,且进来的线程不是第一个排队的线程,那么该线程需要乖乖的排队。

当然该方法不是并发安全的方法,即可能存在多个线程觉得自己无需排队,最终还是依靠CAS来争夺锁。

 if (!hasQueuedPredecessors() &&  compareAndSetState(0, acquires)) {
     //线程安全   
     setExclusiveOwnerThread(current);  
     return true;  
 }  

同一时刻,只有一个线程可以成功改变state的状态。记录该线程为独占锁线程,一般后续可以重入。

没成功获取锁那么会调用2.2 中的方法,将该线程加入到阻塞队列中

2.3. 阻塞线程——acquireQueued

  • 若执行到该方法,说明addWaiter方法已经成功将该线程包装为Node节点放到了队尾。
  • 在该方法中依旧尝试获取锁;
  • 再次获取锁失败后,会将其阻塞;
final boolean acquireQueued(final Node node, int arg) {  
    boolean failed = true;  
    try {  
        boolean interrupted = false;  
        for (;;) {  
            //获取node的前驱节点
            final Node p = node.predecessor();  
            //若前驱节点在办理业务,那么它将再次获取一次锁。
            if (p == head && tryAcquire(arg)) {  
                //获取锁成功,此处便是线程安全。
                //将自己设置为头节点,并将自己设置为哑节点
                setHead(node);  
                p.next = null; // help GC  
                failed = false;  
                return interrupted;  
            }  
            //获取锁失败,将自己挂起。
            if (shouldParkAfterFailedAcquire(p, node) &&  
                parkAndCheckInterrupt())  
                interrupted = true;  
        }  
    } finally {  
        if (failed)  
            cancelAcquire(node);  
    }  
}  

上述方法时自旋方法,而出口就是获取到锁。若线程获取不到锁,便会将自己阻塞。

//该方法时node线程获取锁成功后执行的,故是线程安全的。
private void setHead(Node node) {  
    head = node;  
    node.thread = null;  
    node.prev = null;  
}  

深入并发原理和大厂面试(一):AQS解析_第5张图片

private static boolean shouldParkAfterFailedAcquire(Node pred, Node node) {  
    //上一个节点的waitStatus
    int ws = pred.waitStatus;  
    //  Node.SIGNAL==-1
    if (ws == Node.SIGNAL)  
        return true;  
    //ws大于0,则说明该节点已经被取消了。
    if (ws > 0) {  
        do {  
            node.prev = pred = pred.prev;  
        } while (pred.waitStatus > 0);  
        pred.next = node;  
    } else {  
        //CAS变更ws的状态
        compareAndSetWaitStatus(pred, ws, Node.SIGNAL);  
    }  
    return false;  
}  

上述方法是加锁失败开始执行的。也就是一个线程决定挂起之前需要执行的操作。这里就用到了节点中的信号量waitStatus

  1. 判断前驱节点waitStatus的值,会做出如下操作:
    1.1 前驱节点waitStatus若是-1,直接返回true。
    1.2 前驱节点waitStatus若大于0,证明前驱节点已被取消,那么在链表中删除前驱节点,直到node的前驱节点的waitStatus不大于0为止。然后返回false
    1.3. 若前驱节点waitStatus等于0,使用CAS尝试改变前驱节点waitStatus状态,由0到-1,然后返回false。

  2. 若是返回true,那么去阻塞该节点,若是返回false,那么继续自旋,继续上述过程,直至该方法返回true为止,方法返回true,便会执行下列方法,阻塞线程。

private final boolean parkAndCheckInterrupt() {  
    //将线程挂起
    LockSupport.park(this);  
    //线程被唤起时,查看线程的中断标识(注意,查看完毕后,中断标识归位)
    return Thread.interrupted();  
} 

需要注意的是:当前节点在阻塞之前,会将前驱节点的waitStatus设置为-1,就可保证前驱节点在适当的时机唤醒自己。

附录

对象的CAS算法

深入并发原理和大厂面试(一):AQS解析_第6张图片

开始我认为对象的CAS算法,实际上会是B对象去覆盖堆内存上的A对象,其实不然。比较交换的是引用。

//该方法是获取引用。而非堆上的内存。
static {  
    try {  
        valueOffset = unsafe.objectFieldOffset  
            (AtomicReference.class.getDeclaredField("value"));  
    } catch (Exception ex) { throw new Error(ex); }  
}  

3. 加锁总结

  1. 因为AQS的等待队列是延迟加载,只有多个线程并发访问时,才会开始维护队列。
  2. 因为head节点中不包含thread属性的值,又被称为哑节点
  3. head是正在办理业务的节点,而他的后继节点是第一个排队节点。
  1. 尝试加锁过程
  1. 根据status判断当前锁是否被持有,若被持有,直接维护队列
  2. 若未被持有,判断当前队列是否有节点在排队,若有节点排队,直接维护队列
  3. 若无节点排队,则通过CAS修改锁状态标识,修改成功代表线程持有该锁;
  4. 使用exclusiveOwnerThread来保存持有锁的线程(解决线程重入);
  1. 维护队列过程

最终线程的head节点为哑节点。后续线程被组装成node节点,维护在链表中。

  1. 线程阻塞过程
  1. 判断node节点是否为head节点的后续节点(第一个排队节点),若是的话,尝试获取锁。若获取到,将其设置为head节点,并将其设置为哑节点;
  2. 在阻塞前,会将自己的前驱节点的waitStatus设置为SIGNAL。以便可以唤醒自己。

你可能感兴趣的:(android,java,并发编程)