昨天项目组里的一名毕业生询问我, 如何知道异步线程的返回值, 这让我不觉想起来了两年前我参加招商系一个公司的面试, 在技术二面时, 面试官出过一道这样的编程题.
题目大概含义是: 我有一个需求是为了得到一个求和结果, 但是这个结果, 需要两个耗时大概 1s 左右计算功能的结果相加得到的, 麻烦用线程帮我实现, 方案越多越好, 然后默默的递过来纸和笔~~
首先我们分析一下, 这个题目肯定是需要多个线程执行的, 我们要抓住这个题目的关键点 : 主线程必须要等 2 个子线程执行完, 拿到子线程的结果进行相加, 得到最终结果.
其实, 实现这个题目的方式有很多种, 以我现在的观点来看, 面试官应该主要考察如下三点:
1. 多线程相关的基本知识点理解是否透彻, thread 没有返回值如何处理.
2. 思维是否活跃, 知识面是否广, 能想出多少种方案
3. 写代码是否规范
当年 too young, 我只写出了方案一和方案二的部分代码 (由于平时敲代码自动提示的比较多, 所以很多单词敲不出来).
不知道面试官当时是怎样的心境, 反正最终是拿到 offer 了, 不过因为个人原因, 去了另一家公司.
接下来, 我将两个数相加, 相个数相乘模仿两个耗时的计算功能, 用三种方案来解决这个问题.(我面试的时候写的代码可没有下面这么详细)
方案一: 使用 thread.join() 实现
java 中的 join 方法可以控制线程的执行顺序, 这个方案主要是考察线程的 join 方法原理, 以及 thread 的实现方式.
join() method suspends the execution of the calling thread until the object called finishes its execution.
大概的意思是: 如果在主线程 mian() 中调用子线程的 join() 方法, 就会阻塞主线程, 直到子线程执行完, 在唤起主线程继续执行.
至于为什么会阻塞主线程, 有兴趣的同学可以继续找资料学习, 这里就不多扩展了.
- package day01;
- public class JoinTest {
- public static void main(String[] args) {
- CalculateThread addTread = new CalculateThread(2, 3, "add");
- CalculateThread mutlTread = new CalculateThread(2, 3, "mutl");
- addTread.start(); // 子线程处理两个数相加
- mutlTread.start(); // 子线程处理两个数相乘
- try {
- addTread.join(); // 暂停主线程, 让 addTread 先执行完
- mutlTread.join(); // 暂停主线程, 让 mutlTread 子线程先执行完
- } catch (InterruptedException e) {
- e.printStackTrace();
- }
- int threadResult = addTread.getResult() + mutlTread.getResult();
- System.out.println("主线程获得两个子线程结果的和:" + threadResult);
- }
- }
- class CalculateThread extends Thread {
- private int param1;
- private int param2;
- private String type;
- private int result; // 用来保存线程执行结果
- public CalculateThread(int param1, int param2, String type) {
- this.param1 = param1;
- this.param2 = param2;
- this.type = type;
- }
- // 为了获取每个子线程计算结果
- public int getResult() {
- return result;
- }
- // 两个数相加
- public void add() {
- System.out.println(this.getName() + ": 子线程处理两个参数相加开始");
- try {
- Thread.sleep(2000L); // 模拟加法子线程需要执行 2s
- result = param1 + param2;
- } catch (InterruptedException e) {
- e.printStackTrace();
- }
- System.out.println(this.getName() + ": 子线程处理两个参数相加结束");
- }
- // 两个数相乘
- public void mult() {
- System.out.println(this.getName() + ": 子线程处理两个参数相乘开始");
- try {
- Thread.sleep(1000L); // 模拟乘法子线程需要执行 1s
- result = param1 * param2;
- } catch (InterruptedException e) {
- e.printStackTrace();
- }
- System.out.println(this.getName() + ": 子线程处理两个参数相乘结束");
- }
- @Override
- public void run() {
- // TODO Auto-generated method stub
- switch (type) {
- case "add": // switch 的 string 类型, jdk 1.7 才开始支持
- add();
- break;
- case "mutl":
- mult();
- break;
- default:
- break;
- }
- }
方案一注意:
1. join() 一定要在 start() 方法之后调用. 所以如果多个子线程执行, 要先循环执行完子线程的 start() 方法, 再循环执行 join() 方法, 这样才能变成并行执行.
如果执行一个子线程的 start() 方法后, 就直接执行这个子线程的 join() 方法, 由于主线程阻塞主了, 所以需要等这个线程执行完, 后面的线程才能执行, 就变成串行的了.
2. 每个子线程计算的返回值, 我们目前是用子线程里的变量保存实现的, 我们也可以用主线程的引用类型当作共享变量实现 (这个要考虑并发下, 线程安全问题).
方案二: 使用 Future 和 Callable 实现
Future 是可以保存返回值的, 这也是很多人知道的方案, Future 封装了多个方法, 可以很好的获取线程执行状态, 以及异常处理, 这里我们就不扩展了, 有兴趣的同学可以自己再去学习.
- package day01;
- import java.util.concurrent.Callable;
- import java.util.concurrent.ExecutionException;
- import java.util.concurrent.ExecutorService;
- import java.util.concurrent.Executors;
- import java.util.concurrent.Future;
- public class FutureTest {
- public static void main(String[] args) {
- try {
- CalculateCallable addCalable = new CalculateCallable(2, 3, "add");
- CalculateCallable mutlCalable = new CalculateCallable(2, 3, "mutl");
- // new 两个线程, 固定线程
- ExecutorService execute = Executors.newFixedThreadPool(2);
- // 返回 Futrue 对象后, 可以使用 Futrue.get() 方法获取返回值
- Future<Integer> addFuture = execute.submit(addCalable);
- Future<Integer> mutlFuture = execute.submit(mutlCalable);
- int futurResult = addFuture.get() + mutlFuture.get();
- System.out.println("主线程获得两个子线程结果的和:" + futurResult);
- } catch (ExecutionException e) {
- e.printStackTrace();
- } catch (InterruptedException e1) {
- e1.printStackTrace();
- }
- }
- }
- class CalculateCallable implements Callable<Integer> {
- private int param1;
- private int param2;
- private String type;
- public CalculateCallable(int param1, int param2, String type) {
- this.param1 = param1;
- this.param2 = param2;
- this.type = type;
- }
- // 两个数相加
- public int add() {
- System.out.println("加法子线程处理两个参数相加开始");
- try {
- Thread.sleep(2000L); // 模拟加法子线程需要执行 2s
- } catch (InterruptedException e) {
- e.printStackTrace();
- }
- System.out.println("加法子线程处理两个参数相加结束");
- return param1 + param2;
- }
- // 两个数相乘
- public int mult() {
- System.out.println("乘法子线程处理两个参数相乘开始");
- try {
- Thread.sleep(1000L); // 模拟乘法子线程需要执行 1s
- } catch (InterruptedException e) {
- e.printStackTrace();
- }
- System.out.println("乘法子线程处理两个参数相乘结束");
- return param1 * param2;
- }
- @Override
- public Integer call() throws Exception {
- // TODO Auto-generated method stub
- switch (type) {
- case "add": // switch 的 string 类型, jdk 1.7 才开始支持
- return add();
- case "mutl":
- return mult();
- default:
- return null;
- }
- }
- }
方案二注意:
1. Future 接口调用 get() 方法取得处理的结果值时是阻塞性的, 如果调用 Future 对象的 get() 方法时, 任务尚未执行完成, 则调用 get() 方法时一直阻塞到此任务完成为止.
如果这样, 则前面先执行的任务一旦耗时很多, 后面的任务调用 get() 方法就呈阻塞状态, 也就是排队进行等待. 主线程并不能保证首先获得结果的是最先完成任务线程的返回值, 大大影响运行效率. 那么使用多线程就没什么意义了.
幸运的是 JDK 并发包也提供了 CompletionService 接口可以解决这个问题, 它的 take() 方法哪个线程先完成就先获取谁的 Futrue 对象, 有兴趣的可以去仔细了解下相关知识点.
方案三: 使用 CountDownLatch 实现
这个方案, 当时面试的时候没想到, 但是是一个很好很强大的并发类.
CountDownLatch 存在于 java.util.concurrent 包下. CountDownLatch 这个类能够使一个线程等待其他线程完成各自的工作后再执行. 例如, 应用程序的主线程希望负责启动框架服务的线程在已经启动所有的框架服务之后再执行.
CountDownLatch 是通过一个计数器来实现的, 计数器的初始值为线程的数量. 每当一个线程完成了自己的任务后, 计数器的值就会减 1. 当计数器值到达 0 时, 它表示所有的线程已经完成了任务, 然后在闭锁上等待的线程就可以恢复执行任务.
- package day01;
- import java.util.concurrent.CountDownLatch;
- import java.util.concurrent.TimeUnit;
- public class CountDownLatchTest {
- public static void main(String[] args) {
- // 初始值为 2, 因为我们目前就 2 个子线程
- CountDownLatch countDownLatch = new CountDownLatch(2);
- // 每个线程中传入 countDownLatch
- CountDownThread addTread = new CountDownThread(2,3,"add",countDownLatch);
- CountDownThread mutlTread = new CountDownThread(2,3,"mutl",countDownLatch);
- addTread.start();
- mutlTread.start();
- try{
- // 设置超时时间为 3 秒, 3s 如果线程没有执行完, 返回 false
- boolean timeoutFlag = countDownLatch.await(3,TimeUnit.SECONDS);
- if(timeoutFlag){
- int threadResult = addTread.getResult() + mutlTread.getResult();
- System.out.println("主线程获得两个子线程结果的和:" + threadResult);
- }else{
- int threadResult = addTread.getResult() + mutlTread.getResult();
- System.out.println("主线程等待子线程执行超时:" + threadResult);
- }
- }catch (InterruptedException e){
- e.printStackTrace();
- }
- }
- }
- class CountDownThread extends Thread{
- private int param1;
- private int param2;
- private String type;
- private int result;
- private CountDownLatch countDownLatch;
- public CountDownThread(int param1,int param2,String type,CountDownLatch countDownLatch){
- this.param1 = param1;
- this.param2 = param2;
- this.type = type;
- this.countDownLatch = countDownLatch;
- }
- // 为了获取每个子线程计算结果
- public int getResult() {
- return result;
- }
- // 两个数相加
- public void add(){
- System.out.println(this.getName()+": 子线程处理两个参数相加开始");
- try{
- Thread.sleep(2000L); // 模拟加法子线程需要执行 2s
- result = param1 + param2;
- System.out.println(this.getName()+": 子线程处理两个参数相加结束");
- }catch(InterruptedException e){
- e.printStackTrace();
- }finally{
- // 计数器减 1
- countDownLatch.countDown();
- }
- }
- // 两个数相乘
- public void mult(){
- System.out.println(this.getName()+": 子线程处理两个参数相乘开始");
- try{
- Thread.sleep(1000L); // 模拟乘法子线程需要执行 1s
- result = param1 * param2;
- System.out.println(this.getName()+": 子线程处理两个参数相乘结束");
- }catch(InterruptedException e){
- e.printStackTrace();
- }finally{
- // 计数器减 1
- countDownLatch.countDown();
- }
- }
- @Override
- public void run() {
- // TODO Auto-generated method stub
- switch (type) {
- case "add": //switch 的 string 类型, jdk 1.7 才开始支持
- add();
- break;
- case "mutl":
- mult();
- break;
- default:
- break;
- }
- }
- }
如果我们将超时时间改成 1s ,boolean timeoutFlag = countDownLatch.await(1,TimeUnit.SECONDS); 由于计算相加时睡眠了 2s, 相乘时睡眠了 1s
所以, 相加的计算是直接超时的, timeoutFlag 会返回 false, 最终计算结果是 2 * 3 = 6.
方案三注意:
1. CountDownLatch 的缺点是 CountDownLatch 是一次性的, 计数器的值只能在构造方法中初始化一次, 之后没有任何机制再次对其设置值, 当 CountDownLatch 使用完毕后, 它不能再次被使用.
2.countDownLatch.countDown() 最好在 finally 块中执行, 防止子线程没有执行完, 就自减 1 了, 导致主线程没有等到所有子线程执行完, 便执行了.
其实我们在工作中遇到的业务场景往往比较复杂, 对并发, 异常的处理都比较严格, 这里只是给大家提供一个方向, 以后遇到类似的功能需求, 我们不会像无头苍蝇一样, 至少有了解决问题的方向.
来源: https://www.cnblogs.com/haly/p/10732332.html