[Java][并发编程]AQS以及其相关同步器的源码解析

AQS以及其相关同步器的源码解析

概念

AQSAbstractQueuedSynchronizer)抽象的队列同步器。是用来构建锁或者其他同步器组件的重量级基础框架以及整个JUC体系的基石。通过内置的 FIFO 队列(先入先出队列)来完成资源获取线程的排队工作,将每条要去抢占资源的线程封装成一个 Node 节点来实现锁的分配,并通过一个 volatile 类型的 int 类型变量表示持有锁的状态,使用 CAS 完成对 state 值的修改。

所以我对AQS简单粗暴的理解:AQS = state + FIFO队列。

其中NodeCLH队列中的节点,是AQS中最基本的数据结构。

Node = waitStatus + Thread

AQS有关的类:

  • ReentrantLock
  • CountDownLatch
  • ReentrantReadWriteLock
  • Semephore
  • CyclicBarrier
  • .......

这些类的底层都有继承了AQS,所以可以说AQSJUC中最重要的基石。

下面为AQS相关的UML图:

ReentrantLock

ReentrantLock可重入锁,指一个线程可以对一个临界资源重复加锁。为独占锁,不支持共享。

ReentrantLock提供两个构造方法,有参构造是根据参数创建公平锁或非公平锁,而无参构造方式默认是非公平锁,因为非公平锁的性能高,大部分业务使用的都是非公平锁。

public ReentrantLock() {
    sync = new NonfairSync();
}

public ReentrantLock(boolean fair) {
    sync = fair ? new FairSync() : new NonfairSync();
}

所以可以看到,ReentrantLock有两种加锁方式:公平锁(FairSync)与非公平锁(NonfairSync)。

使用方式

public class AQSDemo {
    // static ReentrantLock lock = new ReentrantLock();
    static ReentrantLock lock = new ReentrantLock(true);

    static int num = 10;

    public static void main(String[] args) {
        new Thread(AQSDemo::test, "线程1").start();
        new Thread(AQSDemo::test, "线程2").start();
        new Thread(AQSDemo::test, "线程3").start();
    }

    static void test() {
        String name = Thread.currentThread().getName();
        while (num > 1) {
            lock.lock();
            try{
                num--;
                System.out.printf("%s:%d%n", name, num);
            } finally {
                lock.unlock();
            }
        }
    }
}

上面这段代码,非公平锁和公平锁的打印结果是不同的。

非公平锁结果:

线程19
线程18
线程17
线程16
线程15
线程14
线程13
线程12
线程11
线程20
线程3:-1

公平锁结果:

线程19
线程28
线程37
线程16
线程25
线程34
线程13
线程22
线程31
线程10
线程2:-1

源码

ReentrantLock的内部类Sync,该类实现了AQS。又有两个子类实现了Sync类:FairSyncNonfairSync

公平锁:讲究的是先来后到,先进先出。如果这个锁的等待队列已经有线程在等待,那么当前线程就会进入等待队列当中排队。

非公平锁:没有排队的概念,谁抢到是谁的。是需要抢占的锁。

公平锁和非公平锁的不同就是,非公平锁首先会调用CASstate从0改为1,成功则表示获取到了锁,使用这个setExclusiveOwnerThread方法记录下当前占用state的线程;否则与公平锁一样调用acquire方法获取锁。acquireAQS中提供的模板方法:

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

tryAcquire 尝试着获取 state,如果成功,跳过后面的步骤;如果失败,则执行 acquireQueued 方法将线程加入 CLH 等待队列中。

tryAcquire

非公平锁和公平锁的实现方式有些不同。

首先看看非公平锁的tryAcquire

protected final boolean tryAcquire(int acquires) {
    return nonfairTryAcquire(acquires);
}

/**
 * Performs non-fair tryLock.  tryAcquire is implemented in
 * subclasses, but both need nonfair try for trylock method.
 */
