JAVA利用CompletableFuture优雅实现多线程的异步编排

CompletableFuture多线程

一、CompletableFuture 介绍

CompletableFuture 是 Java 8 引入的一个类,位于 java.util.concurrent 包中。它是对 Future 接口的增强,提供了非常强大的功能来支持异步编程模型。CompletableFuture 实现了 FutureCompletionStage 接口,可以用来创建非阻塞代码,处理异步任务的结果,并且能够方便地进行链式调用。

主要特点

  1. 异步执行CompletableFuture 可以异步执行任务,这意味着任务可以在另一个线程上执行,而不会阻塞当前线程。
  2. 组合性:多个 CompletableFuture 实例可以组合在一起,形成复杂的异步流程。例如,一个 CompletableFuture 的结果可以作为另一个 CompletableFuture 的输入。
  3. 回调机制:提供了多种方法来注册回调函数,这些回调函数会在未来的某个时刻(如异步操作完成时)被自动调用。
  4. 异常处理:提供了一种处理异步任务中抛出的异常的方式。

二、常用方法

说明:使用CompletableFuture异步编排大多方法都会有一个重载方法,会多出一个executor参数,用来传来自定义的线程池,如果不传就会使用默认的线程池。

1、并行执行

1.1、有返回值 supplyAsync()
public static <U> CompletableFuture<U> supplyAsync(Supplier<U> supplier);
public static <U> CompletableFuture<U> supplyAsync(Supplier<U> supplier, Executor executor);

实例:

import java.util.concurrent.CompletableFuture;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;

public class CompletableFutureSupplyAsyncExample {
    public static void main(String[] args) throws ExecutionException, InterruptedException {
        // 使用默认的 ForkJoinPool.commonPool()
        CompletableFuture<String> future1 = CompletableFuture.supplyAsync(() -> {
            return "Hello from default pool";
        });

        // 使用自定义的 Executor
        ExecutorService customExecutor = Executors.newFixedThreadPool(2);
        CompletableFuture<String> future2 = CompletableFuture.supplyAsync(() -> { 
            return "Hello from custom pool";
        }, customExecutor);

        // 获取并打印结果,(get()这里需要处理异常,我直接抛出去了)
        System.out.println(future1.get());
        System.out.println(future2.get());

        // 关闭自定义的 Executor
        customExecutor.shutdown();
    }
}

打印:

Hello from default pool
Hello from custom pool
1.2、无返回值 runAsync()
public static CompletableFuture<Void> runAsync(Runnable runnable);
public static CompletableFuture<Void> runAsync(Runnable runnable, Executor executor);

2、串行执行

2.1、无入参,无返回值
// 使线程串行执行,无入参,无返回值
public CompletableFuture<Void> thenRun(Runnable action);
public CompletableFuture<Void> thenRunAsync(Runnable action);
public CompletableFuture<Void> thenRunAsync(Runnable action, Executor executor);

下面是对这三个方法的详细解释和示例。

  1. public CompletableFuture<Void> thenRun(Runnable action);
  • 描述:当 CompletableFuture 完成时,异步执行给定的 Runnable 任务。这个任务将在 CompletableFuture 完成后立即执行,但不会阻塞当前线程。
  • 执行环境:任务将在调用 thenRun 方法的线程中执行,或者在 CompletableFuture 完成时的线程中执行。
  1. public CompletableFuture<Void> thenRunAsync(Runnable action);
  • 描述:当 CompletableFuture 完成时,异步执行给定的 Runnable 任务。这个任务将在 ForkJoinPool.commonPool() 中执行。
  • 执行环境:任务将在 ForkJoinPool.commonPool() 中执行,这是一个共享的线程池。
  1. public CompletableFuture<Void> thenRunAsync(Runnable action, Executor executor);
  • 描述:当 CompletableFuture 完成时,异步执行给定的 Runnable 任务。这个任务将在指定的 Executor 中执行。
  • 执行环境:任务将在指定的 Executor 中执行,这使得你可以控制任务的执行环境,例如使用自定义的线程池或单线程执行器。

如果看不懂直接看实例吧(其实我也看不懂,哈哈哈)

实例:

package multithreading;

