AQS-lock分析

AQS-lock分析,第1张

AQS的理解

AQS(abstractQueuedSynchronizer)也就是同步队列器,这个类非常重要,可以说是并发编程核心中的核心。他是锁具备synchronized功能的原因,也就是可以阻塞一个线程。

ReenTrantLock、Semaphore、Countdownlatch都是基于AQS来实现的。

阻塞一个线程的条件

既然AQS实现的功能是使锁具备synchronized的功能,也就是可以阻塞一个线程,那么需要满足四个条件

  1. 需要有一个state变量记录锁的状态,是否持有锁,持有的话是1,没有持有则是0。若是重入的则增加。
  2. 需要有一个变量,记录当前是哪个线程持有锁。
  3. 需要有一个底层支持,C,对线程的唤醒和阻塞 *** 作。
  4. 需要有一个队列,维护所有的阻塞线程,并且这个队列是线程安全的无锁队列。

对于1,维护了一个state来记录状态
对于2,Thread exclusiveOwnerThread可以记录
对于3,LockSupport封装了park,unpark方法实现线程的精准唤醒和阻塞

对于4,AQS维护了一个阻塞队列
它是核心中核心,这个我们需要首先熟知。如下可以知道,AQS的阻塞队列是一个双向链表。

具体如下:
Node是CLH队列的节点,代表“等待锁的线程队列”。
(01) 每个Node都会一个线程对应。
(02) 每个Node会通过prev和next分别指向上一个节点和下一个节点,这分别代表上一个等待线程和下一个等待线程。
(03) Node通过waitStatus保存线程的等待状态。
(04) Node通过nextWaiter来区分线程是“独占锁”线程还是“共享锁”线程。如果是“独占锁”线程,则nextWaiter的值为EXCLUSIVE;如果是“共享锁”线程,则nextWaiter的值是SHARED。

 private transient volatile Node head;

    /**
     * Tail of the wait queue, lazily initialized.  Modified only via
     * method enq to add new wait node.
     */
    private transient volatile Node tail;

    /**
     * The synchronization state.
     */
    private volatile int state;
private transient volatile Node head;    // CLH队列的队首
private transient volatile Node tail;    // CLH队列的队尾

// CLH队列的节点
static final class Node {
    static final Node SHARED = new Node();
    static final Node EXCLUSIVE = null;

    // 线程已被取消,对应的waitStatus的值
    static final int CANCELLED =  1;
    // “当前线程的后继线程需要被unpark(唤醒)”,对应的waitStatus的值。
    // 一般发生情况是:当前线程的后继线程处于阻塞状态,而当前线程被release或cancel掉,因此需要唤醒当前线程的后继线程。
    static final int SIGNAL    = -1;
    // 线程(处在Condition休眠状态)在等待Condition唤醒,对应的waitStatus的值
    static final int CONDITION = -2;
    // (共享锁)其它线程获取到“共享锁”,对应的waitStatus的值
    static final int PROPAGATE = -3;

    // waitStatus为“CANCELLED, SIGNAL, CONDITION, PROPAGATE”时分别表示不同状态,
    // 若waitStatus=0,则意味着当前线程不属于上面的任何一种状态。
    volatile int waitStatus;

    // 前一节点
    volatile Node prev;

    // 后一节点
    volatile Node next;

    // 节点所对应的线程
    volatile Thread thread;

    // nextWaiter是“区别当前CLH队列是 ‘独占锁’队列 还是 ‘共享锁’队列 的标记”
    // 若nextWaiter=SHARED,则CLH队列是“独占锁”队列;
    // 若nextWaiter=EXCLUSIVE,(即nextWaiter=null),则CLH队列是“共享锁”队列。
    Node nextWaiter;

    // “共享锁”则返回true,“独占锁”则返回false。
    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
    }

    // 构造函数。thread是节点所对应的线程,mode是用来表示thread的锁是“独占锁”还是“共享锁”。
    Node(Thread thread, Node mode) {     // Used by addWaiter
        this.nextWaiter = mode;
        this.thread = thread;
    }

    // 构造函数。thread是节点所对应的线程,waitStatus是线程的等待状态。
    Node(Thread thread, int waitStatus) { // Used by Condition
        this.waitStatus = waitStatus;
        this.thread = thread;
    }
}

入队,将新的Node加到tail后面,然后对tail进行CAS *** 作,出队就是对head进行CAS *** 作
一句话,入队添加到尾部,出队从头部出来。

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;
                }
            }
        }
    }

锁是如何进行阻塞和唤醒 *** 作的

在使用ReentrantLock时候,正常的使用场景是用 ReentrantLock.lock() 加锁,ReentrantLock.unlock() 解锁,try catch之中捕捉异常。
下面就从lock() 方法出发,探究一下lock底层是怎么做的。

			final void lock() {
            if (compareAndSetState(0, 1))
                setExclusiveOwnerThread(Thread.currentThread());
            else
                acquire(1);
        }

关注点放在 acquire(1) 上面。

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

(注意这里的判断是&&,只要有一个条件是false,后面的条件就终止判断了。)
1、先tryAcquire(arg)尝试获取锁,如果获取到了,就进行独占 *** 作(更改state,exclusiveOwnerThread设置等)。
2、如果tryAcquire(arg)没有获取锁(!tryAcquire(arg))== true,那么就要执行 //acquireQueued(addWaiter(Node.EXCLUSIVE), arg)先通过//addWaiter(Node.EXCLUSIVE)来将“当前线程”加入到"CLH队列(非阻塞的FIFO队列)"末尾。CLH队列就是线程等待队列。
执行完//addWaiter(Node.EXCLUSIVE)之后,会调用//acquireQueued()来获取锁。由于此时ReentrantLock是公平锁,它会根据公平性原则来获取锁.
“当前线程”在执行acquireQueued()时,会进入到CLH队列中休眠等待,直到获取锁了才返回!
3、如果acquireQueued(addWaiter(Node.EXCLUSIVE), arg)还是为true,也就是说在加入到阻塞队列之中,尝试获取锁,依旧没有成功,该线程已经被设置为阻塞状态。那么 selfInterrupt() 方法执行。