final boolean nonfairTryAcquire(int acquires) {
    final Thread current = Thread.currentThread();
    int c = getState();
    if (c == 0) {
        // 如果 c(锁状态)为0,表示此时的资源是空闲的,则使用CAS获取锁,记录当前占用锁的线程,并返回
        // 这里可以看出是非公平锁,因为所有线程都是用CAS获取锁的,不需要排队
        if (compareAndSetState(0, acquires)) {
            setExclusiveOwnerThread(current);
            return true;
        }
    }
    // 如果已有线程获得锁,且该线程再次获得了锁,获取资源数 +1
    // 这里可以印证 ReentrantLock 是可重入锁
    else if (current == getExclusiveOwnerThread()) {
        int nextc = c + acquires;
        if (nextc < 0) // overflow
            throw new Error("Maximum lock count exceeded");
        // 更新state,这里不需要用CAS更新,此时的锁就是当前线程占有的,其他线程没机会执行。此时更新state是线程安全的
        setState(nextc);
        return true;
    }
    return false;
}

然后是公平锁的tryAcquire。基本上和非公平锁的代码是一样的,区别在于加锁的时候需要判断是否已经有队列存在,没有采取加锁,有则直接返回false

/**
 * Fair version of tryAcquire.  Don't grant access unless
 * recursive call or no waiters or is first.
 */
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;
}


public final boolean hasQueuedPredecessors() {
    Node t = tail; 
    Node h = head;
    Node s;
    // 首尾不相等,且head节点线程不是当前线程则表示需要进入队列
    return h != t &&
        ((s = h.next) == null || s.thread != Thread.currentThread());
}

acquireQueued

如果tryAcquire尝试加锁失败,则会执行acquireQueued这个方法,也就是将线程加入到FIFO等待队列。

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

可以看到调用了addWaiter方法,将当前线程的Node节点入队。从Node.EXCLUSIVE可以看出,这个节点是独占模式。

addwaiter代码:

private Node addWaiter(Node mode) {
    Node node = new Node(Thread.currentThread(), mode);
    Node pred = tail;
    // 尾节点不为null表示已经存在队列,直接将当前线程作为尾节点
    if (pred != null) {
        node.prev = pred;
        // 如果尾节点存在,使用CAS将等待线程入队
        if (compareAndSetTail(pred, node)) {
            pred.next = node;
            return node;
        }
    }
    // 如果节点是空的,则执行 enq 方法
    enq(node);
    return node;
}

enq方法代码:

private Node enq(final Node node) {
    for (;;) {
        Node t = tail;
        // 判断尾节点是否为空,如果是空的,说明FIFO队列的head、tail还未构建,需要先构建head节点,然后使用CAS入队
        if (t == null) { // Must initialize
            if (compareAndSetHead(new Node()))
                tail = head;
        } else {
            // 尾节点不为空,将等待线程入队
            node.prev = t;
            if (compareAndSetTail(t, node)) {
                t.next = node;
                return t;
            }
        }
    }
}

问:使用 CAS 创建 head 节点的时候只是简单调用了 new Node() 方法,并不像其他节点那样记录 thread,为什么?

答:因为 head 节点只代表当前有线程占用了 state,至于占用 state 的是哪个线程,之前有调用了 setExclusiveOwnerThread 方法,即已经记录在 exclusiveOwnerThread 属性里了。

现在有个问题,就是入队后的线程,要如何处理?是马上阻塞吗?

如果是马上阻塞意味着线程从运行态转为阻塞态,涉及到用户态向内核态的切换,而且唤醒后也要从内核态转为用户态,开销相对比较大。所以AQS对这种入队的线程采用的方式,是通过让它们自旋来竞争锁,

但是当前锁是独占锁,如果锁一直被被某个线程占有,其他等待队列中的线程一直自旋没太大意义,反而会占用CPU影响性能。

所以更合适的方式是,它们自旋一两次次数,竞争不到锁后识趣地阻塞以等待前置节点释放锁后再来唤醒它。