import java.util.concurrent.CompletableFuture;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;

public class CompletableFutureThenRunExample {
    public static void main(String[] args) throws Exception {
        // 创建一个异步任务
        //myTask
        CompletableFuture<String> future = CompletableFuture.supplyAsync(() -> {

            for (int i = 0; i < 10; i++) {
                System.out.print("myTask ");
            }
            System.out.println();

            return "Hello, World!";
        });

        // 使用 thenRun 方法
        //Task1
        CompletableFuture<Void> thenRunFuture = future.thenRun(() -> {
            System.out.println("thenRun: Task1 completed in " + Thread.currentThread().getName());
        });

        // 使用 thenRunAsync 方法
        //Task2
        CompletableFuture<Void> thenRunAsyncFuture = future.thenRunAsync(() -> {
            System.out.println("thenRunAsync: Task2 completed in " + Thread.currentThread().getName());
        });

        // 使用 thenRunAsync 方法并指定 Executor
        //Task3
        ExecutorService customExecutor = Executors.newFixedThreadPool(2);
        CompletableFuture<Void> thenRunAsyncWithExecutorFuture = future.thenRunAsync(() -> {
            System.out.println("thenRunAsyncWithExecutor: Task3 completed in " + Thread.currentThread().getName());
        }, customExecutor);

        // 等待所有任务完成
        thenRunFuture.get();
        thenRunAsyncFuture.get();
        thenRunAsyncWithExecutorFuture.get();

        // 关闭自定义的 Executor
        customExecutor.shutdown();
    }
}

打印:

myTask myTask myTask myTask myTask myTask myTask myTask myTask myTask 
thenRunAsync: Task2 completed in ForkJoinPool.commonPool-worker-2
thenRun: Task1 completed in ForkJoinPool.commonPool-worker-1
thenRunAsyncWithExecutor: Task3 completed in pool-1-thread-1

由此可见:

当执行完supplyAsync之后才会并行执行thenRun和thenRunAsync方法,而且supplyAsync只会执行一次。而且thenRun和thenRunAsync共用默认线程池。

2.2、有入参,无返回值
// 使线程串行执行,有入参,无返回值
public CompletableFuture<Void> thenAccept(Consumer<? super T> action);
public CompletableFuture<Void> thenAcceptAsync(Consumer<? super T> action);
public CompletableFuture<Void> thenAcceptAsync(Consumer<? super T> action, Executor executor);

实例:

package multithreading;

import java.util.concurrent.CompletableFuture;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;

public class CompletableFutureThenAcceptExample {
    public static void main(String[] args) throws Exception {
        // 创建一个异步任务
        CompletableFuture<String> future = CompletableFuture.supplyAsync(() -> {
            return "Hello, World!";
        });

        // 使用 thenAccept 方法
        CompletableFuture<Void> thenAcceptFuture = future.thenAccept(result -> {
            System.out.println("thenAccept: Result is " + result + " in " + Thread.currentThread().getName());
        });

        // 使用 thenAcceptAsync 方法
        CompletableFuture<Void> thenAcceptAsyncFuture = future.thenAcceptAsync(result -> {
            System.out.println("thenAcceptAsync: Result is " + result + " in " + Thread.currentThread().getName());
        });

        // 使用 thenAcceptAsync 方法并指定 Executor
        ExecutorService customExecutor = Executors.newFixedThreadPool(2);
        CompletableFuture<Void> thenAcceptAsyncWithExecutorFuture = future.thenAcceptAsync(result -> {
            System.out.println("thenAcceptAsyncWithExecutor: Result is " + result + " in " + Thread.currentThread().getName());
        }, customExecutor);

        // 等待所有任务完成
        thenAcceptFuture.get();
        thenAcceptAsyncFuture.get();
        thenAcceptAsyncWithExecutorFuture.get();

        // 关闭自定义的 Executor
        customExecutor.shutdown();
    }
}

打印:

thenAccept: Result is Hello, World! in main
thenAcceptAsync: Result is Hello, World! in ForkJoinPool.commonPool-worker-1
thenAcceptAsyncWithExecutor: Result is Hello, World! in pool-1-thread-1

