多线程情况下,如何让主线程在子线程任务运行完成时再进行?
new 线程用 join
-
当希望提升程序效率时,多线程通常是一种常用的方式,尤其是在支线任务耗时的情况下。考虑如下场景:多文件上传时,需要在所有文件上传完成后,将结果汇总返回。
-
无论线程类是继承Thread类还是实现Runnable接口,只要在线程启动时,采用的是new的方式,就可以使用 join来达到主线程等待的目的。如下 :
-
UploadReq和UploadRes为两个自定义对象,UploadReq用于组合要上传的参数,如InputStream或File,以及一个唯一的key,UploadRes用于组合返回结果,如是否成功,上传后的路径等。通过key -> UploadRes的形式将上传结果组合成一个Map,便于使用
-
doUpload方法中,执行了真正的上传。并通过多线程的方式,使得多个文件可以同时上传,以提高程序效率。采用join方法,使主线程在子线程完成后再返回,以得到最终的完整结果集
线程池用CountDownLatch
-
如果在开启任务时,使用的是线程池的形式,则要实现主任务在所有子任务执行完成后再返回的目的时,无法再使用join,这是因为,join的一个关键节点是子线程执行完成,但是,在线程池的模式下,子任务执行完成,其线程不一定销毁,而是会归还到线程池中保留,这种情况下,join将会一直等待下去。
-
此时,需要使用CountDownLatch。如下:
-
-
CountDownLatch的基本执行流程是,在构建时,给定一个初始值,如5,当调用countDown方法时,初始值会减1,当调用await方法时,会对初始值进行判断,如大于0,则等待,等于0,则向下执行。CountDownLatch的判断与线程状态无关,所以在线程池的情况下仍可满足需求。