下面到每个方法中具体看看。

先看 tryAcquire(arg) 这个方法,这个方法是先尝试去获取到锁,这个方法有多个实现,以非公平锁的实现为例,这个方法是先获取状态 getState() ,无锁状态时上来就抢锁compareAndSetState(0, acquires)

final boolean nonfairTryAcquire(int acquires) {
            final Thread current = Thread.currentThread();
            int c = getState();
            if (c == 0) {
                if (compareAndSetState(0, acquires)) {
                    setExclusiveOwnerThread(current);
                    return true;
                }
            }
            else if (current == getExclusiveOwnerThread()) {
                int nextc = c + acquires;
                if (nextc < 0) // overflow
                    throw new Error("Maximum lock count exceeded");
                setState(nextc);
                return true;
            }
            return false;
        }

公平锁则不同,**!hasQueuedPredecessors()**需要先判断是否在队列的第一位,并且没有人持有锁

protected final boolean tryAcquire(int acquires) {
            final Thread current = Thread.currentThread();
            int c = getState();
            if (c == 0) {
                if (!hasQueuedPredecessors() &&
                    compareAndSetState(0, acquires)) {
                    setExclusiveOwnerThread(current);
                    return true;
                }
            }
            else if (current == getExclusiveOwnerThread()) {
                int nextc = c + acquires;
                if (nextc < 0)
                    throw new Error("Maximum lock count exceeded");
                setState(nextc);
                return true;
            }
            return false;
        }

获取到锁设置state,exclusiveOwnerThread等 *** 作,如果没有再执行下面的方法

之后关注acquireQueued(addWaiter(Node.EXCLUSIVE), arg)这个条件,先看(addWaiter(Node.EXCLUSIVE),这个是先增加一个节点到阻塞队列的尾节点上。

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;
            //先尝试添加到队列的尾部,如果不成功,再执行下面的enq
            if (compareAndSetTail(pred, node)) {
                pred.next = node;
                return node;
            }
        }
        //enq内部会进行队列的初始化
        enq(node);
        return node;
    }

当addWaiter加入到阻塞队列之后,接下来的工作就交给acquireQueued做了,p == head && tryAcquire(arg),如果发现自己是在队列头部,那么就去尝试获取锁,如果拿锁也成功,那么就需要更改队列结构(tryAcquire(arg)这个方法已经设置了当前线程之类的)。

如果失败,那么就会去执行shouldParkAfterFailedAcquire(p, node) && parkAndCheckInterrupt() 这俩个方法,

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;
                }
                // 执行到这里,说明
                // 已经尝试过获取锁了,但还是失败了(当然有可能是因为p != head)
                if (shouldParkAfterFailedAcquire(p, node) &&
                    parkAndCheckInterrupt())
                    interrupted = true;
            }
        } finally {
            if (failed)
                cancelAcquire(node);
        }
    }

shouldParkAfterFailedAcquire(p, node) && parkAndCheckInterrupt() 这俩个方法非常重要!!
这段代码主要检查当前线程是否需要被阻塞,具体规则如下:

  1. 如果当前线程的前驱节点状态为SINNAL,则表明当前线程需要被阻塞,调用unpark()方法唤醒,直接返回true,当前线程阻塞
  2. 如果当前线程的前驱节点状态为CANCELLED(ws > 0),则表明该线程的前驱节点已经等待超时或者被中断了,则需要从CLH队列中将该前驱节点删除掉,直到回溯到前驱节点状态 <= 0 ,返回false
  3. 如果前驱节点非SINNAL,非CANCELLED,则通过CAS的方式将其前驱节点设置为SINNAL,返回false:
private static boolean shouldParkAfterFailedAcquire(Node pred, Node node) {
    //前驱节点的状态
    int ws = pred.waitStatus;
    //状态为SIGNAL表示当前线程处于等待状态,直接返回true
    if (ws == Node.SIGNAL)
        /*
         * This node has already set status asking a release
         * to signal it, so it can safely park.
         */
        return true;
    if (ws > 0) {
        //前驱节点状态>0,则为Canclled,表明该节点已经超时或者被中断,需要从同步队列删除该前驱节点,直到前驱节点状态小于0
        /*
         * Predecessor was cancelled. Skip over predecessors and
         * indicate retry.
         */
        do {
            node.prev = pred = pred.prev;
        } while (pred.waitStatus > 0);
        pred.next = node;
    } else {
        //前驱节点状态为Condition、PROPAGATE
        /*
         * waitStatus must be 0 or PROPAGATE.  Indicate that we
         * need a signal, but don't park yet.  Caller will need to
         * retry to make sure it cannot acquire before parking.
         */
        //通过CAS的方式将前驱节点的状态设置为SIGNAL
        compareAndSetWaitStatus(pred, ws, Node.SIGNAL);
    }
    return false;
}

```java
private final boolean parkAndCheckInterrupt() {
			//这里是对需要阻塞的线程进行阻塞
        LockSupport.park(this);
        return Thread.interrupted();
    }

欢迎分享,转载请注明来源:内存溢出

原文地址: http://outofmemory.cn/langs/788203.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2022-05-05
下一篇 2022-05-05

发表评论

登录后才能评论

评论列表(0条)

保存