多任务并发:如何判断线程池中的任务都已经执行完毕?

前言:

多线程并发,我们往往采用线程池来管理并发的线程。但是,我们往往有这样的需要:要求在线程池中的任务都完成后才能执行后续的任务,或者需要任务都完成后释放资源或向数据库写入状态。这些都需要我们判断线程池的任务是否都已经完成。
判断线程池中的任务是否全部完成,方式有不少,这里我来整理一下。

一、使用线程池的原生函数isTerminated();

优点:操作简便;

缺点:需要主线程阻塞;

executor提供一个原生函数isTerminated()来判断线程池中的任务是否全部完成。全部完成返回true,否则返回false。

栗子:
package my.thread.test;


import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;

class AnyTask implements Runnable{
	
	public void run() {
		try {
			Thread.sleep(3000);
			System.out.println("普通任务执行执行完毕!");
				
		}catch(Exception e) {
			
			e.printStackTrace();
			
		}
	}
}

class EndTask {
	public void taskRun() {
		try {
			System.out.println("开始执行最终任务");
			
		}catch(Exception e) {
			e.printStackTrace();
			
		}
	}
}


public class TeminatedTest {
	
	public static void main(String[] args) {
		try {
			ExecutorService exector = Executors.newFixedThreadPool(7);
		
			for(int i =0;i<10;i++) {
				AnyTask anyTask = new AnyTask();
				exector.execute(anyTask);
			}
			exector.shutdown();
		
			while(true) {
				if(exector.isTerminated()) {
					
					EndTask endTask = new EndTask();
					endTask.taskRun();
					
					break;
					
				}else {
					Thread.sleep(3000);
				}
			}
			
		}catch(Exception e) {
			e.printStackTrace();
		}
		
	}
}

操作很方便,但要求主线程阻塞,这样对接口化编程很不友好。

二、使用CountDownLatch:

优点:操作相对简便,可以把等待线程池中任务完成后的后续工作做成任务,同样放到线程池中运行,简单来说,就是可以控制线程池中任务执行的顺序。

缺点:
需要提前知道任务的数量。

原理:其工作原理是赋给CountDownLatch一个计数值,普通的任务执行完毕后,调用countDown()执行计数值减一。最后执行的任务在调用方法的开始调用await()方法,这样整个任务会阻塞,直到这个计数值(Count)为零,才会继续执行。

栗子:
package my.thread.test;


import java.util.concurrent.CountDownLatch;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;


	
	class AnyTask implements Runnable{
		
		private CountDownLatch endTaskLatch;
		
		public AnyTask(CountDownLatch latch) {
			
			this.endTaskLatch = latch;
			
		}
		
		public void run() {
			try {
				Thread.sleep(10000);
				System.out.println("普通任务执行执行完毕!");
				endTaskLatch.countDown();
				System.out.println(endTaskLatch.getCount());
				
			}catch(Exception e) {
				
				e.printStackTrace();
				
			}
		}
	}
	
class EndTask implements Runnable{
		
		private CountDownLatch endTaskLatch;
		
		EndTask(CountDownLatch latch) {
			
			this.endTaskLatch =latch;
		}
		
		public void run() {
			
			try {
				
				endTaskLatch.await();
				System.out.println("开始执行最终任务");
				System.out.println(endTaskLatch.getCount());
			}catch(Exception e) {
				e.printStackTrace();
				
			}
		}
	}
	
public class CountDownLatchTest {

	public static void main(String[] args) {
		
		ExecutorService exector = Executors.newFixedThreadPool(7);
		CountDownLatch TaskLatch = new CountDownLatch(10);
		EndTask endTask = new EndTask(TaskLatch);
		
		exector.execute(endTask);
		
		for(int i=0;i<10;i++) {
			exector.execute(new AnyTask(TaskLatch));
		}
		System.out.println("主线程已经执行完了");	
	}
}

根据打印结果我们可以清楚的看到你,endTask等待所有的AnyTask执行完毕才继续执行,否则会一直阻塞下去。

当然,我们也可以使用await()一直阻塞。在主线程调用即可。

三、使用重入锁,维持一个公共计数:

第三种方式,实行起来基本上和CountDownLatch 的原理差不多。

所有的普通任务维持一个static的计数器,当任务完成时计数器加一(这里要用锁),当计数器的值等于任务数时,这时所有的任务已经执行完毕了,这时endTask就会自动执行。

所有的任务,包括endTask都可以放到线程池中,当普通任务未执行完毕时,如果endTask开始执行,也会一直阻塞等待,直到公共计数等于等于任务数。

优点:灵活、可控
缺点:需要预知运算的任务数、操作相对有点繁琐

栗子:
package my.thread.test;

import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.locks.Lock;
import java.util.concurrent.locks.ReentrantLock;

class AnyTask implements Runnable{
	
	public static int taskNum = 0;
	private Lock lock = new ReentrantLock();
	
	public void run() {
		try {
			Thread.sleep(3000);
			
			System.out.println("普通任务执行执行完毕!");
			
			lock.lock();
			AnyTask.taskNum++;
			lock.unlock();	
		}catch(Exception e) {
			
			e.printStackTrace();
			
		}
	}
}

class EndTask implements Runnable{
	
	private int tnum = 0;
	public EndTask(int tnum) {
		this.tnum = tnum;
		
	}
	
	public void run() {
		
		try {
			while(true) {
				if(tnum==AnyTask.taskNum) {
			
					System.out.println("开始执行最终任务");
					break;
					
				}else {
					System.out.println("线程池的任务没有完成,等待。。。");
					Thread.sleep(3000);
				}
			}
			
		}catch(Exception e) {
			e.printStackTrace();
			
		}
	}
}

public class JishuTest {
	
	public static void main(String[] args) {
		
		ExecutorService exector = Executors.newFixedThreadPool(7);
		
		int taskNum = 10;
		
		for(int i=0;i<taskNum;i++) {
			
			AnyTask anyTask = new AnyTask(); 
			exector.execute(anyTask);
			
		}
		EndTask endTask = new EndTask(taskNum);
		exector.execute(endTask);
		exector.shutdown();
		
	}
}

四、submit向线程池提交任务,Future判断任务执行状态:

使用submit向线程池提交任务与execute提交不同,submit会有Future类型的返回值。通过返回返回值可以用于判断线程任务的执行状态。
举例:

package com.thread;

import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.Future;

	
public class Test {

	public static void main(String[] args) {
		
		ExecutorService exector = Executors.newFixedThreadPool(7);
		Future<?> future = exector.submit(new Runnable(){
			@Override
			public void run() {
				int i = 5;
				while(i>=0) {
					try {
						System.out.println("正在执行线程!i="+i);
						Thread.sleep(10000);
						i--;
					}catch(Exception e) {
						e.printStackTrace();
					}
				}
			}	
		});
		while(true) {
			if(future.isDone()) {
				break;
			}
			System.out.println("主线程等待...");
			try {
				Thread.sleep(5000);
			}catch(Exception e) {
				e.printStackTrace();
			}
		}
		System.out.println("主线程已经执行完了");	
	}
}

打印日志:

主线程等待...
正在执行线程!i=5
主线程等待...
主线程等待...
正在执行线程!i=4
主线程等待...
正在执行线程!i=3
主线程等待...
主线程等待...
正在执行线程!i=2
主线程等待...
主线程等待...
正在执行线程!i=1
主线程等待...
主线程等待...
正在执行线程!i=0
主线程等待...
主线程等待...
主线程已经执行完了

不同的应用场景应采用适合的方式,以上、谢谢 !

评论 7
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值