同步工具-CountDownLatch

最新推荐文章于 2025-04-15 08:45:12 发布

原创最新推荐文章于 2025-04-15 08:45:12 发布 · 142 阅读

0 ·

CC 4.0 BY-SA版权

并发专栏收录该内容

6 篇文章

订阅专栏

本文介绍如何使用CountDownLatch将异步接口转换为同步接口，详细解析其基于AQS的实现原理，包括状态管理、线程唤醒机制。

    我在本币系统用了一个CountDownLatch实现了一个把一个异步接口变成同步接口的用法。http接口调用的方法，主要是在servlet中的service方法中返回对应的response，然而该service方法中调用的一个异步接口的方法，如果不做限制的话直接返回为空。这里其实可以用future接口，但当时我只会CDL所以这篇文章就看看cdl的源码把。
    CountDownLatch的实现很简单，也和同步工具一样，基于aqs实现的

	private final Sync sync;

    private static final class Sync extends AbstractQueuedSynchronizer {
        private static final long serialVersionUID = 4982264981922014374L;
		//初始化状态
        Sync(int count) {
            setState(count);
        }
	
        int getCount() {
            return getState();
        }
		//获取状态是否是0，如果是0则countdown的次数已经完毕，所有等待条件的线程将会释放，不是0则还需要等待
        protected int tryAcquireShared(int acquires) {
            return (getState() == 0) ? 1 : -1;
        }
		//这里会有多个线程调用，所以自旋设置state-=1，如果state变为0则释放所有等待线程。
        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;
            }
        }
    }

    public void await() throws InterruptedException {
        sync.acquireSharedInterruptibly(1);
    }

    public void countDown() {
        sync.releaseShared(1);
    }

可见由于aqs的强大，这里的实现反而非常简单。state表示的是“次数”，每次countdown就是state-=1
当state=0时则之前的await线程将会被唤醒，由于是共享模式，会唤醒同步队列中的所有等待线程，之后调用await的线程直接返回。aqs提供同步工具中需要的判断是否阻塞，判断是否释放的功能，并且提供state来表示一个子类需要的状态。reentrantLock中的state表示线程是否获取锁，cdl的state表示是否阻塞条件满足。