另外如果锁在自旋过程中被中断了,或者自旋超时了,应该处于「取消」状态。

所以,在Node节点中就定义了waitStatus这个变量,来记录每个Node可能所处的状态。在前面的概念介绍时,有对各个状态进行了展示。

// 由于超时或中断,节点已被取消
static final int CANCELLED = 1; 
// 节点阻塞(park)必须在其前驱结点为 SIGNAL 的状态下才能进行,如果结点为 SIGNAL,则其释放锁或取消后,可以通过 unpark 唤醒下一个节点,
static final int SIGNAL = -1;  
// 表示线程在等待条件变量(先获取锁,加入到条件等待队列,然后释放锁,等待条件变量满足条件;只有重新获取锁之后才能返回)
static final int CONDITION = -2;
// 表示后续结点会传播唤醒的操作,共享模式下起作用
static final int PROPAGATE = -3;

//等待状态
volatile int waitStatus;

接下来才真正到了acquireQueued的代码,加锁的核心逻辑:

final boolean acquireQueued(final Node node, int arg) {
    boolean failed = true;
    try {
        boolean interrupted = false;
        for (;;) {
            final Node p = node.predecessor();
             // 如果前一个节点是 head,且尝试自旋获取锁成功
            if (p == head && tryAcquire(arg)) {
                // 将 head 结点指向当前节点,原 head 结点出队。这样原 head 不可达,会被GC
                setHead(node);
                p.next = null; // help GC
                failed = false;
                return interrupted;
            }
            // 如果前一个节点不是 head 或者竞争锁失败,则判断锁是否应该停止自旋进入阻塞状态
            // s为true表示线程可以进入阻塞中断,调用parkAndCheckInterrupt让线程阻塞
            if (shouldParkAfterFailedAcquire(p, node) &&
                parkAndCheckInterrupt())
                interrupted = true;
        }
    } finally {
        if (failed)
            // 如果线程自旋中因为异常等原因获取锁最终失败,则取消加锁节点
            cancelAcquire(node);
    }
}


private static boolean shouldParkAfterFailedAcquire(Node pred, Node node) {
    int ws = pred.waitStatus;
    // 如果前置节点的状态是 SIGNAL,则当前节点可以进入阻塞状态
    if (ws == Node.SIGNAL)
        return true;
    // 如果前置节点为取消状态,则前置节点需要移除
    if (ws > 0) {
        do {
            node.prev = pred = pred.prev;
        } while (pred.waitStatus > 0);
        pred.next = node;
    } else {
        // 将前置节点的 waitStatus 设为 SIGNAL
        compareAndSetWaitStatus(pred, ws, Node.SIGNAL);
    }
    return false;
}

unlock

无论是非公平锁还是公平锁,解锁中也是调用AQS中的模板方法:

public final boolean release(int arg) {
    // 尝试释放锁是否成功
    if (tryRelease(arg)) {
        Node h = head;
        if (h != null && h.waitStatus != 0)
            // 锁释放成功后,唤醒 head 之后的节点,让它来竞争锁
            unparkSuccessor(h);
        return true;
    }
    return false;
}

tryRelease是定义在ReentrantLock的内部类Sync中的:

protected final boolean tryRelease(int releases) {
    int c = getState() - releases;
    // 只有持有锁的线程才能释放锁,所以如果当前锁不是持有锁的线程,则抛异常
    if (Thread.currentThread() != getExclusiveOwnerThread())
        throw new IllegalMonitorStateException();
    boolean free = false;
    // 说明线程持有的锁全部释放了,需要释放 exclusiveOwnerThread 的持有线程
    if (c == 0) {
        free = true;
        setExclusiveOwnerThread(null);
    }
    setState(c);
    return free;
}

unparkSuccessor唤醒节点代码:


private void unparkSuccessor(Node node) {
    // 获取节点的 waitStatus
    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);
}

