温馨提示×

springboot怎么批量请求接口

小亿
150
2023-10-20 04:55:40
栏目: 编程语言

在Spring Boot中,可以使用RestTemplate类来发送HTTP请求并获取响应。要实现批量请求接口,可以使用多线程来异步发送多个请求。

以下是一个示例代码,演示如何使用RestTemplate发送批量请求接口:

import org.springframework.http.HttpMethod;
import org.springframework.http.ResponseEntity;
import org.springframework.util.concurrent.ListenableFuture;
import org.springframework.util.concurrent.ListenableFutureCallback;
import org.springframework.web.client.AsyncRestTemplate;
import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.CountDownLatch;
public class ApiBatchRequest {
public static void main(String[] args) throws InterruptedException {
List<String> urls = new ArrayList<>();
urls.add("http://api.example.com/endpoint1");
urls.add("http://api.example.com/endpoint2");
urls.add("http://api.example.com/endpoint3");
// 创建异步RestTemplate
AsyncRestTemplate restTemplate = new AsyncRestTemplate();
// 设置并发请求数量
int concurrentRequests = 3;
CountDownLatch latch = new CountDownLatch(concurrentRequests);
for (int i = 0; i < concurrentRequests; i++) {
// 发送异步请求
ListenableFuture<ResponseEntity<String>> future = restTemplate.exchange(urls.get(i), HttpMethod.GET, null, String.class);
// 添加请求完成后的回调
future.addCallback(new ListenableFutureCallback<ResponseEntity<String>>() {
@Override
public void onSuccess(ResponseEntity<String> result) {
System.out.println("Response received: " + result.getBody());
latch.countDown();
}
@Override
public void onFailure(Throwable ex) {
System.out.println("Request failed: " + ex.getMessage());
latch.countDown();
}
});
}
// 等待所有请求完成
latch.await();
}
}

在上面的示例中,我们首先创建一个包含多个URL的列表。然后使用AsyncRestTemplate类发送每个URL的GET请求,并使用ListenableFutureCallback回调处理每个请求的结果。最后,使用CountDownLatch类等待所有请求完成。

请注意,上述示例中的代码仅支持GET请求。如果需要发送其他类型的请求(例如POST、PUT等),可以修改HttpMethod参数。另外,还可以根据实际需求来修改并发请求数量和回调处理逻辑。

0