多线程批量调用接口

首先定义一个类,该类封装批量请求参数

public class Result {
    private Integer id;
    private String requestParam;
    private Date date;

    public Integer getId() {
        return id;
    }

    public void setId(Integer id) {
        this.id = id;
    }

    public String getRequestParam() {
        return requestParam;
    }

    public void setRequestParam(String requestParam) {
        this.requestParam = requestParam;
    }

    public Date getDate() {
        return date;
    }

    public void setDate(Date date) {
        this.date = date;
    }
}

接下来模拟批量调用其他服务接口:

import com.google.common.collect.Lists;
import org.apache.commons.collections.CollectionUtils;

import java.util.List;
import java.util.concurrent.CompletableFuture;
import java.util.concurrent.TimeUnit;
import java.util.stream.Collectors;

public class ManyThreadCallTest {

    private static final int dataSize = 50;

    public static void main(String[] args){

        // 模拟从数据库查询数据
        List<Result> params = queryFromDB();
        // 2,模拟多线程批量调用接口
       ManyThreadCall(params);
    }

    // 模拟从数据库查询请求入参记录
    private static List<Result> queryFromDB() {
        List<Result> resultLists = Lists.newArrayList();
        for (int i = 0; i < dataSize; i++) {
            Result resultDto = new Result();
            resultDto.setRequestParam("WJ00000" + i);
            resultLists.add(resultDto);
        }
        return resultLists;
    }

    private static void ManyThreadCall(List<Result> params) {
        Long startTime= System.currentTimeMillis();
        List<Result> resultLists = Lists.newArrayList();
        // params集合拆分,List<List<Result>>里面每个List<Result>包含3条记录,这里也可以根据业务情况做调整,调整为4或者5都可以。
        List<List<Result>> partList = Lists.partition(params,3);
        for(List<Result> requestParms:partList){
            List<Result> resultData = getResultData(requestParms);
            if(CollectionUtils.isNotEmpty(resultData)){
                resultLists.addAll(resultData);
            }
        }
        System.out.println("ManyThreadCall 花费时间:"+(System.currentTimeMillis()-startTime));
    }

// CompletableFuture多线程调用并返回值
    public static List<Result> getResultData(List<Result> params) {

        List<CompletableFuture<Result>> completableFutureList = Lists.newArrayList();
        params.stream().forEach(param -> {
            try {
                CompletableFuture<Result> future = CompletableFuture.supplyAsync(() -> callRemoteInterface(param));
                completableFutureList.add(future);
            } catch (Exception e) {
                System.out.println("callRemoteInterface fail,requestParam:"+param.getRequestParam());
            }
        });
        List<Result> results = completableFutureList.stream().map(CompletableFuture::join).collect(Collectors.toList());
        return results;
    }

    // 伪代码模拟调用外部接口
    private static Result callRemoteInterface(Result resultDto) {
        try {
            // 这里假设调接口花费2秒
            TimeUnit.SECONDS.sleep(2L);//call壹次花費2秒鐘
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
        return resultDto;
    }

}
  • 0
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
在Spring Boot中,可以使用RestTemplate来调用接口批量调用接口的方式有多种。 一种方式是使用多线程并发调用接口。可以使用Java的Executor框架来实现多线调用,将需要调用接口放入一个List中,使用Executor框架的invokeAll方法批量执行这些任务,最后等待所有任务执行完成。示例代码如下: ```java RestTemplate restTemplate = new RestTemplate(); List<Callable<String>> tasks = new ArrayList<>(); tasks.add(() -> restTemplate.postForObject(url1, request1, String.class)); tasks.add(() -> restTemplate.postForObject(url2, request2, String.class)); tasks.add(() -> restTemplate.postForObject(url3, request3, String.class)); ExecutorService executorService = Executors.newFixedThreadPool(3); List<Future<String>> futures = executorService.invokeAll(tasks); executorService.shutdown(); for (Future<String> future : futures) { String result = future.get(); // 处理返回结果 } ``` 另一种方式是使用Java8的Stream API来批量调用接口。将需要调用接口放入一个List中,使用Stream API的map方法将每个接口调用转换成一个CompletableFuture,再使用CompletableFuture的allOf方法等待所有任务执行完成。示例代码如下: ```java RestTemplate restTemplate = new RestTemplate(); List<String> urls = Arrays.asList(url1, url2, url3); List<CompletableFuture<String>> futures = urls.stream() .map(url -> CompletableFuture.supplyAsync(() -> restTemplate.postForObject(url, request, String.class))) .collect(Collectors.toList()); CompletableFuture<Void> allFutures = CompletableFuture.allOf(futures.toArray(new CompletableFuture[0])); allFutures.join(); List<String> results = futures.stream() .map(CompletableFuture::join) .collect(Collectors.toList()); // 处理返回结果 ``` 以上两种方式都可以实现批量调用接口同步数据的功能,具体选择哪种方式要根据实际情况和需求来决定。需要注意的是,批量调用接口可能会对接口服务造成较大的负载,应该避免并发量过大而导致服务瘫痪。
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值