java并发编程的重要工具类CountDownLatch和CyclicBarrier

本文详细介绍了CountDownLatch的两种使用场景,包括让多个线程等待的并发模拟和等待单个线程完成的任务汇总。通过实例展示了CountDownLatch如何作为并发控制工具,与Thread.join和CyclicBarrier进行对比,强调了各自的特点和适用情况。同时,还给出了CyclicBarrier的使用示例,展示了其在一组线程互相等待并共同执行的场景中的应用。
摘要由CSDN通过智能技术生成

CountDownLatch的两种使用场景:

场景1:让多个线程等待
场景2:和让单个线程等待。
场景1 让多个线程等待:模拟并发,让并发线程一起执行
为了模拟高并发,让一组线程在指定时刻(秒杀时间)执行抢购,这些线程在准备就绪后,进行等待(CountDownLatch.await()),直到秒杀时刻的到来,然后一拥而上;
这也是本地测试接口并发的一个简易实现。

在这个场景中,CountDownLatch充当的是一个发令枪的角色;
就像田径赛跑时,运动员会在起跑线做准备动作,等到发令枪一声响,运动员就会奋力奔跑。和上面的秒杀场景类似,代码实现如下:

CountDownLatch countDownLatch = new CountDownLatch(1);
for (int i = 0; i < 5; i++) {
    new Thread(() -> {
        try {
            //准备完毕……运动员都阻塞在这,等待号令
            countDownLatch.await();
            String parter = "【" + Thread.currentThread().getName() + "】";
            System.out.println(parter + "开始执行……");
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
    }).start();
}

Thread.sleep(2000);// 裁判准备发令
countDownLatch.countDown();// 发令枪:执行发令

运行结果:

【Thread-0】开始执行……
【Thread-1】开始执行……
【Thread-4】开始执行……
【Thread-3】开始执行……
【Thread-2】开始执行……
我们通过CountDownLatch.await(),让多个参与者线程启动后阻塞等待,然后在主线程 调用CountDownLatch.countdown(1) 将计数减为0,让所有线程一起往下执行;
以此实现了多个线程在同一时刻并发执行,来模拟并发请求的目的。

场景2 让单个线程等待:多个线程(任务)完成后,进行汇总合并
很多时候,我们的并发任务,存在前后依赖关系;比如数据详情页需要同时调用多个接口获取数据,并发请求获取到数据后、需要进行结果合并;或者多个数据操作完成后,需要数据check;
这其实都是:在多个线程(任务)完成后,进行汇总合并的场景。

代码实现如下:

CountDownLatch countDownLatch = new CountDownLatch(5);
for (int i = 0; i < 5; i++) {
    final int index = i;
    new Thread(() -> {
        try {
            Thread.sleep(1000 + ThreadLocalRandom.current().nextInt(1000));
            System.out.println("finish" + index + Thread.currentThread().getName());
            countDownLatch.countDown();
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
    }).start();
}

countDownLatch.await();// 主线程在阻塞,当计数器==0,就唤醒主线程往下执行。
System.out.println("主线程:在所有任务运行完成后,进行结果汇总");

运行结果:

finish4Thread-4
finish1Thread-1
finish2Thread-2
finish3Thread-3
finish0Thread-0
主线程:在所有任务运行完成后,进行结果汇总
在每个线程(任务) 完成的最后一行加上CountDownLatch.countDown(),让计数器-1;
当所有线程完成-1,计数器减到0后,主线程往下执行汇总任务。

从上面两个例子的代码,可以看出 CountDownLatch 的API并不多;

CountDownLatch的构造函数中的count就是闭锁需要等待的线程数量。这个值只能被设置一次,而且不能重新设置;
await():调用该方法的线程会被阻塞,直到构造方法传入的 N 减到 0 的时候,才能继续往下执行;
countDown():使 CountDownLatch 计数值 减 1;
CountDownLatch 工作原理
CountDownLatch是通过一个计数器来实现的,计数器的初始值为线程的数量;
调用await()方法的线程会被阻塞,直到计数器 减到 0 的时候,才能继续往下执行;

调用了await()进行阻塞等待的线程,它们阻塞在Latch门闩/栅栏上;只有当条件满足的时候(countDown() N次,将计数减为0),它们才能同时通过这个栅栏;以此能够实现,让所有的线程站在一个起跑线上。

countDown()方法则是将计数器减1;

在CountDownLatch的构造函数中,指定的线程数量,只能指定一次;由于CountDownLatch采用的是减计数,因此当计数减为0时,计数器不能被重置。
这是和CyclicBarrier的一个重要区别。

CountDownLatch 的源码在JUC并发工具中,也相对算是简单的;
底层基于 AbstractQueuedSynchronizer 实现,CountDownLatch 构造函数中指定的count直接赋给AQS的state;每次countDown()则都是release(1)减1,最后减到0时unpark阻塞线程;这一步是由最后一个执行countdown方法的线程执行的。
而调用await()方法时,当前线程就会判断state属性是否为0,如果为0,则继续往下执行,如果不为0,则使当前线程进入等待状态,直到某个线程将state属性置为0,其就会唤醒在await()方法中等待的线程。

CountDownLatch与Thread.join
CountDownLatch的作用就是允许一个或多个线程等待其他线程完成操作,看起来有点类似join() 方法,但其提供了比 join() 更加灵活的API。
CountDownLatch可以手动控制在n个线程里调用n次countDown()方法使计数器进行减一操作,也可以在一个线程里调用n次执行减一操作。
而 join() 的实现原理是不停检查join线程是否存活,如果 join 线程存活则让当前线程永远等待。所以两者之间相对来说还是CountDownLatch使用起来较为灵活。

CountDownLatch与CyclicBarrier
CountDownLatch和CyclicBarrier都能够实现线程之间的等待,只不过它们侧重点不同:

CountDownLatch一般用于一个或多个线程,等待其他线程执行完任务后,再才执行
CyclicBarrier一般用于一组线程互相等待至某个状态,然后这一组线程再同时执行
另外,CountDownLatch是减计数,计数减为0后不能重用;而CyclicBarrier是加计数,可置0后复用。

CountDownLatch和CyclicBarrier两个同为java并发编程的重要工具类
前者更多依赖经典的AQS机制和CAS机制来控制器内部状态的更迭和计数器本身的变化,而后者更多依靠可重入Lock等机制来控制其内部并发安全性和一致性。

CyclicBarrier例子

package com.ld.task;

import java.util.Random;
import java.util.concurrent.BrokenBarrierException;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.CyclicBarrier;

/**
 * CyclicBarrier:放行条件由一组线程本身决定
 */
public class UseCyclicBarrier {
    //     static CyclicBarrier cyclicBarrier = new CyclicBarrier(6);
    //6个线程都执行到await()方法时,会先执行AfterWork()线程,AfterWork()线程执行完成后在继续向下执行。
    static CyclicBarrier cyclicBarrier = new CyclicBarrier(6, new AfterWork());
    //    List<String> threadNames = new ArrayList<>();
    static ConcurrentHashMap<String, Object> concurrentHashMap = new ConcurrentHashMap<>(); //线程安全的集合

    static class WorkTask implements Runnable {
        @Override
        public void run() {
            try {
                concurrentHashMap.put(Thread.currentThread().getName(), Thread.currentThread().getId());
                Random r = new Random();
                if (r.nextBoolean()) {
                    System.out.println("线程" + Thread.currentThread().getName() + "做点别的事情...");
                    Thread.sleep(2000);
                }
                System.out.println("线程" + Thread.currentThread().getName() + "进入等待状态...");
                cyclicBarrier.await();//线程运行到此处等待的数量达到设定的初始值后,就会自动向下进行。
                Thread.sleep(1000);
                System.out.println("线程" + Thread.currentThread().getName() + "开始向下执行...");
            } catch (InterruptedException e) {
                e.printStackTrace();
            } catch (BrokenBarrierException e) {
                e.printStackTrace();
            }
        }
    }

    static class AfterWork implements Runnable {
        @Override
        public void run() {
            if (!concurrentHashMap.isEmpty()) {
                for (String name : concurrentHashMap.keySet()) {
                    System.out.println("AfterWork:" + name);
                }
            }
        }
    }

    public static void main(String[] args) throws InterruptedException {
        for (int i = 0; i < 6; i++) {
//            Thread.sleep(1000);
            new Thread(new WorkTask()).start();
        }
    }
}

CyclicBarrier和CountDownLatch比较
1、CountDownLatch放行由第三者控制,CyclicBarrier放行由一组线程本身控制
2、CountDownLatch放行条件》=线程数,CyclicBarrier放行条件=线程数

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值