公平锁和非公平锁的区别就在于,非公平锁不管是否有线程在排队,先抢锁;而公平锁则会判断是否存在队列,有线程在排队则直接进入队列排队。

另外线程在park被唤醒后非公平锁还会抢锁,公平锁仍然需要排队。所以非公平锁的性能比公平锁高很多,大部分情况下我们使用非公平锁即可。

总结

ReentrantLock是可重入的互斥锁,虽然具有与synchronized相同功能,但是会比synchronized更加灵活,且能提供更多的方法。与synchronized的异同如下:

ReentrantLock synchronized
锁实现机制 依赖AQS 监视器模式
灵活性 支持响应中断、超时、尝试获取锁 无法设置超时时间,无法提供外部方法
释放形式 必须显示调用unlock()释放锁 自动释放监视器
锁类型 公平锁和非公平锁 非公平锁
条件队列 可关联多个条件队列 关联一个条件队列
可重入性 可重入 可重入

ReentrantLock底层基于AQS实现,UML关系图如下。

Semaphore

使用方式

Semaphore(信号量)是AQS共享模式的一个应用,允许多个线程同时对共享资源进行操作,并且可以有效的控制并发数,利用它可以很好的实现流量控制。Semaphore提供了两个带参构造器,这两个构造器都必须传入一个初始的许可证数量,使用构造器一构造出来的信号量在获取许可证时会采用非公平方式获取,使用构造器二可以通过参数指定获取许可证的方式(公平或者非公平)。


public Semaphore(int permits) {
    sync = new NonfairSync(permits);
}

public Semaphore(int permits, boolean fair) {
    sync = fair ? new FairSync(permits) : new NonfairSync(permits);
}

Semaphore提供了很多方法来获取许可,这里列举一下:

  • void acquire() throws InterruptedException 获取一个许可,会阻塞等待其他线程释放许可
  • void acquire(int permits) throws InterruptedException 获取指定的许可数,会阻塞等待其他线程释放
  • void acquireUninterruptibly() 获取一个许可,会阻塞等待其他线程释放许可,可以被中断
  • void acquireUninterruptibly(int permits) 获取指定的许可数,会阻塞等待其他线程释放许可,可以被中断
  • boolean tryAcquire() 尝试获取许可,不会进行阻塞等待
  • boolean tryAcquire(int permits) 尝试获取指定的许可数,不会阻塞等待
  • boolean tryAcquire(long timeout, TimeUnit unit) 尝试获取许可,可指定等待时间
  • boolean tryAcquire(int permits, long timeout, TimeUnit unit) 尝试获取指定的许可数,可指定等待时间

释放许可则是使用了release方法。

源码

获取许可的源码先浅看这四个:

public void acquire() throws InterruptedException {
   sync.acquireSharedInterruptibly(1);
}
public void acquireUninterruptibly() {
   sync.acquireShared(1);
}
public boolean tryAcquire() {
   return sync.nonfairTryAcquireShared(1) >= 0;
}
public boolean tryAcquire(long timeout, TimeUnit unit) throws InterruptedException {
   return sync.tryAcquireSharedNanos(1, unit.toNanos(timeout));
}

上面的是Semaphore提供的默认获取许可证操作,每次只获取一个许可证。

acquire

先看看acquire获取许可里面的acquireSharedInterruptibly这个方法。这个方法是AQS里面的方法。

public final void acquireSharedInterruptibly(int arg)
        throws InterruptedException {
    // 如果线程中断抛出异常
    if (Thread.interrupted())
        throw new InterruptedException();
    // 尝试获取锁(共享锁)
    if (tryAcquireShared(arg) < 0)
        doAcquireSharedInterruptibly(arg);
}

其中tryAcquireShared也是在AQS里面是抽象方法,FairSyncNonfairSync这两个派生类实现了该方法的逻辑。FairSync实现的是公平获取的逻辑,而NonfairSync实现的非公平获取的逻辑。

