背景
在编写业务代码时经常遇到并发执行多个任务的需求,因为串行执行太慢,会影响业务代码性能。特别对于直接面向普通用户的业务来说用户体验至关重要,保证用户体验重要的一点是要“快”。业务代码中经常需要调用其它业务接口或者同时从多个数据源取数据再处理等,这种情况下势必要走网络请求,网络消耗必不可少,最好的情况是毫秒级别,一般情况下是几十毫秒级别,甚至几百毫秒,TimeoutException恐怕大家并不陌生。
例子
当你浏览微信朋友圈时,微信会把你朋友最近的动态展示在你朋友圈里,并且按照时间顺序最近的排在前面。你会发现,朋友圈里不会展示把你删除了或者你把他删除了的好友,不会展示被你设置过“不看他(她)的朋友圈”的好友或者对方把你设置过“不让他(她)看我的朋友圈”的好友,不会展示被你拉黑或把你拉黑的好友,不会展示被微信系统标记为spam的好友,等等。对于微信这种支持数亿人聊天的应用,其系统必定很复杂,解耦做得也比较好。 下面模拟下这个接口的实现:
1 public List<Feed> lastestFeeds(String wxId) { 2 //串行执行 3 //1. 获取你得好友列表 4 //2. 去掉把你删除的好友 5 //3. 去掉被你删除的好友 6 //4. 去掉被你设置过"不看他(她)的朋友圈"的好友 7 //5. 去掉对你设置过"不让他(她)看我的朋友圈"的好友 8 //6. 去掉被你拉黑和把你拉黑的好友 9 //7. 去掉被微信系统标记为作弊的好友 10 //... 11 //8. 获取好友最近动态再返回 12 }
对于微信这种复杂的系统,通常不可能从一个接口获取到这些信息,必须从多个接口获取到这些信息后再处理。如果说串行实现这些功能,你可以想象一下是不是慢到吐血,相信微信也不会这么干,否则朋友圈会刷半天也没响应,那么这个用户体验就太糟糕了。那么这个时候并发执行这些子任务就可以很高效的处理掉这种情况。具体到这个接口也就是会把1-7拆解成单个子任务,再丢到线程池异步的执行。最后执行完了,再汇总处理。
1 public List<Feed> lastestFeedsV2(String wxId) { 2 //并发运行,无先后先后执行 3 //1. 获取你得好友列表 4 //2. 去掉把你删除的好友 5 //3. 去掉被你删除的好友 6 //4. 去掉被你设置过"不看他(她)的朋友圈"的好友 7 //5. 去掉对你设置过"不让他(她)看我的朋友圈"的好友 8 //6. 去掉被你拉黑和把你拉黑的好友 9 //7. 去掉被微信系统标记为作弊的好友 10 //... 11 //等待所有子任务完成,汇总处理 12 //8. 获取好友最近动态再返回 13 }
那么如何实现并发运行呢,下面讨论几种实现。
CountDownLatch.await() VS ExecutorService.invokeAll()
从功能上讲这两者均可以实现并发执行多个任务并等待的功能。先看代码如何完成以上功能: CountDownLatch实现:
1 public List<Feed> lastestFeeds(String wxId) { 2 ThreadPoolExecutor executor = ...; 3 CountDownLatch latch = new CountDownLatch(7); 4 executor.execute(new Runnable() { 5 @Override 6 public void run() { 7 try{ 8 //1. 获取你得好友列表 9 } catch (Exception e) { 10 11 } finally { 12 latch.countDown(); 13 } 14 } 15 }); 16 17 executor.execute(new Runnable() { 18 @Override 19 public void run() { 20 try{ 21 //2. 去掉把你删除的好友 22 } catch (Exception e) { 23 24 } finally { 25 latch.countDown(); 26 } 27 } 28 }); 29 30 executor.execute(new Runnable() { 31 @Override 32 public void run() { 33 try{ 34 //3. 去掉被你删除的好友 35 } catch (Exception e) { 36 37 } finally { 38 latch.countDown(); 39 } 40 } 41 }); 42 43 executor.execute(new Runnable() { 44 @Override 45 public void run() { 46 try{ 47 //4. 去掉被你设置过"不看他(她)的朋友圈"的好友 48 } catch (Exception e) { 49 50 } finally { 51 latch.countDown(); 52 } 53 } 54 }); 55 56 executor.execute(new Runnable() { 57 @Override 58 public void run() { 59 try{ 60 //5. 去掉对你设置过"不让他(她)看我的朋友圈"的好友 61 } catch (Exception e) { 62 63 } finally { 64 latch.countDown(); 65 } 66 } 67 }); 68 69 executor.execute(new Runnable() { 70 @Override 71 public void run() { 72 try{ 73 //6. 去掉被你拉黑和把你拉黑的好友 74 } catch (Exception e) { 75 76 } finally { 77 latch.countDown(); 78 } 79 } 80 }); 81 82 executor.execute(new Runnable() { 83 @Override 84 public void run() { 85 try{ 86 //7. 去掉被微信系统标记为作弊的好友 87 } catch (Exception e) { 88 89 } finally { 90 latch.countDown(); 91 } 92 } 93 }); 94 try { 95 //latch.await(); 96 latch.await(500, TimeUnit.MILLISECONDS); 97 } catch (InterruptedException e) { 98 e.printStackTrace(); 99 } 100 101 //等待所有子任务完成,汇总处理 102 //8. 获取好友最近动态再返回 103 }
ExecutorService实现:
1 public List<Feed> lastestFeeds(String wxId) { 2 ThreadPoolExecutor executor = ...; 3 CountDownLatch latch = new CountDownLatch(7); 4 List<Callable<String>> tasks = new ArrayList<>(7); 5 tasks.add(new Callable<String>() { 6 @Override 7 public String call() throws Exception { 8 //1. 获取你得好友列表 9 return ...; 10 } 11 }); 12 tasks.add(new Callable<String>() { 13 @Override 14 public String call() throws Exception { 15 //2. 去掉把你删除的好友 16 return ...; 17 } 18 }); 19 tasks.add(new Callable<String>() { 20 @Override 21 public String call() throws Exception { 22 //3. 去掉被你删除的好友 23 return ...; 24 } 25 }); 26 tasks.add(new Callable<String>() { 27 @Override 28 public String call() throws Exception { 29 //4. 去掉被你设置过"不看他(她)的朋友圈"的好友 30 return ...; 31 } 32 }); 33 tasks.add(new Callable<String>() { 34 @Override 35 public String call() throws Exception { 36 //5. 去掉对你设置过"不让他(她)看我的朋友圈"的好友 37 return ...; 38 } 39 }); 40 tasks.add(new Callable<String>() { 41 @Override 42 public String call() throws Exception { 43 //6. 去掉被你拉黑和把你拉黑的好友 44 return ...; 45 } 46 }); 47 tasks.add(new Callable<String>() { 48 @Override 49 public String call() throws Exception { 50 //7. 去掉被微信系统标记为作弊的好友 51 return ...; 52 } 53 }); 54 try { 55// List<Future<String>> futureList = executor.invokeAll(tasks); 56 List<Future<String>> futureList = executor.invokeAll(tasks, 500, TimeUnit.MILLISECONDS); 57 for(Future<String> future : futureList) { 58 if(future.isCancelled()) { 59 //处理 60 } 61 try { 62 future.get(); 63 } catch (ExecutionException e) { 64 e.printStackTrace(); 65 } 66 } 67 } catch (InterruptedException e) { 68 e.printStackTrace(); 69 } 70 }
区别
两者实现的区别是什么?从性能上讲应该是没有太大差异,在此要感谢Doug Lea大神,感谢他优雅的实现,具体的实现,请各位看官自行百度和google。 我主要是谈谈这两者使用上的细微差异,可以根据具体业务场景选择一个更合适的。两者在等待子任务结束时,均提供了限时和不限时版本。
如果使用限时版本
关键的差异在于CountDownLatch超时后不再阻塞主线程,而是会继续执行,对于没有完成的子任务,它不知道也不会处理,也就是说没有完成的子任务还是会继续执行。而ExecutorService.invokeAll超时后,会取消所有在线程池任务队列中等待运行的子任务,说白了就是超时的子任务会被取消。 对于我举的微信这个例子来说,主线程等待超时后,子任务再返回数据是没有意义的,继续执行超时的子任务只会浪费CPU而已,尤其是对于QPS较大的业务来说,影响更明显。这种情况下选择ExecutorService.invokeAll可能更好些,当然如果你提供额外的状态标识如定义个AtomicBoolean来辅助实现超时不执行子任务的功能也是可以的。当然有些场景,即便主线程超时了,子任务也必须执行(例如子任务中涉及数据存储,否则数据可能丢失),这种情况下,应该是使用CountDownLatch。 希望读者诸君能够认真体会这点差异。
对于不限时版本
两者无太大差异,但在生产环境中要慎用,因为它不可控。
其它
另外再一点要注意的是,使用CountDownLatch时子任务的结果需要通过线程安全的容器收集如ConcurrentHashMap等,再处理。ExecutorService.invokeAll因为返回的是List<Future>,所以可以拿到结果,但如果想通过Future找到是哪个子任务,就只能根据顺序来确认了。