CountDownLatch的学习

发表时间:2017-09-07 16:42:30 浏览量( 16 ) 留言数( 0 )

学习目标:

1、了解CountDownLatch的功能。

2、在实践中灵活使用CountDownLatch。


学习过程:

一、CountDownLatch的简介

   CountDownLatch是在java1.5被引入的 CountDownLatch是通过一个计数器来实现的,计数器的初始值为线程的数量。每当一个线程完成了自己的任务后,计数器的值就会减1。当计数器值到达0时,它表示所有的线程已经完成了任务,然后在闭锁上等待的线程就可以恢复执行任务。例如,应用程序的主线程希望在负责启动框架服务的线程已经启动所有的框架服务之后再执行。

   构造器中的计数值(count)实际上就是闭锁需要等待的线程数量。这个值只能被设置一次,而且CountDownLatch没有提供任何机制去重新设置这个计数值。与CountDownLatch的第一次交互是主线程等待其他线程。主线程必须在启动其他线程后立即调用CountDownLatch.await()方法。这样主线程的操作就会在这个方法上阻塞,直到其他线程完成各自的任务。

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

   可见CountDownLatch的方法非常简单,构造方法设置数值,await方法会等待count等于0,countDown使得count减一。

二、示例代码

    有一个任务如下,主线程需要启动4个子线程完成一项任务,必须要要等3个任务都发出完成通知后主线程收到了这个通知后就可以继续操作了,使用CountDownLatch可以非常简单直观的完整这个操作,不想Join方法需要等子线程运行结束。

  定义一个Ruannable类:

public class MyTask implements Runnable {

	private int taskId;
	private String taskName;
	private CountDownLatch countDownLatch;

	public MyTask(int taskId, String taskName,CountDownLatch countDownLatch) {
		this.taskId = taskId;
		this.taskName = taskName;
		this.countDownLatch=countDownLatch;
	}

	@Override
	public void run() {
		try {
			System.out.println("run taskId =" + this.taskId);
			Thread.sleep(2000);
			System.out.println("run taskId =" + this.taskId+"执行了通知");
			countDownLatch.countDown();
			Thread.sleep(2000);
			System.out.println("end taskId =" + this.taskId);
		} catch (InterruptedException e) {
			e.printStackTrace();
		}
	}

	public String toString() {
		return Integer.toString(this.taskId);
	}

}

  定义运行测试类:

public class TestCountDownLatch {

	public static void main(String[] args) throws InterruptedException {

		CountDownLatch countDownLatch = new CountDownLatch(3);// 定义为3

		ExecutorService pool = Executors.newFixedThreadPool(3);
		// 创建线程
		MyTask mt1 = new MyTask(1, "任务1",countDownLatch);
		MyTask mt2 = new MyTask(2, "任务2",countDownLatch);
		MyTask mt3 = new MyTask(3, "任务3",countDownLatch);
		// 将线程放入池中进行执行
		pool.execute(mt1);
		pool.execute(mt2);
		pool.execute(mt3);

		System.out.println("主线程启动了。。。。");
		
		countDownLatch.await();// 主线程会阻塞在这里,等待倒数为0后才能继续执行,必须要等子线程结束就可以执行了。

		System.out.println("主线程可以继续执行了。。。");

	}

}

运行结果如下:

主线程启动了。。。。

run taskId =1

run taskId =2

run taskId =3

run taskId =1执行了通知

run taskId =3执行了通知

run taskId =2执行了通知

主线程可以继续执行了。。。

end taskId =1

end taskId =3

end taskId =2