多线程等待任务结束的几种方法

比如,主线程创建线程池,提交n个任务,想让主线程在任务全部结束之后再继续做其他的事情。


1、使用awaitTermination方法

public static void main(String[] args) { ExecutorService executor = Executors.newFixedThreadPool(3); int i = 0; AtomicInteger result = new AtomicInteger(0); while (i < 10) { i++; executor.execute(() -> { //每执行一次,对result加1 System.out.println(result.addAndGet(1)); }); } System.out.println("调用shutdown()方法时,result的值为:" + result.get()); executor.shutdown(); try { //等待所有任务结束,最多等待30分钟 executor.awaitTermination(30, TimeUnit.MINUTES); } catch (InterruptedException e) { e.printStackTrace(); } System.out.println("计算完成,result的值为:" + result.get() + ",可以继续处理其他事情了"); } 

boolean awaitTermination(long timeout, TimeUnit unit)的作用:

Blocks until all tasks have completed execution after a shutdown request,
or the timeout occurs, or the current thread is interrupted,
whichever happens first.


2、借助工具类CountDownLatch

public static void main(String[] args) { ExecutorService executor = Executors.newFixedThreadPool(3); CountDownLatch latch = new CountDownLatch(10); AtomicInteger result = new AtomicInteger(0); for (int i = 1; i <= 10; i++) { executor.execute(() -> { //do something... System.out.println(result.addAndGet(1)); //then latch.countDown(); }); } System.out.println("调用shutdown()方法时,result的值为:" + result.get()); executor.shutdown(); try { latch.await();//等待任务结束:其实是等待latch值为0 } catch (InterruptedException e) { e.printStackTrace(); } System.out.println("计算完成,result的值为:" + result.get() + ",可以继续处理其他事情了"); } 

适用于知道任务的数量,因为CountDownLatch在创建时就要指定其大小,并且不能重新初始化。

//TODO:CountDownLatch是否会禁止指令重排序?(从官方的例子看,会!有待研究)

public void example() { CountDownLatch doneSignal = new CountDownLatch(10); Executor e = Executors.newFixedThreadPool(10); for (int i = 0; i < 10; ++i) // create and start threads e.execute(new WorkerRunnable(doneSignal, i)); try { doneSignal.await(); // wait for all to finis } catch (InterruptedException e1) { e1.printStackTrace(); } } class WorkerRunnable implements Runnable { private final CountDownLatch doneSignal; private final int i; WorkerRunnable(CountDownLatch doneSignal, int i) { this.doneSignal = doneSignal; this.i = i; } public void run() { doWork(i); doneSignal.countDown(); } void doWork(int i) { } } 

其他

//TODO



作者:maxwellyue
链接:https://www.jianshu.com/p/bcbfb58d0da5
来源:简书
简书著作权归作者所有,任何形式的转载都请联系作者获得授权并注明出处。

猜你喜欢

转载自www.cnblogs.com/xiaoshen666/p/11258556.html