Java 信号量(semaphore)搭配CountDownLatch 实现多线程处理循环内逻辑并限制创建线程数

Java 信号量(semaphore)搭配CountDownLatch 实现多线程处理循环内逻辑并限制创建线程数

  • 使用场景
    for循环中的处理逻辑较慢,这个时候可以在for循环中创建线程实现多线程处理数据,结合CountDownLatch 可以等待所有线程全部执行结束之后主线程再执行。
    但是有的时候循环次数比较多,并不适合循环创建,因为线程会创建太多。
    比如下面这段代码
 ExecutorService threadPools = Executors.newFixedThreadPool(20);
        CountDownLatch countDownLatch = new CountDownLatch(100);
        for (int i = 0; i < 100; i++) {
            final int token = i;
                threadPools.execute(()->{
                    try {
                        System.out.println("线程" + Thread.currentThread().getName());
                        Thread.sleep(1000);
                    } catch (InterruptedException e) {
                        e.printStackTrace();
                    }finally {
                        countDownLatch.countDown();
                    }
                });
        }
        try {
            countDownLatch.await();
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
  • 这时循环100次会执行100个任务,如果循环次数特别大,会直接导致系统内存占满死机。

这个时候可以使用信号量来限制任务执行的次视

直接上代码

@Service
public class XxlService {

    private final Semaphore semaphore = new Semaphore(10);

    private final ExecutorService threadPools = Executors.newFixedThreadPool(20);

    public void syncSaiWordData() {

        // 2. 创建计数器
        CountDownLatch countDownLatch = new CountDownLatch(needSyncDataList.size());

        // 4. 开启多线程完善数据 最多10条线程
        for (WordSaiwordLibraryPO wordSaiwordLibraryPO : needSyncDataList) {
            threadPools.execute(() -> {
                try {
                // 获取许可
                    semaphore.acquire();
                    // 已经补全信息的无需再次查询
                    // 处理业务逻辑...
                    // ... 
                } catch (Exception e) {
                    e.printStackTrace();
                } finally {
                // 归还许可
                    semaphore.release();
                    //计数器 -1
                    countDownLatch.countDown();
                }
            });
        }
        try {
        // 等待全部执行结束
            countDownLatch.await();
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
}

  • 1
    点赞
  • 0
    收藏
  • 打赏
    打赏
  • 0
    评论
©️2022 CSDN 皮肤主题:大白 设计师:CSDN官方博客 返回首页
评论

打赏作者

今天的砖很烫

你的鼓励将是我创作的最大动力

¥2 ¥4 ¥6 ¥10 ¥20
输入1-500的整数
余额支付 (余额:-- )
扫码支付
扫码支付:¥2
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值