package tech.powerjob.client.service.impl;
|
|
import com.google.common.collect.Maps;
|
import lombok.SneakyThrows;
|
import lombok.extern.slf4j.Slf4j;
|
import okhttp3.*;
|
import tech.powerjob.client.ClientConfig;
|
import tech.powerjob.client.common.Protocol;
|
import tech.powerjob.client.service.HttpResponse;
|
import tech.powerjob.client.service.PowerRequestBody;
|
import tech.powerjob.common.OmsConstant;
|
import tech.powerjob.common.serialize.JsonUtils;
|
|
import javax.net.ssl.*;
|
import java.io.IOException;
|
import java.security.SecureRandom;
|
import java.util.Map;
|
import java.util.Optional;
|
import java.util.Set;
|
import java.util.concurrent.TimeUnit;
|
|
/**
|
* desc
|
*
|
* @author tjq
|
* @since 2024/2/20
|
*/
|
@Slf4j
|
public class ClusterRequestServiceOkHttp3Impl extends AppAuthClusterRequestService {
|
|
private final OkHttpClient okHttpClient;
|
|
|
public ClusterRequestServiceOkHttp3Impl(ClientConfig config) {
|
super(config);
|
|
// 初始化 HTTP 客户端
|
if (Protocol.HTTPS.equals(config.getProtocol())) {
|
okHttpClient = initHttpsNoVerifyClient();
|
} else {
|
okHttpClient = initHttpClient();
|
}
|
}
|
|
@Override
|
protected HttpResponse sendHttpRequest(String url, PowerRequestBody powerRequestBody) throws IOException {
|
|
// 添加公共 header
|
powerRequestBody.addHeaders(config.getDefaultHeaders());
|
|
Object obj = powerRequestBody.getPayload();
|
|
RequestBody requestBody = null;
|
|
switch (powerRequestBody.getMime()) {
|
case APPLICATION_JSON:
|
MediaType jsonType = MediaType.parse(OmsConstant.JSON_MEDIA_TYPE);
|
String body = obj instanceof String ? (String) obj : JsonUtils.toJSONStringUnsafe(obj);
|
requestBody = RequestBody.create(jsonType, body);
|
|
break;
|
case APPLICATION_FORM:
|
FormBody.Builder formBuilder = new FormBody.Builder();
|
Map<String, String> formObj = (Map<String, String>) obj;
|
formObj.forEach(formBuilder::add);
|
requestBody = formBuilder.build();
|
}
|
|
Request request = new Request.Builder()
|
.post(requestBody)
|
.headers(Headers.of(powerRequestBody.getHeaders()))
|
.url(url)
|
.build();
|
|
try (Response response = okHttpClient.newCall(request).execute()) {
|
|
int code = response.code();
|
HttpResponse httpResponse = new HttpResponse()
|
.setCode(code)
|
.setSuccess(code == HTTP_SUCCESS_CODE);
|
|
ResponseBody body = response.body();
|
if (body != null) {
|
httpResponse.setResponse(body.string());
|
}
|
|
Headers respHeaders = response.headers();
|
Set<String> headerNames = respHeaders.names();
|
Map<String, String> respHeaderMap = Maps.newHashMap();
|
headerNames.forEach(hdKey -> respHeaderMap.put(hdKey, respHeaders.get(hdKey)));
|
|
httpResponse.setHeaders(respHeaderMap);
|
|
return httpResponse;
|
}
|
}
|
|
@SneakyThrows
|
private OkHttpClient initHttpClient() {
|
OkHttpClient.Builder okHttpBuilder = commonOkHttpBuilder();
|
return okHttpBuilder.build();
|
}
|
|
@SneakyThrows
|
private OkHttpClient initHttpsNoVerifyClient() {
|
|
X509TrustManager trustManager = new NoVerifyX509TrustManager();
|
|
SSLContext sslContext = SSLContext.getInstance("TLS");
|
sslContext.init(null, new TrustManager[]{trustManager}, new SecureRandom());
|
SSLSocketFactory sslSocketFactory = sslContext.getSocketFactory();
|
|
OkHttpClient.Builder okHttpBuilder = commonOkHttpBuilder();
|
|
// 不需要校验证书
|
okHttpBuilder.sslSocketFactory(sslSocketFactory, trustManager);
|
// 不校验 url中的 hostname
|
okHttpBuilder.hostnameVerifier((String hostname, SSLSession session) -> true);
|
|
|
return okHttpBuilder.build();
|
}
|
|
private OkHttpClient.Builder commonOkHttpBuilder() {
|
return new OkHttpClient.Builder()
|
// 设置读取超时时间
|
.readTimeout(Optional.ofNullable(config.getReadTimeout()).orElse(DEFAULT_TIMEOUT_SECONDS), TimeUnit.SECONDS)
|
// 设置写的超时时间
|
.writeTimeout(Optional.ofNullable(config.getWriteTimeout()).orElse(DEFAULT_TIMEOUT_SECONDS), TimeUnit.SECONDS)
|
// 设置连接超时时间
|
.connectTimeout(Optional.ofNullable(config.getConnectionTimeout()).orElse(DEFAULT_TIMEOUT_SECONDS), TimeUnit.SECONDS)
|
.callTimeout(Optional.ofNullable(config.getConnectionTimeout()).orElse(DEFAULT_TIMEOUT_SECONDS), TimeUnit.SECONDS);
|
}
|
|
@Override
|
public void close() throws IOException {
|
|
// 关闭 Dispatcher
|
okHttpClient.dispatcher().executorService().shutdown();
|
// 清理连接池
|
okHttpClient.connectionPool().evictAll();
|
// 清理缓存（如果有使用）
|
Cache cache = okHttpClient.cache();
|
if (cache != null) {
|
cache.close();
|
}
|
}
|
}
|
