欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

CountDownLatch实战指南

程序员文章站 2022-07-10 10:47:44
...
CountDownLatch是concuurent包里的一个工具类,它是一个计数器,它允许一个或者多个线程一直在等待,直到所有线程全部执行完成之后再执行。
CountDownLatch是什么
untDownLatch是在java1.5被引入的,跟它一起被引入的并发工具类还有CyclicBarrier、Semaphore、ConcurrentHashMap和BlockingQueue,它们都存在于java.util.concurrent包下。CountDownLatch这个类能够使一个线程等待其他线程完成各自的工作后再执行。例如,应用程序的主线程希望在负责启动框架服务的线程已经启动所有的框架服务之后再执行。

CountDownLatch是通过一个计数器来实现的,计数器的初始值为线程的数量。每当一个线程完成了自己的任务后,计数器的值就会减1。当计数器值到达0时,它表示所有的线程已经完成了任务,然后在闭锁上等待的线程就可以恢复执行任务。



CountDownLatch的伪代码如下所示:

//Main thread start
//Create CountDownLatch for N threads
//Create and start N threads
//Main thread wait on latch
//N threads completes there tasks are returns
//Main thread resume execution
CountDownLatch如何工作

CountDownLatch.java类中定义的构造函数:

1
2
//Constructs a CountDownLatch initialized with the given count.
public void CountDownLatch(int count) {...}
构造器中的计数值(count)实际上就是闭锁需要等待的线程数量。这个值只能被设置一次,而且CountDownLatch没有提供任何机制去重新设置这个计数值。

与CountDownLatch的第一次交互是主线程等待其他线程。主线程必须在启动其他线程后立即调用CountDownLatch.await()方法。这样主线程的操作就会在这个方法上阻塞,直到其他线程完成各自的任务。

其他N 个线程必须引用闭锁对象,因为他们需要通知CountDownLatch对象,他们已经完成了各自的任务。这种通知机制是通过 CountDownLatch.countDown()方法来完成的;每调用一次这个方法,在构造函数中初始化的count值就减1。所以当N个线程都调 用了这个方法,count的值等于0,然后主线程就能通过await()方法,恢复执行自己的任务。

在实时系统中的使用场景

让我们尝试罗列出在java实时系统中CountDownLatch都有哪些使用场景。我所罗列的都是我所能想到的。如果你有别的可能的使用方法,请在留言里列出来,这样会帮助到大家。

实现最大的并行性:有时我们想同时启动多个线程,实现最大程度的并行性。例如,我们想测试一个单例类。如果我们创建一个初始计数为1的CountDownLatch,并让所有线程都在这个锁上等待,那么我们可以很轻松地完成测试。我们只需调用 一次countDown()方法就可以让所有的等待线程同时恢复执行。
开始执行前等待n个线程完成各自任务:例如应用程序启动类要确保在处理用户请求前,所有N个外部系统已经启动和运行了。
死锁检测:一个非常方便的使用场景是,你可以使用n个线程访问共享资源,在每次测试阶段的线程数目是不同的,并尝试产生死锁。

在这个例子中,我模拟了一个应用程序启动类,它开始时启动了n个线程类,这些线程将检查外部系统并通知闭锁,并且启动类一直在闭锁上等待着。一旦验证和检查了所有外部服务,那么启动类恢复执行。

CountDownLatch使用例子
AbstractBaseRuleHandler.java:这个类是一个Runnable,负责所有特定的外部服务健康的检测。它删除了重复的代码和闭锁的中心控制代码
public abstract class AbstractBaseRuleHandler implements Runnable {

	private CountDownLatch latch;
	
	private List<Long> errorList;
	
	private Long primaryId;
	
	public AbstractBaseRuleHandler(CountDownLatch latch,List<Long> errorList,Long primaryId) {
		this.latch =latch;
		this.errorList = errorList;
		this.primaryId =primaryId;
	}

	@Override
	public void run() {
		try {
			  int i = handleOpration(primaryId);
			  if(i == 0){
				  errorList.add(primaryId);
			  }
		}catch(Exception e){
			ThreadLocalLog.infoPro(AbstractBaseRuleHandler.class.getName(), "run",primaryId + ":删除模版关联规则出错"+  e.getMessage(), null, null);
		}finally{
			latch.countDown();
		}
	}

	 public abstract int handleOpration(Long id);
}


TemplateRuleDelBean.java 这个类继承了AbstractBaseRuleHandler,实现了handleOpration()方法。
public class TemplateRuleDelBean extends AbstractBaseRuleHandler{

	
	/**
	 * @param latch
	 * @param errorList
	 * @param primaryId
	 */
	public TemplateRuleDelBean(CountDownLatch latch, List<Long> errorList, Long primaryId) {
		super(latch, errorList, primaryId);
	}
	
	@Override
	public int handleOpration(Long id) {
	return 0;//ruleBaseConfigEntityMapper.updateByPrimaryKeySelective(entity);
     //本来做跟新操作,现在直接返回0
	}

}

TempateDelSingleton.java 这个类是一个主启动类,它负责初始化闭锁,然后等待,直到所有服务都被检测完。
public class TempateDelSingleton {

	private TempateDelSingleton() {
	}
	
	private static class TemplateInner{
		private static final TempateDelSingleton incetance = new TempateDelSingleton();
	}
	
	public static TempateDelSingleton getIncetance(){
		return TemplateInner.incetance;
	}
	
	public List<Long> handleMain(List<Long> primaryList){
		CountDownLatch _latch= new CountDownLatch(primaryList.size());
		List<Long> errorList = new ArrayList<>();
		List<AbstractBaseRuleHandler> baseList = new ArrayList<>();
		for(Long ab : primaryList){
			baseList.add(new TemplateRuleDelBean(_latch, errorList, ab));
		}
		ExecutorService executor = Executors.newFixedThreadPool(baseList.size());
		for(final AbstractBaseRuleHandler v :  baseList){
			executor.execute(v);
		}
		 try {
			_latch.await();
		} catch (InterruptedException e) {
		}finally{			
			executor.shutdown();
		}
		return errorList;
	}
	

}

最后用main 跑一下这个单例 即可去检测一下闭锁的功能了。
	public static void main(String[] args) {
		List<Long> primaryList = Arrays.asList(1l,2l,3l);
		TempateDelSingleton.getIncetance().handleMain(primaryList);
	}

输出结果:
1
3
2
一些其他的CountDownLatch相关的问题:

解释一下CountDownLatch概念?
CountDownLatch 和CyclicBarrier的不同之处?
给出一些CountDownLatch使用的例子?
CountDownLatch 类中主要的方法?
相关标签: thread java