zoukankan      html  css  js  c++  java
  • JUC中AQS简介

    AQS,在java.util.concurrent.locks包中,AbstractQueuedSynchronizer这个类是并发包中的核心,了解其他类之前,需要先弄清楚AQS。在JUC的很多类中都会存在一个内部类Sync,Sync都是继承自AbstractQueuedSynchronizer,相信不用说就能明白AQS有多重要。

    AQS原理

    AQS就是一个同步器,要做的事情就相当于一个锁,所以就会有两个动作:一个是获取,一个是释放。获取释放的时候该有一个东西来记住他是被用还是没被用,这个东西就是一个状态。如果锁被获取了,也就是被用了,还有很多其他的要来获取锁,总不能给全部拒绝了,这时候就需要他们排队,这里就需要一个队列。这大概就清楚了AQS的主要构成了:

    • 获取和释放两个动作
    • 同步状态(原子操作)
    • 阻塞队列

    state

    AQS用32位整形来表示同步状态。

    1
    private volatile int state;

    在互斥锁中表示线程是否已经获取了锁,0未获取,1已经获取,大于1表示重入数。

    AQS提供了getState(),setState(),compareAndSetState()来获取和修改state的值,这些操作需要atomic包的支持,采用CAS操作,保证其原子性和可见性。

    AQS的CLH锁队列

    CLH其实就是一个FIFO的队列,只不过稍微做了点改进。AQS中内部使用内部类Node来实现,是一个链表队列,原始CLH使用自旋锁,AQS的CLH则在每个node里使用一个状态字段来控制阻塞,不是自旋。直接看代码:

    1
    2
    3
    4
    5
    6
    7
    8
    9
    10
    11
    12
    13
    14
    15
    16
    17
    18
    19
    20
    21
    22
    23
    24
    25
    26
    27
    28
    29
    30
    31
    32
    33
    34
    35
    36
    37
    38
    39
    40
    41
    42
    43
    44
    45
    46
    47
    48
    49
    50
    51
    52
    53
    54
    /**
    +------+ prev +-----+ +-----+
    head | | <---- | | <---- | | tail
    +------+ +-----+ +-----+
    /**
    static final class Node {
    //作为共享模式
    static final Node SHARED = new Node();
    //作为独占模式
    static final Node EXCLUSIVE = null;
    //等待状态:表示节点中线程是已被取消的
    static final int CANCELLED = 1;
    //等待状态:表示当前节点的后继节点的线程需要被唤醒
    static final int SIGNAL = -1;
    //等待状态:表示线程正在等待条件
    static final int CONDITION = -2;
    //等待状态:表示下一个共享模式的节点应该无条件的传播下去
    static final int PROPAGATE = -3;
    //等待状态,初始化为0,剩下的状态就是上面列出的
    volatile int waitStatus;
    //当前节点的前驱节点
    volatile Node prev;
    //后继节点
    volatile Node next;
    //当前节点的线程
    volatile Thread thread;
    //
    Node nextWaiter;
    //是否是共享节点
    final boolean isShared() {
    return nextWaiter == SHARED;
    }
    //当前节点的前驱节点
    final Node predecessor() throws NullPointerException {
    Node p = prev;
    if (p == null)
    throw new NullPointerException();
    else
    return p;
    }

    Node() { // Used to establish initial head or SHARED marker
    }

    Node(Thread thread, Node mode) { // Used by addWaiter
    this.nextWaiter = mode;
    this.thread = thread;
    }

    Node(Thread thread, int waitStatus) { // Used by Condition
    this.waitStatus = waitStatus;
    this.thread = thread;
    }
    }

    共享锁和互斥锁

    AQS的CLH队列锁中,每个节点代表着一个需要获取锁的线程,该node中有两个常量SHARED共享模式,EXCLUSIVE独占模式。

    1
    2
    3
    4
    /** Marker to indicate a node is waiting in shared mode */
    static final Node SHARED = new Node();
    /** Marker to indicate a node is waiting in exclusive mode */
    static final Node EXCLUSIVE = null;

    共享模式允许多个线程可以获取同一个锁,独占模式则一个锁只能被一个线程持有,其他线程必须要等待。

    AQS源码

    1
    2
    3
    4
    5
    6
    7
    8
    9
    10
    11
    12
    13
    14
    15
    16
    17
    18
    19
    20
    //阻塞队列的队列头
    private transient volatile Node head;
    //队列尾
    private transient volatile Node tail;
    //同步状态,这就是上面提到的需要原子操作的状态
    private volatile int state;
    //返回当前同步器的状态
    protected final int getState() {
    return state;
    }
    //设置同步器的状态
    protected final void setState(int newState) {
    state = newState;
    }
    //原子的设置当前同步器的状态
    protected final boolean compareAndSetState(int expect, int update) {
    return unsafe.compareAndSwapInt(this, stateOffset, expect, update);
    }
    //
    static final long spinForTimeoutThreshold = 1000L;

    独占模式的获取

    acquire,独占,忽略中断

    1
    2
    3
    4
    5
    6
    7
    8
    9
    10
    //独占模式的获取方法,会忽略中断
    //tryAcquire方法会被至少调用一次,由子类实现
    //如果tryAcquire不能成功,当前线程就会进入队列排队
    public final void acquire(int arg) {
    //首先调用tryAcquire尝试获取
    //获取不成功,就使用acquireQueued使线程进入等待队列
    if (!tryAcquire(arg) &&
    acquireQueued(addWaiter(Node.EXCLUSIVE), arg))
    selfInterrupt();
    }

    tryAcquire方法:

    1
    2
    3
    4
    5
    //由子类来实现
    //尝试在独占模式下获取,会查询该对象的状态是否允许在独占模式下获取
    protected boolean tryAcquire(int arg) {
    throw new UnsupportedOperationException();
    }

    使用指定的模式创建一个节点,添加到AQS链表队列中:

    1
    2
    3
    4
    5
    6
    7
    8
    9
    10
    11
    12
    13
    14
    15
    16
    private Node addWaiter(Node mode) {
    //当前线程,指定的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;
    }
    }
    //使用添加节点的方法
    enq(node);
    return node;
    }

    向队列中插入节点:

    1
    2
    3
    4
    5
    6
    7
    8
    9
    10
    11
    12
    13
    14
    15
    16
    17
    18
    19
    //会插入节点到对列中
    private Node enq(final Node node) {
    for (;;) {
    //尾节点
    Node t = tail;
    //需要实例化一个队列
    if (t == null) { // Must initialize
    //使用cas创建头节点
    if (compareAndSetHead(new Node()))
    tail = head;
    } else {
    node.prev = t;
    if (compareAndSetTail(t, node)) {
    t.next = node;
    return t;
    }
    }
    }
    }

    tryAcquire没有获取到,就会先使用addWaiter添加进队列,然后使用acquireQueued从队列获取,如果这时候获取成功,则替换当前节点为队列头,然后返回:

    1
    2
    3
    4
    5
    6
    7
    8
    9
    10
    11
    12
    13
    14
    15
    16
    17
    18
    19
    20
    21
    22
    23
    24
    25
    26
    27
    28
    29
    30
    31
    32
    33
    34
    35
    36
    //独占模式处理正在排队等待的线程。
    //自旋,直至获取成功才返回
    final boolean acquireQueued(final Node node, int arg) {
    //当前获取是否失败
    boolean failed = true;
    try {
    //获取是否被中断
    boolean interrupted = false;
    for (;;) {
    //获取当前节点的前驱节点
    final Node p = node.predecessor();

    //head节点要么是刚才初始化的节点
    //要么就是成功获取锁的节点
    //如果当前节点的前驱节点是head,当前节点就应该去尝试获取锁了
    //当前节点的前驱节点是头节点,就尝试获取
    if (p == head && tryAcquire(arg)) {
    //获取成功的话,就把当前节点设置为头节点
    setHead(node);
    //之前的head节点的next引用设为null
    p.next = null; // help GC
    failed = false;
    return interrupted;
    }
    //查看当前节点是否应该被park
    //如果应该,就park当前线程
    if (shouldParkAfterFailedAcquire(p, node) &&
    parkAndCheckInterrupt())
    interrupted = true;
    }
    } finally {
    //失败了,取消当前线程
    if (failed)
    cancelAcquire(node);
    }
    }

    设置头节点,只能被获取方法调用:

    1
    2
    3
    4
    5
    private void setHead(Node node) {
    head = node;
    node.thread = null;
    node.prev = null;
    }

    shouldParkAfterFailedAcquire方法,查看是否应该被park:

    1
    2
    3
    4
    5
    6
    7
    8
    9
    10
    11
    12
    13
    14
    15
    16
    17
    18
    19
    20
    21
    private static boolean shouldParkAfterFailedAcquire(Node pred, Node node) {
    //前驱节点中保存的等待状态
    int ws = pred.waitStatus;
    //等待状态是signal,也就是当前节点在等着被唤醒
    //此时当前节点应该park
    if (ws == Node.SIGNAL)
    return true;

    //等待状态大于0表示前驱节点已经取消
    //会向前找到一个非取消状态的节点
    if (ws > 0) {
    do {
    node.prev = pred = pred.prev;
    } while (pred.waitStatus > 0);
    pred.next = node;
    } else {
    //将前驱节点的waitStatus设置为signal,表示当前需要被park
    compareAndSetWaitStatus(pred, ws, Node.SIGNAL);
    }
    return false;
    }

    看下parkAndCheckInterrupt方法:

    1
    2
    3
    4
    5
    6
    //挂起当前线程,并返回当前中断状态
    private final boolean parkAndCheckInterrupt() {
    //挂起当前线程
    LockSupport.park(this);
    return Thread.interrupted();
    }

    cancelAcquire取消当前节点:

    1
    2
    3
    4
    5
    6
    7
    8
    9
    10
    11
    12
    13
    14
    15
    16
    17
    18
    19
    20
    21
    22
    23
    24
    25
    26
    27
    28
    29
    30
    31
    32
    33
    34
    35
    36
    37
    38
    private void cancelAcquire(Node node) {
    //节点不存在
    if (node == null)
    return;
    //节点的线程引用设为null
    node.thread = null;

    //前驱节点
    Node pred = node.prev;
    //大于0表示前驱节点被取消
    while (pred.waitStatus > 0)
    node.prev = pred = pred.prev;

    //前驱节点的下一个是需要移除的节点
    Node predNext = pred.next;

    //设置节点状态为取消
    node.waitStatus = Node.CANCELLED;

    //如果是尾节点,直接取消,将前一个节点设置为尾节点
    if (node == tail && compareAndSetTail(node, pred)) {
    compareAndSetNext(pred, predNext, null);
    } else {//不是尾节点,说明有后继节点,将前驱节点的next纸箱后继节点
    int ws;
    if (pred != head &&
    ((ws = pred.waitStatus) == Node.SIGNAL ||
    (ws <= 0 && compareAndSetWaitStatus(pred, ws, Node.SIGNAL))) &&
    pred.thread != null) {
    Node next = node.next;
    if (next != null && next.waitStatus <= 0)
    compareAndSetNext(pred, predNext, next);
    } else {
    unparkSuccessor(node);
    }

    node.next = node; // help GC
    }
    }

    acquireInterruptibly 独占,可中断

    跟独占忽略中断类似,不再解释。

    tryAcquireNanos,独占,可超时,可中断

    跟上面类似,但是在doAcquireNanos中会获取当前时间,并获取LockSupport.parkNanos之后的时间在做超时时间的重新计算,到了超时时间,就返回false。

    独占模式的释放

    release,独占,忽略中断

    1
    2
    3
    4
    5
    6
    7
    8
    9
    10
    11
    12
    13
    public final boolean release(int arg) {
    //尝试释放,修改状态
    if (tryRelease(arg)) {
    //成功释放
    //head代表初始化的节点,或者是当前占有锁的节点
    //需要unpark后继节点
    Node h = head;
    if (h != null && h.waitStatus != 0)
    unparkSuccessor(h);
    return true;
    }
    return false;
    }

    unparkSuccessor:

    1
    2
    3
    4
    5
    6
    7
    8
    9
    10
    11
    12
    13
    14
    15
    16
    17
    18
    19
    20
    private void unparkSuccessor(Node node) {
    //头节点中保存的waitStatus
    int ws = node.waitStatus;
    //重置头节点状态为0
    if (ws < 0)
    compareAndSetWaitStatus(node, ws, 0);
    //后继节点
    Node s = node.next;
    //后继节点为null或者已经取消
    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;
    }
    //unpark
    if (s != null)
    LockSupport.unpark(s.thread);
    }

    共享模式的获取

    acquireShared,共享,忽略中断

    acquireSharedInterruptibly,共享,可中断

    tryAcquireSharedNanos,共享,可设置超时,可中断

    共享模式的释放

    releaseShared

    共享模式的和独占模式基本差不多,和独占式的acquireQueued方法区别就是在获取成功的节点后会继续unpark后继节点,将共享状态向后传播。

    LockSupport

    用来创建锁和其他同步类的基本线程阻塞原语。每个使用LockSupport的线程都会与一个许可关联,如果该许可可用并且可在进程中使用,则调用park()将会立即返回,否则可能阻塞。如果许可不可用,可调用unpark使其可用。

    许可不可重入,只能调用一次park()方法,否则会一直阻塞。

    park()和unpark()作用分别是阻塞线程和解除阻塞线程,且park和unpark不会遇到suspend和resume可能引发的死锁问题。

    park,如果许可可用,使用该许可,并且该调用立即返回;否则为线程调度禁用当前线程,并在发生以下三种情况之一之前,使其处于休眠状态:

    * 其他某个线程将当前线程作为目标调用unpark
    * 其他某个线程中断当前线程
    * 该调用不合逻辑的返回
    

    unpark,如果给定的线程尚不可用,则使其可用。如果线程在park上受阻塞,则它将解除其阻塞状态。否则,保证下一次调用park不受阻塞。如果给定线程尚未启动,则无法保证此操作有任何效果。

  • 相关阅读:
    C++笔记-智能指针 shared_ptr
    Linux笔记-性能调优工具perf
    git submodule 如何push代码
    性能测试工具gperftools使用
    Linux信号使用及自定义信号
    DNN在推荐系统中的应用参考资料
    vscode远程代码同步
    感知机模型到DNN模型
    c++笔记-libcurl多线程并发时的core【转载】
    go笔记-熔断器
  • 原文地址:https://www.cnblogs.com/zhangboyu/p/7452575.html
Copyright © 2011-2022 走看看