Java：当前线程运行完毕，再运行后续逻辑

一、问题描述

在程序设计中，可能存在这样的情景：主线程中存在一个子线程，子线程需要在执行完毕后为后续代码逻辑提供参数。但在代码执行时，子进程还没执行完毕，后续的代码已经开始执行了，这时候就会出现参数为空的异常，或是直接报错。

public class ThreadTest {
    public static void main(String[] args) {
        String str;
        Thread thread = null;
        thread = new Thread(new Runnable() {
            @Override
            public void run() {
                while(true){
                    System.out.println("Thread is running...");
                    break;
                }
            }
        });
        thread.start();

        System.out.println("Thread is finished...");
    }
}

执行结果：

因此，如何才能使得子进程执行完毕后才开始运行后续代码呢？

二、解决方法

1. join

Thread类提供了一个join( )方法，其原理是不断检查子进程是否执行完毕，当子进程未执行完成时，调用该子进程的进程被阻塞，直到子进程执行完毕后，再继续运行调用进程。

public class ThreadTest {
    public static void main(String[] args) {
        String str;
        Thread thread = null;
        thread = new Thread(new Runnable() {
            @Override
            public void run() {
                while(true){
                    System.out.println("Thread is running...");
                    break;
                }
            }
        });
        thread.start();
        try {
            thread.join();
        } catch (InterruptedException e) {
            e.printStackTrace();
        }

        System.out.println("Thread is finished...");
    }
}

执行结果：

2. CountDownLatch

 CountDownLatch允许一个或多个线程等待其他线程执行完毕后再运行。

•  CountDownLatch的构造函数接收int类型的参数作为计数器，若要等待N个点再执行后续逻辑，就传入N。
• 这里的N可以是N个线程，也可以是N个执行步骤。
• 当我们调用countDown( )方法时，N会减一。
• 调用await( ) 方法来阻塞当前线程，直到N减为0。

public class ThreadTest {
    public static void main(String[] args) {
        CountDownLatch count = new CountDownLatch(2);
        String str;
        Thread thread1, thread2 = null;
        thread1 = new Thread(new Runnable() {
            @Override
            public void run() {
                System.out.println("Thread1" + " is running...");
                count.countDown();
                System.out.println("Thread1 " + " is finished...");
            }
        });
        thread2 = new Thread(new Runnable() {
            @Override
            public void run() {
                System.out.println("Thread2" + " is running...");
                count.countDown();
                System.out.println("Thread2" + " is finished...");
            }
        });
        thread1.start();
        thread2.start();
        try {
            count.await();
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
        System.out.println("All is finished.");
    }
}

执行结果：