CountDownLatch使用synchronized、notifyAll、wait实现

最新推荐文章于 2024-04-22 23:18:47 发布

置顶 hello_java_world

最新推荐文章于 2024-04-22 23:18:47 发布

阅读量269

点赞数

分类专栏： Java 文章标签： java 并发编程

本文链接：https://blog.csdn.net/hello_java_world/article/details/104880907

版权

Java 专栏收录该内容

2 篇文章 0 订阅

订阅专栏

本文分享了一种使用synchronized、notifyAll、wait等方法手工实现CountDownLatch功能的代码示例，通过创建自定义的CountDownLatchDiy类，演示了如何在多线程环境下控制线程的等待和唤醒，确保所有任务完成后再继续执行后续操作。

摘要由CSDN通过智能技术生成

为了加上对CountDownLatch的理解，自己DIY使用synchronized、notifyAll、wait实现CountDownLatch的功能，下面直接贴代码了。如有问题，欢迎大家指出

public class CountDownLatchDiyTest {
    
    public static void main(String[] args) throws Exception {
        int n = 1000;
        final CountDownLatchDiy startSignal = new CountDownLatchDiy(1);
        final CountDownLatchDiy doneSignal = new CountDownLatchDiy(n);
        ExecutorService fixedThreadPool = Executors.newFixedThreadPool(100);
        for (int i = 0; i < n; i++) {
            fixedThreadPool.execute(
                new Thread("任务" + String.valueOf(i)) {
                    public void run() {
                        try {
                            startSignal.await();
                            System.out.println(this.getName() + " thread start");
                            Thread.sleep(random(100));
                            System.out.println(this.getName() + " thread over");
                            doneSignal.countDown();
                        } catch (Exception e) {
                            e.printStackTrace();
                        }
                    }
                });
        }
        System.out.println("main thread start");
        startSignal.countDown();
        Thread.sleep(random(100));
        doneSignal.await();
        System.out.println("main thread over");
        fixedThreadPool.shutdown();
    }

    static class CountDownLatchDiy {
        private int count;

        CountDownLatchDiy(int c) {
            this.count = c;
        }

        public synchronized void countDown() {
            if (count > 0) {
                count = count - 1;
                notifyAll();
                return;
            }

            throw new RuntimeException(Thread.currentThread().getName() + " count <= 0 .");
        }

        public synchronized void await() throws InterruptedException {
            while (count > 0) {
                wait();
            }
        }
    }

    static long random(long max) {
        return 1 + new Double(Math.random() * max).longValue();
    }
}