Spring Bootで一括してAPIリクエストを送信する方法

2年 ago
夏樹, 風
1 minute

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パラメータを変更することができます。さらに、同時リクエストの数やコールバック処理ロジックは、実際のニーズに応じて変更できます。

bannerAds