CompletableFuture

在工作中, 批量任务执行是经常会遇到的需求, 我们会遇到以下场景:

例如 多个线程执行任务, 任务完成不分先后顺序; 启动多个线程, 后执行的依赖上一个任务的返回结果等等

针对上面说的场景, java8中新引入了批量任务处理类CompletableFuture, 此类中涵盖了我们需要的操作方法, 例如CompletableFuture.allOf是与的关系, 每个都要执行完, CompletableFuture.anyOf是或的关系, 其中任意一个执行完
第一种: 批量任务不分先后顺序以下示例代码:

CompletableFuture.allOf(CompletableFuture.runAsync(() -> {
    Thread.currentThread().setName("线程A");
    for (int i = 0; i < 10; i++) {
        try {
            System.out.println(Thread.currentThread().getName()+"-"+i);
            TimeUnit.MILLISECONDS.sleep(new Random().nextInt(2000));
        } catch (InterruptedException e) {
            throw new RuntimeException(e);
        }
    }
}).thenAccept(unused -> {
    //线程A执行结束执行
    System.out.println(Thread.currentThread().getName()+"结束");
}),CompletableFuture.runAsync(() -> {
    Thread.currentThread().setName("线程B");
    for (int i = 0; i < 10; i++) {
        try {
            System.out.println(Thread.currentThread().getName()+"-"+i);
            TimeUnit.MILLISECONDS.sleep(new Random().nextInt(1000));
        } catch (InterruptedException e) {
            throw new RuntimeException(e);
        }
    }
}).thenAccept(unused -> {
    //线程B执行结束执行
    System.out.println(Thread.currentThread().getName()+"结束");
})).get();
System.out.println("End");
} catch (InterruptedException | ExecutionException e) {
throw new RuntimeException(e);
}

执行结果:
线程A-0
线程B-0
线程B-1
线程B-2
线程B-3
线程A-1
线程B-4
线程B-5
线程A-2
线程B-6
线程B-7
线程B-8
线程B-9
线程A-3
线程B结束
线程A-4
线程A-5
线程A-6
线程A-7
线程A-8
线程A-9
线程A结束
End

针对第二种, 执行任务依赖上一个任务的返回结果, 我们可以这么写:

@Test
public void testCalc2() {
   CompletableFuture.supplyAsync(() -> {
       ThreadUtil.sleep(3,TimeUnit.SECONDS);
       //模拟任务执行
       task("任务一");
       return 100;
   }).thenApplyAsync(integer -> {
       task("任务二");
       return integer + 100;
   }).thenAccept(integer -> print("结果: " + integer)).join();
}

输出结果为:

ForkJoinPool.commonPool-worker-9任务一 Start
ForkJoinPool.commonPool-worker-9 值: 1
ForkJoinPool.commonPool-worker-9 值: 2
ForkJoinPool.commonPool-worker-9 值: 3
ForkJoinPool.commonPool-worker-9 值: 4
ForkJoinPool.commonPool-worker-9 值: 5
ForkJoinPool.commonPool-worker-9任务一 End
ForkJoinPool.commonPool-worker-9任务二 Start
ForkJoinPool.commonPool-worker-9 值: 1
ForkJoinPool.commonPool-worker-9 值: 2
ForkJoinPool.commonPool-worker-9 值: 3
ForkJoinPool.commonPool-worker-9 值: 4
ForkJoinPool.commonPool-worker-9 值: 5
ForkJoinPool.commonPool-worker-9任务二 End
ForkJoinPool.commonPool-worker-9 结果: 200

假如我们有批量任务, 当前任务依赖前面的任务的返回结果, 但是要求不按顺序执行, 代码如下:

CompletableFuture.supplyAsync(() -> {
    print("第一个任务开始");
    ThreadUtil.sleep(3, TimeUnit.SECONDS);
     print("第一个任务结束");
    return 100;
}).thenCombine(CompletableFuture.supplyAsync(() -> {
    print("第二个任务开始");
    ThreadUtil.sleep(2, TimeUnit.SECONDS);
    print("第二个任务结束");
    return 100;
}), Integer::sum).thenAccept(r -> {
    print("结果为:" + r);
}).join();

上面的示例, 第二个任务先执行, 第一个后执行, 计算结果
输出如下:

ForkJoinPool.commonPool-worker-9 第一个任务开始
ForkJoinPool.commonPool-worker-2 第二个任务开始
ForkJoinPool.commonPool-worker-2 第二个任务结束
ForkJoinPool.commonPool-worker-9 第一个任务结束
ForkJoinPool.commonPool-worker-9 结果为:200
注意上面任务一和任务二是同时执行的