// 公平锁获取逻辑
static final class FairSync extends Sync {
    // ...省略代码
    protected int tryAcquireShared(int acquires) {
        for (;;) {
            // 如果队列中有线程排队则获取锁失败
            if (hasQueuedPredecessors())
                return -1;
        	// 获取可用许可
            int available = getState();
        	// 获取剩余可用许可
            int remaining = available - acquires;
            // 如果剩余许可小于0则直接返回,否则先更新许可状态之后再返回
            if (remaining < 0 ||
                compareAndSetState(available, remaining))
                return remaining;
        }
    }
}

// 非公平锁获取逻辑
static final class NonfairSync extends Sync {
    // ...省略代码
    protected int tryAcquireShared(int acquires) {
        return nonfairTryAcquireShared(acquires);
    }
}
// 非公平锁获取逻辑
final int nonfairTryAcquireShared(int acquires) {
    for (;;) {
        int available = getState();
        int remaining = available - acquires;
        if (remaining < 0 ||
            compareAndSetState(available, remaining))
            return remaining;
    }
}

可以看到公平锁和非公平锁的获取代码很像,区别在于是否需要排队等待。

对于tryAcquireShared获取锁的返回值含义,返回负数表示获取失败,零表示当前线程获取成功但后续线程不能再获取,正数表示当前线程获取成功并且后续线程也能够获取。

acquireUninterruptibly

可以被中断的获取阻塞的方法。

public void acquireUninterruptibly() {
    sync.acquireShared(1);
}

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

可以看到其实还是调用了tryAcquireShared尝试获取锁的方法,不做赘述。

tryAcquire

public boolean tryAcquire() {
    return sync.nonfairTryAcquireShared(1) >= 0;
}

可以看到直接调用的就是非公平锁的代码,前面的acquire中也有说明具体实现,不做赘述。

release

它的操作很简单,就是调用了AQSreleaseShared方法。

// 释放一个许可
public void release() {
    sync.releaseShared(1);
}

// 释放共享锁
public final boolean releaseShared(int arg) {
    // 尝试释放锁
    if (tryReleaseShared(arg)) {
        // 如果成功了则唤醒其他线程
        doReleaseShared();
        return true;
    }
    return false;
}

// 尝试释放锁
protected final boolean tryReleaseShared(int releases) {
    // 自旋
    for (;;) {
        // 获取锁状态
        int current = getState();
        int next = current + releases;
        if (next < current) // overflow
            throw new Error("Maximum permit count exceeded");
        // 以CAS的方式更新锁状态成功返回true,否则继续自旋
        if (compareAndSetState(current, next))
            return true;
    }
}

总结

Semaphore的底层也是AQS,它内部维护了一个计数器,可加可减,acquire()方法是做减法,release()方法是做加法,可基于Semaphore实现限流操作。

SemaphoreAQSUML图如下:

CountDownLatch

CountDownLatch可被称为门阀、计数器或者闭锁,它能够使一个线程在等待另外一些线程完成各自工作之后,再继续执行。它相当于是一个计数器,这个计数器的初始值就是线程的数量,每当一个任务完成后,计数器的值就会减一,当计数器的值为 0 时,表示所有的线程都已经任务了,然后在 CountDownLatch 上等待的线程就可以恢复执行接下来的任务。

CountDownLatch有一个典型的应用场景就是当一个服务启动时,同时会加载很多组件和服务,这时候主线程会等待组件和服务的加载。当所有的组件和服务都加载完毕后,主线程和其他线程在一起完成某个任务。

提供了一个构造方法,必须指定其初始值。还指定了 countDown 方法,这个方法的作用主要用来减小计数器的值,当计数器变为 0 时,在 CountDownLatchawait 的线程就会被唤醒,继续执行其他任务。

