改造为可配置使用代理服务器
This commit is contained in:
parent
d5a02212ff
commit
2791f9c8db
|
@ -0,0 +1,132 @@
|
|||
package com.ruoyi.platform.utils;
|
||||
import org.apache.http.HttpHost;
|
||||
import org.apache.http.client.methods.*;
|
||||
import org.apache.http.client.utils.URIBuilder;
|
||||
import org.apache.http.conn.ssl.SSLConnectionSocketFactory;
|
||||
import org.apache.http.conn.ssl.TrustSelfSignedStrategy;
|
||||
import org.apache.http.entity.StringEntity;
|
||||
import org.apache.http.impl.client.CloseableHttpClient;
|
||||
import org.apache.http.impl.client.HttpClients;
|
||||
import org.apache.http.ssl.SSLContextBuilder;
|
||||
import org.apache.http.util.EntityUtils;
|
||||
import org.slf4j.Logger;
|
||||
import org.slf4j.LoggerFactory;
|
||||
import org.springframework.beans.factory.annotation.Value;
|
||||
import org.springframework.stereotype.Component;
|
||||
|
||||
import javax.annotation.PostConstruct;
|
||||
import javax.net.ssl.SSLContext;
|
||||
import java.io.IOException;
|
||||
import java.net.URI;
|
||||
import java.net.URISyntaxException;
|
||||
import java.nio.charset.StandardCharsets;
|
||||
|
||||
@Component
|
||||
public class NewHttpUtils {
|
||||
|
||||
private static final Logger log = LoggerFactory.getLogger(NewHttpUtils.class);
|
||||
|
||||
@Value("${proxy.useProxy:false}")
|
||||
private boolean useProxy;
|
||||
|
||||
@Value("${proxy.host}")
|
||||
private String host;
|
||||
|
||||
@Value("${proxy.port}")
|
||||
private Integer port;
|
||||
|
||||
private static CloseableHttpClient httpClient;
|
||||
|
||||
@PostConstruct
|
||||
public void init() {
|
||||
try {
|
||||
SSLContext sslContext = SSLContextBuilder.create()
|
||||
.loadTrustMaterial(new TrustSelfSignedStrategy())
|
||||
.build();
|
||||
|
||||
SSLConnectionSocketFactory sslSocketFactory = new SSLConnectionSocketFactory(sslContext);
|
||||
|
||||
if (useProxy) {
|
||||
httpClient = HttpClients.custom()
|
||||
.setSSLSocketFactory(sslSocketFactory)
|
||||
.setProxy(new HttpHost(host, port))
|
||||
.build();
|
||||
} else {
|
||||
httpClient = HttpClients.custom()
|
||||
.setSSLSocketFactory(sslSocketFactory)
|
||||
.build();
|
||||
}
|
||||
} catch (Exception e) {
|
||||
throw new RuntimeException("Failed to initialize HttpClient", e);
|
||||
}
|
||||
}
|
||||
|
||||
public static String sendGet(String url, String param) {
|
||||
return sendRequest(new HttpGet(), url, param, null, null);
|
||||
}
|
||||
|
||||
public static String sendPost(String url, String param, String body) {
|
||||
return sendRequest(new HttpPost(), url, param, null, body);
|
||||
}
|
||||
|
||||
public static String sendGetWithToken(String url, String param, String token) {
|
||||
return sendRequest(new HttpGet(), url, param, token, null);
|
||||
}
|
||||
|
||||
public static String sendPostWithToken(String url, String param, String token, String body) {
|
||||
return sendRequest(new HttpPost(), url, param, token, body);
|
||||
}
|
||||
|
||||
public static String sendDeleteWithToken(String url, String param, String token) {
|
||||
return sendRequest(new HttpDelete(), url, param, token, null);
|
||||
}
|
||||
|
||||
private static String sendRequest(HttpRequestBase request, String url, String param, String token, String body) {
|
||||
if (httpClient == null) {
|
||||
throw new IllegalStateException("HttpClient is not initialized");
|
||||
}
|
||||
|
||||
String result = "";
|
||||
try {
|
||||
URIBuilder uriBuilder = new URIBuilder(url);
|
||||
if (param != null && !param.isEmpty()) {
|
||||
String[] pairs = param.split("&");
|
||||
for (String pair : pairs) {
|
||||
int idx = pair.indexOf("=");
|
||||
if (idx != -1) {
|
||||
uriBuilder.setParameter(pair.substring(0, idx), pair.substring(idx + 1));
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
URI uri = uriBuilder.build();
|
||||
request.setURI(uri);
|
||||
|
||||
request.setHeader("Content-Type", "application/json");
|
||||
if (token != null) {
|
||||
request.setHeader("Authorization", "Bearer " + token);
|
||||
}
|
||||
|
||||
if (request instanceof HttpEntityEnclosingRequestBase && body != null) {
|
||||
((HttpEntityEnclosingRequestBase) request).setEntity(new StringEntity(body, StandardCharsets.UTF_8));
|
||||
}
|
||||
|
||||
log.info("Executing request: " + request.getRequestLine());
|
||||
|
||||
try (CloseableHttpResponse response = httpClient.execute(request)) {
|
||||
int statusCode = response.getStatusLine().getStatusCode();
|
||||
if (statusCode >= 200 && statusCode < 300) {
|
||||
result = EntityUtils.toString(response.getEntity(), StandardCharsets.UTF_8);
|
||||
} else {
|
||||
throw new IOException("HTTP request failed with response code: " + statusCode);
|
||||
}
|
||||
log.info("Response: " + result);
|
||||
}
|
||||
} catch (URISyntaxException e) {
|
||||
log.error("URISyntaxException, url=" + url + ", param=" + param, e);
|
||||
} catch (IOException e) {
|
||||
log.error("IOException, url=" + url + ", param=" + param, e);
|
||||
}
|
||||
return result;
|
||||
}
|
||||
}
|
Loading…
Reference in New Issue