CompletableFuture真香,可以替代CountDownLatch!

开发 前端
Future相当于一个占位符,代表一个操作将来的结果。一般通过get可以直接阻塞得到结果,或者让它异步执行然后通过callback回调结果。

[[420499]]

在对类的命名这篇长文中,我们提到了Future和Promise。

Future相当于一个占位符,代表一个操作将来的结果。一般通过get可以直接阻塞得到结果,或者让它异步执行然后通过callback回调结果。

但如果回调中嵌入了回调呢?如果层次很深,就是回调地狱。Java中的CompletableFuture其实就是Promise,用来解决回调地狱问题。Promise是为了让代码变得优美而存在的。

有多优美?这么说吧,一旦你使用了CompletableFuture,就会爱不释手,就像初恋女友一样,天天想着她。

一系列静态方法

从它的源代码中,我们可以看到,CompletableFuture直接提供了几个便捷的静态方法入口。其中有run和supply两组。

run的参数是Runnable,而supply的参数是Supplier。前者没有返回值,而后者有,否则没有什么两样。

这两组静态函数,都提供了传入自定义线程池的功能。如果你用的不是外置的线程池,那么它就会使用默认的ForkJoin线程池。默认的线程池,大小和用途你是控制不了的,所以还是建议自己传递一个。

典型的代码,写起来是这个样子。

  1. CompletableFuture<String> future = CompletableFuture.supplyAsync(()->{ 
  2.  return "test"
  3. }); 
  4. String result = future.join(); 

拿到CompletableFuture后,你就可以做更多的花样。

这些花样有很多

我们说面说了,CompletableFuture的主要作用,就是让代码写起来好看。配合Java8之后的stream流,可以把整个计算过程抽象成一个流。前面任务的计算结果,可以直接作为后面任务的输入,就像是管道一样。

  1. thenApply 
  2. thenApplyAsync 
  3. thenAccept 
  4. thenAcceptAsync 
  5. thenRun 
  6. thenRunAsync 
  7. thenCombine 
  8. thenCombineAsync 
  9. thenCompose 
  10. thenComposeAsync 

比如,下面代码的执行结果是99,并不因为是异步就打乱代码执行的顺序了。

  1. CompletableFuture<Integer> cf = CompletableFuture.supplyAsync(() -> 10) 
  2.                 .thenApplyAsync((e) -> { 
  3.                     try { 
  4.                         Thread.sleep(10000); 
  5.                     } catch (InterruptedException ex) { 
  6.                         ex.printStackTrace(); 
  7.                     } 
  8.                     return e * 10; 
  9.                 }).thenApplyAsync(e -> e - 1); 
  10.  
  11. cf.join(); 
  12. System.out.println(cf.get()); 

同样的,函数的作用还要看then后面的动词。

  • apply 有入参和返回值,入参为前置任务的输出
  • accept 有入参无返回值,会返回CompletableFuture
  • run 没有入参也没有返回值,同样会返回CompletableFuture
  • combine 形成一个复合的结构,连接两个CompletableFuture,并将它们的2个输出结果,作为combine的输入
  • compose 将嵌套的CompletableFuture平铺开,用来串联两个CompletableFuture

when和handle

上面的函数列表,其实还有很多。比如:

  1. whenComplete 

when的意思,就是任务完成时候的回调。比如我们上面的例子,打算在完成任务后,输出一个done。它也是属于只有入参没有出参的范畴,适合放在最后一步进行观测。

  1. CompletableFuture<Integer> cf = CompletableFuture.supplyAsync(() -> 10) 
  2.                 .thenApplyAsync((e) -> { 
  3.                     try { 
  4.                         Thread.sleep(1000); 
  5.                     } catch (InterruptedException ex) { 
  6.                         ex.printStackTrace(); 
  7.                     } 
  8.                     return e * 10; 
  9.                 }).thenApplyAsync(e -> e - 1) 
  10.                 .whenComplete((r, e)->{ 
  11.                     System.out.println("done"); 
  12.                 }) 
  13.                 ; 
  14.  
  15. cf.join(); 
  16. System.out.println(cf.get()); 

handle和exceptionally的作用,和whenComplete是非常像的。

  1. public CompletableFuture<T> exceptionally(Function<Throwable, ? extends T> fn); 
  2.  
  3. public <U> CompletionStage<U> handle(BiFunction<? super T, Throwable, ? extends U> fn); 

CompletableFuture的任务是串联的,如果它的其中某一步骤发生了异常,会影响后续代码的运行的。