// 构造函数
public CountDownLatch(int count) {
    if (count < 0) throw new IllegalArgumentException("count < 0");
    this.sync = new Sync(count);
}
// countDown方法 
// 这个方法与前面Semaphore提供的释放许可的代码是一样的,用的都是AQS中的方法。
public void countDown() {
    sync.releaseShared(1);
}
// await方法
// 同样与之前Semaphore获取锁acquire中的方法相同,也是AQS的方法,这个方法将使当前线程在计数器减到0之前一直等待,除非线程被中断。
public void await() throws InterruptedException {
    sync.acquireSharedInterruptibly(1);
}

可以看到countDown方法和await方法都是基于的AQS中的方法。

CountDownLatchAQS之间的UML图如下:

CyclicBarrier

CyclicBarrier类可以实现一组线程相互等待,当所有线程都到达某个屏障点后再进行后续的操作。其实说白了,就是等待线程数达到指定的数量后才会执行指定的程序。

CyclicBarrier提供两个构造函数,

CyclicBarrier类的内部有一个计数器,每个线程在到达屏障点的时候都会调用await方法将自己阻塞,此时计数器会减1,当计数器减为0的时候所有因调用await方法而被阻塞的线程将被唤醒。这就是实现一组线程相互等待的原理。

可以先看看CyclicBarrier内的一些成员变量

private static class Generation {
    boolean broken = false;
}

// 同步操作锁
private final ReentrantLock lock = new ReentrantLock();
// 线程拦截器
private final Condition trip = lock.newCondition();
// 每次拦截线程的数量
private final int parties;
// 换代前执行任务。在唤醒所有线程之前可以通过指定barrierCommand来执行自己的任务
private final Runnable barrierCommand;
//当前generation
private Generation generation = new Generation();
//计数器
private int count;

可以看到CyclicBarrier内部是通过条件队列trip来对线程进行阻塞的,并且其内部维护了两个int型的变量partiescountparties表示每次拦截的线程数,该值在构造时进行赋值。count是内部计数器,它的初始值和parties相同,以后随着每次await方法的调用而减1,直到减为0就将所有线程唤醒。

CyclicBarrier有两种构造函数

// 可指定需要拦截的线程数
public CyclicBarrier(int parties) {
    this(parties, null);
}
// 除了可以指定要拦截的线程数,还可以指定当所有线程都被唤醒后需要执行的任务。
public CyclicBarrier(int parties, Runnable barrierAction) {
    if (parties <= 0) throw new IllegalArgumentException();
    this.parties = parties;
    this.count = parties;
    this.barrierCommand = barrierAction;
}

CyclicBarrier类最主要的功能就是使先到达屏障点的线程阻塞并等待后面的线程,其中它提供了两种等待的方法,分别是定时等待和非定时等待。

// 定时等待
public int await(long timeout, TimeUnit unit)
    throws InterruptedException,
           BrokenBarrierException,
           TimeoutException {
    return dowait(true, unit.toNanos(timeout));
}
// 非定时等待
public int await() throws InterruptedException, BrokenBarrierException {
    try {
        return dowait(false, 0L);
    } catch (TimeoutException toe) {
        throw new Error(toe); // cannot happen
    }
}

可以看到无论是定时等待还是非定时等待,它们都调用了dowait方法,只不过是传入的参数不同。

