时间:2023-02-18 10:16:00 | 栏目:JAVA代码 | 点击:次
是否会遇到这样的场景,你向线程池提交了多个任务,你希望这批任务全部完成后能够反向通知你。
你可能会使用线程计数的方式,等到计数器累加到提交的线程数量,然后通知。emmm,不是不可以,只是不够优雅。本文提供优雅的实现方式,Phaser同步屏障。
也可以不依赖,本人习惯把代码简单化,使用了hutool,所以依赖只有这个。
<dependency> <groupId>cn.hutool</groupId> <artifactId>hutool-all</artifactId> <version>5.7.15</version> </dependency>
废话不多说,上代码。
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; /** * @Program: csdn @ClassName: PhaserUtil @Author: huyi @Date: 2021-11-06 21:03 @Description: * 多线程监控回调工具 @Version: V1.0 */ 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); } } /** * 提交任务以及完成后需要执行的内容 * * @param tasks 任务 * @param complete 完成任务 */ public static void submit(List<Runnable> 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<Runnable> 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,又废话了。
分享一下:
没必要的事就不做,必要的事就尽快做。---冰果
如果本文对你有用,请不要吝啬你的赞,谢谢。