解释:supplyAsync() 的返回结果可以通过 thenAccept(result ->{}) 的 result 传递到串行线程。

2.3、有入参,有返回值
// 使线程串行执行,有入参,有返回值
public <U> CompletableFuture<U> thenApply(Function<? super T,? extends U> fn);
public <U> CompletableFuture<U> thenApplyAsync(Function<? super T,? extends U> fn);
public <U> CompletableFuture<U> thenApplyAsync(Function<? super T,? extends U> fn, Executor executor);

实例:

import java.util.concurrent.CompletableFuture;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;

public class CompletableFutureThenApplyExample {
    public static void main(String[] args) throws Exception {
        // 创建一个异步任务
        CompletableFuture<String> future = CompletableFuture.supplyAsync(() -> {
            return "Hello, World!";
        });

        // 使用 thenApply 方法
        CompletableFuture<Integer> thenApplyFuture = future.thenApply(result -> {
            System.out.println("thenApply: Result is " + result + " in " + Thread.currentThread().getName());
            return result.length();
        });

        // 使用 thenApplyAsync 方法
        CompletableFuture<Integer> thenApplyAsyncFuture = future.thenApplyAsync(result -> {
            System.out.println("thenApplyAsync: Result is " + result + " in " + Thread.currentThread().getName());
            return result.length();
        });

        // 使用 thenApplyAsync 方法并指定 Executor
        ExecutorService customExecutor = Executors.newFixedThreadPool(2);
        CompletableFuture<Integer> thenApplyAsyncWithExecutorFuture = future.thenApplyAsync(result -> {
            System.out.println("thenApplyAsyncWithExecutor: Result is " + result + " in " + Thread.currentThread().getName());
            return result.length();
        }, customExecutor);

        // 等待所有任务完成并获取结果
        int length1 = thenApplyFuture.get();
        int length2 = thenApplyAsyncFuture.get();
        int length3 = thenApplyAsyncWithExecutorFuture.get();

        System.out.println("Length from thenApply: " + length1);
        System.out.println("Length from thenApplyAsync: " + length2);
        System.out.println("Length from thenApplyAsyncWithExecutor: " + length3);

        // 关闭自定义的 Executor
        customExecutor.shutdown();
    }
}

打印

thenApply: Result is Hello, World! in ForkJoinPool.commonPool-worker-1
thenApplyAsyncWithExecutor: Result is Hello, World! in pool-1-thread-1
thenApplyAsync: Result is Hello, World! in ForkJoinPool.commonPool-worker-2
Length from thenApply: 13
Length from thenApplyAsync: 13
Length from thenApplyAsyncWithExecutor: 13

3、等待所有任务完成

public static CompletableFuture<Void> allOf(CompletableFuture<?>... cfs);

实例:

import java.util.concurrent.CompletableFuture;
import java.util.concurrent.ExecutionException;

public class CompletableFutureAllOfExample {
    public static void main(String[] args) throws ExecutionException, InterruptedException {
        // 创建三个异步任务
        CompletableFuture<String> future1 = CompletableFuture.supplyAsync(() -> {
            return "Task 1";
        });

        CompletableFuture<String> future2 = CompletableFuture.supplyAsync(() -> {
            return "Task 2";
        });

        CompletableFuture<String> future3 = CompletableFuture.supplyAsync(() -> {
            return "Task 3";
        });

        // 使用 allOf 等待所有任务完成
        CompletableFuture<Void> allFutures = CompletableFuture.allOf(future2, future1, future3);

        // 等待 allOf 返回的 CompletableFuture 完成
        allFutures.get();

        // 打印每个任务的结果
        System.out.println("Task 1 result: " + future1.join());
        System.out.println("Task 2 result: " + future2.join());
        System.out.println("Task 3 result: " + future3.join());
    }
}

打印:

Task 1 result: Task 1
Task 2 result: Task 2
Task 3 result: Task 3

说明:会等待CompletableFuture. allOf 传入的线程会执行完,才继续往下执行代码

二、关于异常处理

1、使用 exceptionally方法:
  • public CompletableFuture<T> exceptionally(Function<Throwable, ? extends T> fn);
  • 如果之前的阶段因为异常而完成,那么 exceptionally 方法提供的函数会被调用,并且可以返回一个替代的结果或处理异常。例如:
