多线程情况下,如何让主线程在子线程任务运行完成时再进行?

多线程情况下,如何让主线程在子线程任务运行完成时再进行?

new 线程用 join

  • 当希望提升程序效率时,多线程通常是一种常用的方式,尤其是在支线任务耗时的情况下。考虑如下场景:多文件上传时,需要在所有文件上传完成后,将结果汇总返回。

  • 无论线程类是继承Thread类还是实现Runnable接口,只要在线程启动时,采用的是new的方式,就可以使用 join来达到主线程等待的目的。如下 :

  • 多线程情况下,如何让主线程在子线程任务运行完成时再进行?_第1张图片

  • UploadReq和UploadRes为两个自定义对象,UploadReq用于组合要上传的参数,如InputStream或File,以及一个唯一的key,UploadRes用于组合返回结果,如是否成功,上传后的路径等。通过key -> UploadRes的形式将上传结果组合成一个Map,便于使用

  • doUpload方法中,执行了真正的上传。并通过多线程的方式,使得多个文件可以同时上传,以提高程序效率。采用join方法,使主线程在子线程完成后再返回,以得到最终的完整结果集

线程池用CountDownLatch

  • 如果在开启任务时,使用的是线程池的形式,则要实现主任务在所有子任务执行完成后再返回的目的时,无法再使用join,这是因为,join的一个关键节点是子线程执行完成,但是,在线程池的模式下,子任务执行完成,其线程不一定销毁,而是会归还到线程池中保留,这种情况下,join将会一直等待下去。

  • 此时,需要使用CountDownLatch。如下:

  • 多线程情况下,如何让主线程在子线程任务运行完成时再进行?_第2张图片 

  • CountDownLatch的基本执行流程是,在构建时,给定一个初始值,如5,当调用countDown方法时,初始值会减1,当调用await方法时,会对初始值进行判断,如大于0,则等待,等于0,则向下执行。CountDownLatch的判断与线程状态无关,所以在线程池的情况下仍可满足需求。

你可能感兴趣的:(面试技巧,java,开发语言)