目录

前言

Maven依赖

代码

总结

前言

是否会遇到这样的场景，你向线程池提交了多个任务，你希望这批任务全部完成后能够反向通知你。

你可能会使用线程计数的方式，等到计数器累加到提交的线程数量，然后通知。emmm，不是不可以，只是不够优雅。本文提供优雅的实现方式，Phaser同步屏障。

Maven依赖

也可以不依赖，本人习惯把代码简单化，使用了hutool，所以依赖只有这个。

        
            cn.hutool
            hutool-all
            5.7.15
        

 

代码

废话不多说，上代码。

package com.huyi.csdn.tools;

import cn.hutool.core.thread.ThreadUtil;

import java.util.ArrayList;
import java.util.List;
import java.util.Random;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.Phaser;
import java.util.concurrent.TimeUnit;


public class PhaserUtil {
  public static final ExecutorService executorService = Executors.newFixedThreadPool(50);

  public static class CustomPharser extends Phaser {
    private final Runnable runnable;

    public CustomPharser(Runnable runnable) {
      this.runnable = runnable;
    }

    @Override
    protected boolean onAdvance(int phase, int registeredParties) {
      this.runnable.run();
      return super.onAdvance(phase, registeredParties);
    }
  }

  
  public static void submit(List tasks, Runnable complete) {
    Phaser phaser = new CustomPharser(complete);
    for (Runnable runnable : tasks) {
      executorService.submit(
          () -> {
            phaser.register();
            runnable.run();
            System.out.println(Thread.currentThread().getName() + "完成任务！");
            phaser.arriveAndAwaitAdvance();
          });
    }
  }

  
  public static void destroy() {
    System.out.println("摧毁线程池");
    executorService.shutdown();
  }

  public static void main(String[] args) {
    List tasks = new ArrayList<>();
    Random random = new Random();
    for (int i = 0; i < 10; i++) {
      tasks.add(
          () -> {
            ThreadUtil.sleep(random.nextInt(10), TimeUnit.SECONDS);
          });
    }
    submit(tasks, () -> System.out.println("所有任务已完成"));
    ThreadUtil.sleep(20, TimeUnit.SECONDS);
    destroy();
  }
}

代码说明

1、提交任务执行的方式是Runnable也好，Callable也好，或者Consumer、Function等等，不影响，你可以看着调整。

2、完成后的Runnable也和第一点同理。

验证一下

OK，没什么问题。

总结

其实我一直想分享一些可以让读者工作中能用到的东西，想到牧神记里面的一句话，圣人之道，无非就是百姓日用。emmmm，又废话了。

分享一下：

                没必要的事就不做，必要的事就尽快做。---冰果

如果本文对你有用，请不要吝啬你的赞，谢谢。