JUC学习笔记 -- (4) CountDownLatch闭锁

亦凉 2022-05-04 00:30 219阅读 0赞

CountDownLatch类位于java.util.concurrent包下，利用它可以实现类似计数器的功能。比如有一个任务A，它要等待其他4个任务执行完毕之后才能执行，此时就可以利用CountDownLatch来实现这种功能了。

CountDownLatch类只提供了一个构造器：

public CountDownLatch(int count) {  };  //参数count为计数值

然后下面这3个方法是CountDownLatch类中最重要的方法：

//调用await()方法的线程会被挂起，它会等待直到count值为0才继续执行
    public void await() throws InterruptedException { };   
    
    //和await()类似，只不过等待一定的时间后count值还没变为0的话就会继续执行
    public boolean await(long timeout, TimeUnit unit) throws InterruptedException { }; 
    
    //将count值减1
    public void countDown() { };

比如计算所有子线程工作时间的案例:

public class TestCountDownLatch {
    
    	public static void main(String[] args) {
    		//创建计数器
    		CountDownLatch latch = new CountDownLatch(5);
    		LatchDemo ld = new LatchDemo(latch);
    		long startTm = System.currentTimeMillis();
    		for(int i=0;i<5;i++){
    			new Thread(ld).start();
    		}
    		
    		try {
    			//主线程挂起，知道所有的子线程执行结束再继续
    			latch.await();
    		} catch (InterruptedException e) {
    			e.printStackTrace();
    		}
    		long endTm = System.currentTimeMillis();
    		System.out.println("耗费时间为:"+(endTm - startTm));
    	}
    }
    class LatchDemo implements Runnable{
    
    	//计数器
    	private CountDownLatch latch;
    	
    	public LatchDemo(CountDownLatch latch) {
    		this.latch = latch;
    	}
    	
    	@Override
    	public void run() {
    		for(int i=0;i<50000;i++){
    			if(i%2 == 0){
    				System.out.println(i);
    			}
    		}
    		//计数器 -1
    		latch.countDown();
    	}
    	
    }