使用 CompletableFuture 和 ThreadPoolExecutor 实现并行的远程调用,并在所有调用完成后收集结果。
public class CallableDemo {
/**
* 创建线程池,并行处理多个rpc远程调用
*/
private ThreadPoolExecutor executor = new ThreadPoolExecutor(
5,
10,
1000,
TimeUnit.SECONDS,
new LinkedBlockingDeque<>());
public UserInfo getUserInfo(Long id) throws InterruptedException, ExecutionException {
final UserInfo userInfo = new UserInfo();
CompletableFuture userFuture = CompletableFuture.supplyAsync(() -> {
// 调用第一个prc接口
return Boolean.TRUE;
}, executor);
CompletableFuture bonusFuture = CompletableFuture.supplyAsync(() -> {
// 调用第二个rpc接口
return Boolean.TRUE;
}, executor);
CompletableFuture growthFuture = CompletableFuture.supplyAsync(() -> {
// 调用第三个rpc接口
return Boolean.TRUE;
}, executor);
CompletableFuture.allOf(userFuture, bonusFuture, growthFuture).join();
userFuture.get();
bonusFuture.get();
growthFuture.get();
return userInfo;
}
}
使用CompletableFuture进行异步调用
userFuture、bonusFuture 和 growthFuture:分别是三个 CompletableFuture 对象,每个都用 supplyAsync 方法异步执行一个不同的 RPC 调用。
CompletableFuture<Void> userFuture = CompletableFuture.supplyAsync(() -> {
// 调用第一个RPC接口的模拟
return Boolean.TRUE;
}, executor);
这里的 supplyAsync 方法接受一个 Supplier,它是一个无参数的函数接口,返回调用结果。
等待所有的异步任务完成
CompletableFuture.allOf(userFuture, bonusFuture, growthFuture).join();
CompletableFuture.allOf 方法接收一个 CompletableFuture 数组(或可变参数),返回一个新的 CompletableFuture,当所有传入的 CompletableFuture 完成时,它自己也完成。在这里,使用 .join() 方法等待所有任务完成。
获取每个异步任务的结果
userFuture.get();
bonusFuture.get();
growthFuture.get();
get() 方法会阻塞当前线程,直到 CompletableFuture 的计算完成并返回结果。在这里,我们调用 get() 方法来确保每个异步任务都完成了,尽管我们实际上并没有利用每个任务的返回值。