CompletableFuture.supplyAsync(() -> {
    throw new RuntimeException("Error");
}).exceptionally(ex -> {
    System.out.println("Caught exception: " + ex.getMessage());
    return null; // 返回一个默认值
});
2、使用 handle方法
  • public <U> CompletableFuture<U> handle(BiFunction<? super T, Throwable, ? extends U> fn);
  • 不管前面的阶段是否正常完成还是因为异常而完成,handle 方法都会被调用。这使得你可以同时处理结果和异常。例如:
CompletableFuture.supplyAsync(() -> {
    throw new RuntimeException("Error");
}).handle((result, ex) -> {
    if (ex != null) {
        System.out.println("Caught exception: " + ex.getMessage());
        return null; // 返回一个默认值
    } else {
        return result;
    }
});
3、使用 whenComplete方法
CompletableFuture.supplyAsync(() -> {
    throw new RuntimeException("Error");
}).whenComplete((result, ex) -> {
    if (ex != null) {
        System.out.println("Caught exception: " + ex.getMessage());
    } else {
        System.out.println("Result: " + result);
    }
});
4、使用 completeExceptionally方法
  • public boolean completeExceptionally(Throwable ex);
  • 这个方法允许你手动地将一个异常与 CompletableFuture 关联起来。这通常用于模拟错误情况或在特定条件下提前完成未来。
5、链式调用
  • 当多个 CompletableFuture 被链式调用时(如 thenApply, thenAccept, thenRun 等),如果任何一步抛出异常,那么后续的步骤将不会被执行,除非它们是通过 exceptionallyhandle 方法定义的。

以上就是 CompletableFuture 对异常处理的主要方法。正确地处理异常对于确保应用程序的健壮性和可靠性至关重要,尤其是在进行异步编程时。

CompletableFutureJava 8 引入的一个类,用于实现多线程异步编程。它提供了丰富的方法来处理异步任务的结果、组合多个异步任务以及处理异常情况。 要实现超时处理,你可以结合 CompletableFuture 的方法和 Java 的 ScheduledExecutorService 来实现。下面是一个示例: ```java import java.util.concurrent.CompletableFuture; import java.util.concurrent.ExecutionException; import java.util.concurrent.Executors; import java.util.concurrent.ScheduledExecutorService; import java.util.concurrent.TimeUnit; import java.util.concurrent.TimeoutException; public class CompletableFutureTimeoutExample { public static void main(String[] args) { // 创建一个 ScheduledExecutorService 用于超时处理 ScheduledExecutorService executorService = Executors.newScheduledThreadPool(1); // 创建一个 CompletableFuture 对象 CompletableFuture<String> future = new CompletableFuture<>(); // 设置超时时间 long timeout = 5000; // 5秒 // 在指定时间后取消 CompletableFuture executorService.schedule(() -> { if (!future.isDone()) { future.completeExceptionally(new TimeoutException("Timeout")); } }, timeout, TimeUnit.MILLISECONDS); // 异步任务 CompletableFuture.runAsync(() -> { try { // 执行异步操作,例如网络请求、数据库查询等 String result = performAsyncTask(); // 当任务完成时,设置结果给 CompletableFuture future.complete(result); } catch (Exception e) { future.completeExceptionally(e); } }); // 获取结果或处理超时异常 try { String result = future.get(); // 处理成功结果 } catch (TimeoutException e) { // 处理超时异常 } catch (InterruptedException | ExecutionException e) { // 处理其他异常 } // 关闭 ScheduledExecutorService executorService.shutdown(); } private static String performAsyncTask() { // 执行异步任务的逻辑 return "Async task result"; } } ``` 在上面的示例中,我们创建了一个 CompletableFuture 对象,并使用 ScheduledExecutorService 在指定的超时时间后取消 CompletableFuture。然后,我们使用 CompletableFuture.runAsync 方法执行异步任务,并在任务完成时设置结果给 CompletableFuture。最后,我们使用 future.get() 方法来获取结果或处理超时异常。 希望以上信息能够帮助到你!如果你还有其他问题,请随时提问。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值