编程老司机带你玩转 CompletableFuture 异步编程
以下文章来源于程序通事 ,作者楼下小黑哥
点击上方“JavaGuide“
回复”面试突击“获取Github 70k+ Star项目精华集合而成的《JavaGuide面试突击版》
本文从实例出发,介绍 CompletableFuture
基本用法。不过讲的再多,不如亲自上手练习一下。所以建议各位小伙伴看完,上机练习一把,快速掌握 CompletableFuture
。
全文摘要:
Future
VSCompletableFuture
CompletableFuture
基本用法
0x00. 前言
一些业务场景我们需要使用多线程异步执行任务,加快任务执行速度。Java 提供 Runnable
Future<V>
两个接口用来实现异步任务逻辑。
虽然 Future<V>
可以获取任务执行结果,但是获取方式十方不变。我们不得不使用Future#get
阻塞调用线程,或者使用轮询方式判断 Future#isDone
任务是否结束,再获取结果。
这两种处理方式都不是很优雅,JDK8 之前并发类库没有提供相关的异步回调实现方式。没办法,我们只好借助第三方类库,如 Guava
,扩展 Future
,增加支持回调功能。相关代码如下:
虽然这种方式增强了 Java 异步编程能力,但是还是无法解决多个异步任务需要相互依赖的场景。
举一个生活上的例子,假如我们需要出去旅游,需要完成三个任务:
任务一:订购航班
任务二:订购酒店
任务三:订购租车服务
很显然任务一和任务二没有相关性,可以单独执行。但是任务三必须等待任务一与任务二结束之后,才能订购租车服务。
为了使任务三时执行时能获取到任务一与任务二执行结果,我们还需要借助 CountDownLatch
。
0x01. CompletableFuture
JDK8 之后,Java 新增一个功能十分强大的类:CompletableFuture
。单独使用这个类就可以轻松的完成上面的需求:
大家可以先不用管
CompletableFuture
相关API
,下面将会具体讲解。
对比 Future<V>
,CompletableFuture
优点在于:
不需要手工分配线程,JDK 自动分配
代码语义清晰,异步任务链式调用
支持编排异步任务
怎么样,是不是功能很强大?接下来抓稳了,小黑哥要发车了。
1.1 方法一览
首先来通过 IDE 查看下这个类提供的方法:
稍微数一下,这个类总共有 50 多个方法,我的天。。。
不过也不要怕,小黑哥帮你们归纳好了,跟着小黑哥的节奏,带你们掌握 CompletableFuture
。
1.2 创建 CompletableFuture 实例
创建 CompletableFuture
对象实例我们可以使用如下几个方法:
第一个方法创建一个具有默认结果的 CompletableFuture
,这个没啥好讲。我们重点讲述下下面四个异步方法。
前两个方法 runAsync
不支持返回值,而 supplyAsync
可以支持返回结果。
这个两个方法默认将会使用公共的 ForkJoinPool
线程池执行,这个线程池默认线程数是 CPU 的核数。
可以设置 JVM option:-Djava.util.concurrent.ForkJoinPool.common.parallelism 来设置 ForkJoinPool 线程池的线程数
使用共享线程池将会有个弊端,一旦有任务被阻塞,将会造成其他任务没机会执行。所以强烈建议使用后两个方法,根据任务类型不同,主动创建线程池,进行资源隔离,避免互相干扰。
1.3 设置任务结果
CompletableFuture
提供以下方法,可以主动设置任务结果。
1 boolean complete(T value)
2 boolean completeExceptionally(Throwable ex)
第一个方法,主动设置 CompletableFuture
任务执行结果,若返回 true
,表示设置成功。如果返回 false
,设置失败,这是因为任务已经执行结束,已经有了执行结果。
示例代码如下:
1// 执行异步任务
2CompletableFuture cf = CompletableFuture.supplyAsync(() -> {
3 System.out.println("cf 任务执行开始");
4 sleep(10, TimeUnit.SECONDS);
5 System.out.println("cf 任务执行结束");
6 return "楼下小黑哥";
7});
8//
9Executors.newSingleThreadScheduledExecutor().execute(() -> {
10 sleep(5, TimeUnit.SECONDS);
11 System.out.println("主动设置 cf 任务结果");
12 // 设置任务结果,由于 cf 任务未执行结束,结果返回 true
13 cf.complete("程序通事");
14});
15// 由于 cf 未执行结束,将会被阻塞。5 秒后,另外一个线程主动设置任务结果
16System.out.println("get:" + cf.get());
17// 等待 cf 任务执行结束
18sleep(10, TimeUnit.SECONDS);
19// 由于已经设置任务结果,cf 执行结束任务结果将会被抛弃
20System.out.println("get:" + cf.get());
21/***
22 * cf 任务执行开始
23 * 主动设置 cf 任务结果
24 * get:程序通事
25 * cf 任务执行结束
26 * get:程序通事
27*/
这里需要注意一点,一旦 complete
设置成功,CompletableFuture
返回结果就不会被更改,即使后续 CompletableFuture
任务执行结束。
第二个方法,给 CompletableFuture
设置异常对象。若设置成功,如果调用 get
等方法获取结果,将会抛错。
示例代码如下:
1// 执行异步任务
2CompletableFuture cf = CompletableFuture.supplyAsync(() -> {
3 System.out.println("cf 任务执行开始");
4 sleep(10, TimeUnit.SECONDS);
5 System.out.println("cf 任务执行结束");
6 return "楼下小黑哥";
7});
8//
9Executors.newSingleThreadScheduledExecutor().execute(() -> {
10 sleep(5, TimeUnit.SECONDS);
11 System.out.println("主动设置 cf 异常");
12 // 设置任务结果,由于 cf 任务未执行结束,结果返回 true
13 cf.completeExceptionally(new RuntimeException("啊,挂了"));
14});
15// 由于 cf 未执行结束,前 5 秒将会被阻塞。后续程序抛出异常,结束
16System.out.println("get:" + cf.get());
17/***
18 * cf 任务执行开始
19 * 主动设置 cf 异常
20 * java.util.concurrent.ExecutionException: java.lang.RuntimeException: 啊,挂了
21 * ......
22 */
1.4 CompletionStage
CompletableFuture
分别实现两个接口 Future
与 CompletionStage
。
Future
接口大家都比较熟悉,这里主要讲讲 CompletionStage
。
CompletableFuture
大部分方法来自CompletionStage
接口,正是因为这个接口,CompletableFuture
才有如此强大功能。
想要理解 CompletionStage
接口,我们需要先了解任务的时序关系的。我们可以将任务时序关系分为以下几种:
串行执行关系
并行执行关系
AND 汇聚关系
OR 汇聚关系
1.5 串行执行关系
任务串行执行,下一个任务必须等待上一个任务完成才可以继续执行。
CompletionStage
有四组接口可以描述串行这种关系,分别为:
thenApply
方法需要传入核心参数为 Function<T,R>
类型。这个类核心方法为:
1 R apply(T t)
所以这个接口将会把上一个任务返回结果当做入参,执行结束将会返回结果。
thenAccept
方法需要传入参数对象为 Consumer<T>
类型,这个类核心方法为:
1void accept(T t)
返回值 void
可以看出,这个方法不支持返回结果,但是需要将上一个任务执行结果当做参数传入。
thenRun
方法需要传入参数对象为 Runnable
类型,这个类大家应该都比较熟悉,核心方法既不支持传入参数,也不会返回执行结果。
thenCompose
方法作用与 thenApply
一样,只不过 thenCompose
需要返回新的 CompletionStage
。这么理解比较抽象,可以集合代码一起理解。
方法中带有 Async ,代表可以异步执行,这个系列还有重载方法,可以传入自定义的线程池,上图未展示,读者只可以自行查看 API。
最后我们通过代码展示 thenApply
使用方式:
1CompletableFuture<String> cf
2 = CompletableFuture.supplyAsync(() -> "hello,楼下小黑哥")// 1
3 .thenApply(s -> s + "@程序通事") // 2
4 .thenApply(String::toUpperCase); // 3
5System.out.println(cf.join());
6// 输出结果 HELLO,楼下小黑哥@程序通事
这段代码比较简单,首先我们开启一个异步任务,接着串行执行后续两个任务。任务 2 需要等待任务1 执行完成,任务 3 需要等待任务 2。
上面方法,大家需要记住了
Function<T,R>
,Consumer<T>
,Runnable
三者区别,根据场景选择使用。
1.6 AND 汇聚关系
AND 汇聚关系代表所有任务完成之后,才能进行下一个任务。
如上所示,只有任务 A 与任务 B 都完成之后,任务 C 才会开始执行。
CompletionStage
有以下接口描述这种关系。
thenCombine
方法核心参数 BiFunction
,作用与 Function
一样,只不过 BiFunction
可以接受两个参数,而 Function
只能接受一个参数。
thenAcceptBoth
方法核心参数BiConsumer
作用也与 Consumer
一样,不过其需要接受两个参数。
runAfterBoth
方法核心参数最简单,上面已经介绍过,不再介绍。
这三组方法只能完成两个任务 AND 汇聚关系,如果需要完成多个任务汇聚关系,需要使用 CompletableFuture#allOf
,不过这里需要注意,这个方法是不支持返回任务结果。
AND 汇聚关系相关示例代码,开头已经使用过了,这里再粘贴一下,方便大家理解:
1.7 OR 汇聚关系
有 AND 汇聚关系,当然也存在 OR 汇聚关系。OR 汇聚关系代表只要多个任务中任一任务完成,就可以接着接着执行下一任务。
CompletionStage
有以下接口描述这种关系:
前面三组接口方法传参与 AND 汇聚关系一致,这里也不再详细解释了。
当然 OR 汇聚关系可以使用 CompletableFuture#anyOf
执行多个任务。
下面示例代码展示如何使用 applyToEither
完成 OR 关系。
1CompletableFuture<String> cf
2 = CompletableFuture.supplyAsync(() -> {
3 sleep(5, TimeUnit.SECONDS);
4 return "hello,楼下小黑哥";
5});// 1
6
7CompletableFuture<String> cf2 = cf.supplyAsync(() -> {
8 sleep(3, TimeUnit.SECONDS);
9 return "hello,程序通事";
10});
11// 执行 OR 关系
12CompletableFuture<String> cf3 = cf2.applyToEither(cf, s -> s);
13
14// 输出结果,由于 cf2 只休眠 3 秒,优先执行完毕
15System.out.println(cf2.join());
16// 结果:hello,程序通事
1.8 异常处理
CompletableFuture
方法执行过程若产生异常,当调用 get
,join
获取任务结果才会抛出异常。
上面代码我们显示使用 try..catch
处理上面的异常。不过这种方式不太优雅,CompletionStage
提供几个方法,可以优雅处理异常。
exceptionally
使用方式类似于 try..catch
中 catch
代码块中异常处理。
whenComplete
与 handle
方法就类似于 try..catch..finanlly
中 finally
代码块。无论是否发生异常,都将会执行的。这两个方法区别在于 handle
支持返回结果。
下面示例代码展示 handle
用法:
1CompletableFuture<Integer>
2 f0 = CompletableFuture.supplyAsync(() -> (7 / 0))
3 .thenApply(r -> r * 10)
4 .handle((integer, throwable) -> {
5 // 如果异常存在,打印异常,并且返回默认值
6 if (throwable != null) {
7 throwable.printStackTrace();
8 return 0;
9 } else {
10 // 如果
11 return integer;
12 }
13 });
14
15
16System.out.println(f0.join());
17/**
18 *java.util.concurrent.CompletionException: java.lang.ArithmeticException: / by zero
19 * .....
20 *
21 * 0
22 */
0x02. 总结
JDK8 提供 CompletableFuture
功能非常强大,可以编排异步任务,完成串行执行,并行执行,AND 汇聚关系,OR 汇聚关系。
不过这个类方法实在太多,且方法还需要传入各种函数式接口,新手刚开始使用会直接会被弄懵逼。这里帮大家在总结一下三类核心参数的作用
Function
这类函数接口既支持接收参数,也支持返回值Consumer
这类接口函数只支持接受参数,不支持返回值Runnable
这类接口不支持接受参数,也不支持返回值
搞清楚函数参数作用以后,然后根据串行,AND 汇聚关系,OR 汇聚关系归纳一下相关方法,这样就比较好理解了
最后再贴一下,文章开头的思维导图,希望对你有帮助。
0x03. 帮助文档
极客时间-并发编程专栏
https://colobu.com/2016/02/29/Java-CompletableFuture
https://www.ibm.com/developerworks/cn/java/j-cf-of-jdk8/index.html
最后说一句(求关注)
CompletableFuture
很早之前就有关注,本以为跟 Future
一样,使用挺简单,谁知道学的时候才发现好难。各种 API 方法看的头有点大。