CountDownLatch总结

CountDownLatch

  • 实现原理:
  • 主要方法

CountDownLatch相当于一个计数锁存器,当一定的操作结束后,某个线程才能执行。每执行一个操作计数减1,计数到0时,才执行某个操作。

实现原理:

CountDownLatch实现了一个继承AQS的共享锁Sync,里面重写的tryAcquireShared()方法,只有state为0的时候锁才可以被获取。在构造对象的时候,初始化一个计数值N,把N赋给sync的state变量,每执行一个操作,N就会减1,N减到0时,此时state也为0,这时锁可以被获取,可以执行特定的操作。
执行countDown()方法的线程让N减1,执行await()方法的线程等待,直到其他线程让N减到0。

主要方法

//只有一个构造方法
public CountDownLatch(int count) {
    if (count < 0) throw new IllegalArgumentException("count < 0");
    this.sync = new Sync(count);
}
//执行这个方法的线程等待,直到计数值为0
public void await() throws InterruptedException {
    sync.acquireSharedInterruptibly(1);
}
//执行这个方法的线程让计数值减1
public void countDown() {
    sync.releaseShared(1);
}
//如果state值为0,返回1,正值代表获取锁成功,负值代表获取锁失败
protected int tryAcquireShared(int acquires) {
        return (getState() == 0) ? 1 : -1;
    }
protected boolean tryReleaseShared(int releases) {
        // Decrement count; signal when transition to zero
        for (;;) {
            int c = getState();
            if (c == 0)
                return false;
            int nextc = c-1;
            if (compareAndSetState(c, nextc))
                return nextc == 0;
        }
    }
}

你可能感兴趣的:(java)