在一個(gè)spring-boot項(xiàng)目中窗轩，需要請求第三方API苹丸，使用JDK原生的URLConnection過于繁瑣检痰，功能強(qiáng)大的Apache.HttpClient又需要額外引入县爬，導(dǎo)致微服務(wù)程序變大蔑担。搜索發(fā)現(xiàn)spring原生的RestTemplate已經(jīng)內(nèi)置到了spring-boot-starter-web笋庄，可以開箱即用滑负，所以實(shí)現(xiàn)了一個(gè)基于RestTemplate的工具類充包，如下
WebHelper
import org.apache.commons.lang.IllegalClassException;
import org.apache.http.impl.client.HttpClients;
import org.springframework.http.*;
import org.springframework.http.client.SimpleClientHttpRequestFactory;
import org.springframework.http.converter.HttpMessageConverter;
import org.springframework.http.converter.StringHttpMessageConverter;
import org.springframework.lang.NonNull;
import org.springframework.lang.Nullable;
import org.springframework.web.client.RestTemplate;
import javax.net.ssl.HttpsURLConnection;
import javax.net.ssl.SSLContext;
import javax.net.ssl.TrustManager;
import javax.net.ssl.X509TrustManager;
import javax.servlet.http.HttpServletRequest;
import java.io.IOException;
import java.net.HttpURLConnection;
import java.nio.charset.StandardCharsets;
import java.security.cert.X509Certificate;
import java.util.List;
import java.util.Map;
/**
* @author frank
* @date 2020-07-31 15:34
*/
public class WebHelper {
private static final String[] IP_HEADER_CANDIDATES = {
"X-Forwarded-For",
"Proxy-Client-IP",
"WL-Proxy-Client-IP",
"HTTP_X_FORWARDED_FOR",
"HTTP_X_FORWARDED",
"HTTP_X_CLUSTER_CLIENT_IP",
"HTTP_CLIENT_IP",
"HTTP_FORWARDED_FOR",
"HTTP_FORWARDED",
"HTTP_VIA",
"REMOTE_ADDR"
};
/**
* 獲取客戶端 IP 地址
*
* @param request 請求實(shí)例
* @return 客戶端 IP
*/
public static @Nullable
String getClientIpAddress(HttpServletRequest request) {
for (String header : IP_HEADER_CANDIDATES) {
String ip = request.getHeader(header);
if (ip != null && ip.length() != 0 && !"unknown".equalsIgnoreCase(ip)) {
return ip;
}
}
return request.getRemoteAddr();
}
/**
* 發(fā)送 HTTPS.GET 請求
*
* @param url 請求地址
* @param params 請求參數(shù)
* @param headers 請求頭
* @return 響應(yīng)實(shí)體挑格，可能需要對響應(yīng)結(jié)果的狀態(tài)碼進(jìn)行處理斤蔓，所以直接返回響應(yīng)實(shí)體
*/
public static ResponseEntity<String> sendGet(String url, Map<String, String> params, Map<String, String> headers) {
return sendRequest(url, params, null, headers, HttpMethod.GET, Boolean.TRUE);
}
/**
* 發(fā)送 HTTPS.POST 請求
*
* @param url 請求地址
* @param body 請求體
* @param headers 請求頭
* @return 響應(yīng)實(shí)體令花，可能需要對響應(yīng)結(jié)果的狀態(tài)碼進(jìn)行處理锦担，所以直接返回響應(yīng)實(shí)體
*/
public static ResponseEntity<String> sendPost(String url, String body, Map<String, String> headers) {
HttpClients.createDefault();
return sendRequest(url, null, body, headers, HttpMethod.POST, Boolean.TRUE);
}
/**
* 使用 springboot 內(nèi)置的 RestTemplate 發(fā)送 http 請求
*
* @param url 請求地址
* @param params 請求參數(shù)
* @param body 請求體
* @param headers 請求頭
* @param method 請求方法
* @param useSSL 是否使用安全套接字
* @return 響應(yīng)實(shí)體俭识，可能需要對響應(yīng)結(jié)果的狀態(tài)碼進(jìn)行處理，所以直接返回響應(yīng)實(shí)體
*/
public static ResponseEntity<String> sendRequest(String url, Map<String, String> params, String body,
Map<String, String> headers, HttpMethod method, boolean useSSL) {
RestTemplate rt = useSSL
? new RestTemplate(new HttpsClientRequestFactory())
: new RestTemplate();
// 字符編碼設(shè)為 UTF-8
List<HttpMessageConverter<?>> converters = rt.getMessageConverters();
converters.set(1, new StringHttpMessageConverter(StandardCharsets.UTF_8));
rt.setMessageConverters(converters);
HttpHeaders httpHeaders = new HttpHeaders();
if (!MapHelper.isEmpty(headers)) {
httpHeaders.setAll(headers);
}
HttpEntity<String> httpEntity = new HttpEntity<>(body, httpHeaders);
if (!MapHelper.isEmpty(params)) {
url = StringHelper.append(url, "?", MapHelper.toUrl(params, "UTF-8"));
}
return rt.exchange(url, method, httpEntity, String.class);
}
/**
* 覆蓋 SimpleClientHttpRequestFactory 的安全套接字層SSL
*/
public static class HttpsClientRequestFactory extends SimpleClientHttpRequestFactory {
@Override
protected void prepareConnection(@NonNull HttpURLConnection connection, @NonNull String httpMethod) throws IOException {
// to https connection
if (!(connection instanceof HttpsURLConnection)) {
throw new IllegalClassException("An instance of HttpsURLConnection is expected");
}
HttpsURLConnection httpsConnection = (HttpsURLConnection) connection;
ExceptionExecutor.withTry(() -> {
// new empty X509TrustManager
TrustManager[] trustAllCerts = new TrustManager[]{
new X509TrustManager() {
@Override
public java.security.cert.X509Certificate[] getAcceptedIssuers() {
return null;
}
@Override
public void checkClientTrusted(X509Certificate[] certs, String authType) {
}
@Override
public void checkServerTrusted(X509Certificate[] certs, String authType) {
}
}
};
SSLContext sslContext = SSLContext.getInstance("SSL");
sslContext.init(null, trustAllCerts, new java.security.SecureRandom());
// set to connection
httpsConnection.setSSLSocketFactory(sslContext.getSocketFactory());
httpsConnection.setHostnameVerifier((s, sslSession) -> true);
return null;
});
super.prepareConnection(httpsConnection, httpMethod);
}
}
}
MapHelper
import java.io.UnsupportedEncodingException;
import java.net.URLEncoder;
import java.util.Map;
import java.util.stream.Collectors;
public class MapHelper {
/**
* 根據(jù)指定字符集完成 URL 編碼洞渔，構(gòu)造 URL 參數(shù)字符串
*
* @param map {@code {"name":"frank", "gender":"male"}}
* @param charset URLEncode 使用的字符集
* @return name=frank&gender=male
*/
public static <K, V> String toUrl(Map<K, V> map, String charset) {
return map.entrySet().stream()
.map(e -> {
K k = e.getKey();
String val = "";
try {
URLEncoder.encode(StringHelper.toStr(e.getValue()), charset);
} catch (UnsupportedEncodingException ex) {
ex.printStackTrace();
}
return StringHelper.join("=", k, val);
})
.collect(Collectors.joining("&"));
}
}
StringHelper
import java.util.Arrays;
import java.util.Objects;
import java.util.stream.Collectors;
public class StringHelper {
private static final String EMPTY = "";
/**
* 以字符串形式拼接多個(gè)參數(shù)
*
* @param params 如果整個(gè)數(shù)組為空套媚，返回空串；如果其中值為空磁椒，將轉(zhuǎn)為空串
*/
public static String append(Object... params) {
return join(EMPTY, params);
}
/**
* 以字符串形式拼接多個(gè)參數(shù)堤瘤，以指定字符串分隔
*
* @param separator 分隔參數(shù)的字符串
* @param params 如果整個(gè)數(shù)組為空，返回空串浆熔；如果其中值為空本辐，將轉(zhuǎn)為空串
*/
public static String join(String separator, Object... params) {
if (params == null || params.length == 0) {
return EMPTY;
}
return Arrays.stream(params)
.map(StringHelper::toStr)
.collect(Collectors.joining(separator));
}
/**
* Object 轉(zhuǎn) String，如果為 null 時(shí)返回值為空串而不是 "null"
*/
public static String toStr(Object obj) {
return Objects.toString(obj, EMPTY);
}
}
