package com.ycl.utils.http;
|
|
import com.alibaba.fastjson2.JSON;
|
import com.ycl.properties.RequestProperties;
|
import lombok.RequiredArgsConstructor;
|
import lombok.extern.slf4j.Slf4j;
|
import org.apache.http.HttpEntity;
|
import org.apache.http.HttpResponse;
|
import org.apache.http.client.HttpClient;
|
import org.apache.http.client.methods.HttpGet;
|
import org.apache.http.client.methods.HttpPost;
|
import org.apache.http.entity.StringEntity;
|
import org.apache.http.impl.client.DefaultHttpRequestRetryHandler;
|
import org.apache.http.impl.client.HttpClientBuilder;
|
import org.apache.http.util.EntityUtils;
|
import org.springframework.stereotype.Component;
|
|
import java.io.IOException;
|
|
/**
|
* 请求可重试
|
*
|
* @author：xp
|
* @date：2024/4/18 14:30
|
*/
|
@Slf4j
|
@Component
|
@RequiredArgsConstructor
|
public class RetryHttpUtil {
|
|
private final RequestProperties requestProperties;
|
|
/**
|
* get请求
|
*
|
* @param url 请求地址(带参数)
|
* @param token token
|
* @param returnType 响应数据类型
|
* @param <T>
|
* @return
|
*/
|
public <T> T get(String url, String token, Class<T> returnType) {
|
HttpClient httpClient = HttpClientBuilder
|
.create()
|
.setRetryHandler(new DefaultHttpRequestRetryHandler(requestProperties.getRetry(), true))
|
.build();
|
try {
|
HttpGet request = new HttpGet(url);
|
request.setHeader("Authorization", "Bearer " + token);
|
HttpResponse response = httpClient.execute(request);
|
int statusCode = response.getStatusLine().getStatusCode();
|
if (statusCode >= 200 && statusCode < 300) {
|
// 请求成功
|
HttpEntity entity = response.getEntity();
|
// json字符串转响应数据
|
String responseBody = EntityUtils.toString(entity);
|
T t = JSON.parseObject(responseBody, returnType);
|
EntityUtils.consume(entity);
|
return t;
|
} else {
|
// 请求失败
|
log.error("Request failed with status code: " + statusCode);
|
}
|
} catch (IOException e) {
|
// 请求发生异常
|
log.error("Request failed with exception: " + e.getMessage());
|
}
|
return null;
|
}
|
|
/**
|
* post请求
|
*
|
* @param url 请求地址
|
* @param data 请求体参数
|
* @param token token
|
* @param returnType 返回的数据类型
|
* @param <T>
|
* @return
|
*/
|
public <T> T post(String url, Object data, String token, Class<T> returnType) {
|
HttpClient httpClient = HttpClientBuilder
|
.create()
|
.setRetryHandler(new DefaultHttpRequestRetryHandler(requestProperties.getRetry(), true))
|
.build();
|
try {
|
HttpPost request = new HttpPost(url);
|
request.setHeader("Authorization", "Bearer " + token);
|
request.setHeader("Content-Type", "application/json;charset=UTF-8");
|
StringEntity stringEntity = new StringEntity(JSON.toJSONString(data));
|
request.setEntity(stringEntity);
|
HttpResponse response = httpClient.execute(request);
|
int statusCode = response.getStatusLine().getStatusCode();
|
if (statusCode >= 200 && statusCode < 300) {
|
// 请求成功
|
HttpEntity entity = response.getEntity();
|
// json字符串转响应数据
|
String responseBody = EntityUtils.toString(entity);
|
T t = JSON.parseObject(responseBody, returnType);
|
EntityUtils.consume(entity);
|
return t;
|
} else {
|
// 请求失败
|
log.error("Request failed with status code: " + statusCode);
|
}
|
} catch (IOException e) {
|
// 请求发生异常
|
log.error("Request failed with exception: " + e.getMessage());
|
}
|
return null;
|
}
|
|
}
|