// 核心等待方法
private int dowait(boolean timed, long nanos)
    throws InterruptedException, BrokenBarrierException,
           TimeoutException {
    final ReentrantLock lock = this.lock;
    lock.lock();
    try {
        final Generation g = generation;

        if (g.broken)
            throw new BrokenBarrierException();
        // 检查当前线程是否被中断
        if (Thread.interrupted()) {
            breakBarrier();
            throw new InterruptedException();
        }
        // 计数器值减1
        int index = --count;
        // 计数器的值减为0则需唤醒所有线程并转换到下一代
        if (index == 0) {  // tripped
            boolean ranAction = false;
            try {
                // 先执行指定的任务
                final Runnable command = barrierCommand;
                if (command != null)
                    command.run();
                ranAction = true;
                // 唤醒所有线程并转到下一代
                nextGeneration();
                return 0;
            } finally {
                if (!ranAction)
                    breakBarrier();
            }
        }
        
        // 如果计数器不为0则自旋
        for (;;) {
            try {
                // 根据传入的参数来决定是定时等待还是非定时等待
                if (!timed)
                    // 非定时任务
                    trip.await();
                else if (nanos > 0L)
                    // 定时任务
                    nanos = trip.awaitNanos(nanos);
            } catch (InterruptedException ie) {
                if (g == generation && ! g.broken) {
                    // 线程中断 打破栅栏 并唤醒其他线程
                    // 可以说明在等待过程中有一个线程被中断则全部结束,所有之前被阻塞的线程都会被唤醒。
                    breakBarrier();
                    throw ie;
                } else {
                    
                    Thread.currentThread().interrupt();
                }
            }

            if (g.broken)
                throw new BrokenBarrierException();
            // 是否是正常的换代操作而被唤醒,如果是则返回计数器的值
            if (g != generation)
                return index;
            // 超时报异常
            if (timed && nanos <= 0L) {
                breakBarrier();
                throw new TimeoutException();
            }
        }
    } finally {
        lock.unlock();
    }
}


//打翻当前栅栏
private void breakBarrier() {
    //将当前栅栏状态设置为打翻
    generation.broken = true;
    //设置计数器的值为需要拦截的线程数
    count = parties;
    //唤醒所有线程
    trip.signalAll();
}

CyclicBarrierAQSUML图。

CyclicBarrier和CountDownLatch

这俩有点像,有很不同。

CountDownLatch主要用来解决一个线程等待多个线程的场景。对于CountDownLatch来说,重点是那一个线程, 是它在等待,而另外那N个线程在把某个事情做完之后可以继续等待,也可以终止。

CyclicBarrier是一组线程之间互相等待,只要有一个线程没有完成,其他线程都要等待。对于CyclicBarrier来说,重点是那一组(N个)线程,他们之间任何一个没有完成,所有的线程都必须等待。

CyclicBarrier的计数器由自己控制,而CountDownLatch的计数器则由使用者来控制,在CyclicBarrier中线程调用await方法不仅会将自己阻塞还会将计数器减1,而在CountDownLatch中线程调用await方法只是将自己阻塞而不会减少计数器的值。

除此之外 CountDownLatch 的计数器是不能循环利用的,也就是说一旦计数器减到 0,再有线程调用 await(),该线程会直接通过。

CyclicBarrier 的计数器是可以循环利用的,而且具备自动重置的功能,一旦计数器减到 0 会自动重置到设置的初始值。除此之外,CyclicBarrier 还可以设置回调函数。

总结

各种同步器总结:

同步器
ReentrantLock 分为公平锁和非公平锁,是可重入独占锁;注意与synchronized关键字的对比;lock加锁unlock解锁,主要是理解内部的实现方式。
Semaphore 允许多个线程同时对共享资源进行操作,并且可以有效的控制并发数。关键方法是acquire获取许可、release释放许可。
CountDownLatch 一个线程在等待另外一些线程完成各自工作之后,再继续往下执行。调用await方法的线程会被挂起,知道count值为0再继续执行;countDown方法就是讲count的值减1。
CyclicBarrier 注意与CountDownLatch的区别,它是一组线程相互等待,当所有线程都到达某个屏障点后再进行后续的操作。调用await方法线程来告诉CyclicBarrier已经到达同步点。
posted @   knqiufan  阅读(189)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· DeepSeek 开源周回顾「GitHub 热点速览」
· 物流快递公司核心技术能力-地址解析分单基础技术分享
· .NET 10首个预览版发布:重大改进与新特性概览!
· AI与.NET技术实操系列(二):开始使用ML.NET
· 单线程的Redis速度为什么快?
点击右上角即可分享
微信分享提示
欢迎阅读『[Java][并发编程]AQS以及其相关同步器的源码解析』