exceptionally从名字就可以看出,是专门处理这种情况的。比如,我们强制某个步骤除以0,发生异常,捕获后返回-1,它将能够继续运行。

  1. CompletableFuture<Integer> cf = CompletableFuture.supplyAsync(() -> 10) 
  2.                 .thenApplyAsync(e->e/0) 
  3.                 .thenApplyAsync(e -> e - 1) 
  4.                 .exceptionally(ex->{ 
  5.                     System.out.println(ex); 
  6.                     return -1; 
  7.                 }); 
  8.  
  9. cf.join(); 
  10. System.out.println(cf.get()); 

handle更加高级一些,因为它除了一个异常参数,还有一个正常的入参。处理方法也都类似,不再赘述。

当然,CompletableFuture的函数不仅仅这些,还有更多,根据函数名称很容易能够了解到它的作用。它还可以替换复杂的CountDownLatch,这要涉及到几个比较难搞的函数。

替代CountDownLatch

考虑下面一个场景。某一个业务接口,需要处理几百个请求,请求之后再把这些结果给汇总起来。

如果顺序执行的话,假设每个接口耗时100ms,那么100个接口,耗时就需要10秒。假如我们并行去获取的话,那么效率就会提高。

使用CountDownLatch可以解决。

  1. ExecutorService executor = Executors.newFixedThreadPool(5); 
  2.  
  3. CountDownLatch countDown = new CountDownLatch(requests.size()); 
  4. for(Request request:requests){ 
  5.     executor.execute(()->{ 
  6.         try{ 
  7.         //some opts 
  8.         }finally{ 
  9.             countDown.countDown(); 
  10.         } 
  11.     }); 
  12. countDown.await(200,TimeUnit.MILLISECONDS); 

我们使用CompletableFuture来替换它。

  1. ExecutorService executor = Executors.newFixedThreadPool(5); 
  2.  
  3. List<CompletableFuture<Result>> futureList = requests 
  4.     .stream() 
  5.     .map(request-> 
  6.         CompletableFuture.supplyAsync(e->{ 
  7.             //some opts 
  8.         },executor)) 
  9.     .collect(Collectors.toList()); 
  10.  
  11. CompletableFuture<Void> allCF = CompletableFuture.allOf(futureList.toArray(new CompletableFuture[0])); 
  12.  
  13. allCF.join(); 

我们这里用到了一个主要的函数,那就是allOf,用来把所有的CompletableFuture组合在一起;类似的还有anyOf,表示只运行其中一个。常用的,还有三个函数:

  • thenAcceptBoth 处理两个任务的情况,有两个任务结果入参,无返回值
  • thenCombine 处理两个任务的情况,有入参有返回值,最喜欢
  • runAfterBoth 处理两个任务的情况,无入参,无返回值

End

自从认识了CompletableFuture,我已经很少硬编码Future了。相对于各种回调的嵌套,CompletableFuture为我们提供了更直观、更优美的API。在“多个任务等待完成状态”这个应用场景,CompletableFuture已经成了我的首选。

唯一的问题是,它的函数有点多,你需要熟悉一小段时间。另外,有一个小小的问题,个人觉得,这个类如果叫做Promise的话,就能够和JS的统一起来,算是锦上添花吧。

 

作者简介:小姐姐味道 (xjjdog),一个不允许程序员走弯路的公众号。聚焦基础架构和Linux。十年架构,日百亿流量,与你探讨高并发世界,给你不一样的味道。

 

责任编辑:武晓燕 来源: 小姐姐味道
相关推荐

2022-08-14 22:35:37

EurekaConsul

2017-12-21 15:48:11

JavaCompletable

2024-01-11 12:14:31

Async线程池任务

2014-02-21 15:17:23

云存储Time Machin

2014-11-28 16:04:36

浪潮

2024-01-08 08:44:06

2021-09-30 16:17:42

开源软件视频通话视频会议

2024-10-29 09:32:33

VMwareGCPAWS

2021-06-06 16:56:49

异步编程Completable

2021-09-08 06:51:53

CountDownLa闭锁源码

2021-06-30 09:20:18

NuShell工具Linux

2024-11-13 15:09:57

Java线程开发

2023-12-25 09:51:13

性能测试

2009-05-21 13:25:50

.NETCountDownLa微软

2024-04-02 09:40:39

多线程Java原子性

2024-08-06 09:43:54

Java 8工具编程

2024-10-28 13:31:33

性能@Async应用

2021-09-27 13:01:52

线程阻塞排查

2015-06-16 11:06:42

JavaCompletable

2021-03-16 15:12:57

CompletableFuture机制java
点赞
收藏

51CTO技术栈公众号