JAVA多线程之CountDownLatch

时间：2024-01-23 14:53:45

前序:

上周测试给开发的同事所开发的模块提出了一个bug，并且还是偶现。

经过仔细查看代码，发现是在业务中启用了多线程，2个线程同时跑，但是新启动的2个线程必须保证一个完成之后另一个再继续运行，才能消除bug。

什么时候用？

多线程是在很多地方都会用到的，但是我们如果想要实现在某个特定的线程运行完之后，再启动另外一个线程呢，这个时候CountDownLatch就可以派上用场了

怎么用？

先看看普通的多线程代码：

 1 package code;
 2 
 3 public class MyThread extends Thread { 
 4     public static void main(String[] args) {      
 5         MyThread th = new MyThread();   
 6         Thread t1 = new Thread(th, "Mythread");        
 7         t1.start();  
 8         System.out.println(Thread.currentThread().getName());  
 9         }        
10         public void run() 
11             {     
12                 Mythread1 th2 = new Mythread1();   
13                 Thread t2 = new Thread(th2, "Mythread1");        
14                 t2.start();  
15                 System.out.println(this.currentThread().getName());  
16             }  
17         class Mythread1 extends Thread
18         {
19             public void run() {    
20                 try {
21                     Thread.sleep(1000);
22                 } catch (InterruptedException e) {
23                     // TODO Auto-generated catch block
24                     e.printStackTrace();
25                 }
26                 System.out.println(this.currentThread().getName()); 
27                 }  
28             
29         }
30     }

代码如上，先用MyThread继承了Thread类，然后在MyThread类内部又写了一个MyThread1类，同样也是继承了Thread类，并且在run方法里面让它睡1秒，这样运行代码，就会打印出：

从上面的输出顺序可以看出，先是启动了main线程，然后再启动了MyThread线程，在MyThread线程中，又启动了MyThread1线程。但是由于让MyThread1线程睡了1秒，模拟处理后续业务，这样他就比MyThread运行完毕的时间晚一些。

现在，在代码中加上CountDownLatch ，要让MyThread1先运行完毕，再让MyThread继续运行。

 1 package code;
 2 
 3 import java.util.concurrent.CountDownLatch;
 4 
 5 public class MyThread extends Thread { 
 6     CountDownLatch countDownLatch = new CountDownLatch(1);
 7     public static void main(String[] args) {      
 8         MyThread th = new MyThread();   
 9         Thread t1 = new Thread(th, "Mythread");        
10         t1.start();  
11         System.out.println(Thread.currentThread().getName());  
12         }        
13         public void run() 
14             {     
15                 Mythread1 th2 = new Mythread1();   
16                 Thread t2 = new Thread(th2, "Mythread1");        
17                 t2.start();  
18                 try {
19                     countDownLatch.await();
20                 } catch (InterruptedException e) {
21                     e.printStackTrace();
22                 }
23                 System.out.println(this.currentThread().getName());  
24             }  
25         class Mythread1 extends Thread
26         {
27             public void run() {    
28                 try {
29                     Thread.sleep(1000);
30                 } catch (InterruptedException e) {
31                     e.printStackTrace();
32                 }
33                 System.out.println(this.currentThread().getName()); 
34                 countDownLatch.countDown();
35                 }  
36             
37         }
38     }

代码写法如上所示，大致分三步

1，我们先new一个CountDownLatch对象入参设置为1（我个人理解的这个就像是new一个数组一样，什么时候数组清空了，那就可以让被中断的线程继续运行了）

2，在MyThread类中调用countDownLatch.await();让当前线程停止运行。

3，在Mythread1类中调用countDownLatch.countDown()方法。当Mythread1全部执行完毕，再最后调用该方法，作用就是把我说的“数组”清空。

看看输出的打印结果

结果如上图，是符合预期的结果的。

最后再说下CountDownLatch countDownLatch = new CountDownLatch(1)的入参，这块设置的是1，那就需要调用一次countDownLatch.countDown()减去1。

如果是其他数字，那就要调用相应的次数，否则调用countDownLatch.await()的线程都不会被继续执行。

 

相关文章