在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參數。另外,還可以根據實際需求來修改并發請求數量和回調處理邏輯。