软件编程
位置:首页>> 软件编程>> java编程>> Java 多线程等待优雅的实现方式之Phaser同步屏障

Java 多线程等待优雅的实现方式之Phaser同步屏障

作者:剑客阿良_ALiang  发布时间:2023-11-29 09:47:43 

标签:Java,多线程等待,Phaser,同步屏障

前言

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

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

Maven依赖

也可以不依赖,本人习惯把代码简单化,使用了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也和第一点同理。

验证一下

Java 多线程等待优雅的实现方式之Phaser同步屏障

OK,没什么问题。

来源:https://huyi-aliang.blog.csdn.net/article/details/121185988

0
投稿

猜你喜欢

手机版 软件编程 asp之家 www.aspxhome.com