验证中...
语言: Java
分类: 常用工具包
最后更新于 2018-07-12 17:59
片段 1 片段 2
RestfulHttpClient.java
原始数据 复制代码
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588
package pres.lnk.utils;
import com.fasterxml.jackson.databind.ObjectMapper;
import org.apache.commons.lang3.ArrayUtils;
import org.apache.commons.lang3.StringUtils;
import java.io.*;
import java.net.HttpURLConnection;
import java.net.URL;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
/**
* <p>RestfulApi请求工具</p>
* <a href=http://blog.csdn.net/lnktoking/article/details/79174187">使用说明</a>
* @author lnk
*/
public class RestfulHttpClient {
/** 全局默认请求头 */
private static final Map<String, String> defaultHeaders = new HashMap<>();
/** 全局默认请求头 */
private static List<URLConnectionInitializer> initializers;
private static final String charset = "UTF-8";
private static final int readTimeout = 60000;
private static final int connectTimeout = 60000;
public static final String METHOD_GET = "GET";
public static final String METHOD_POST = "POST";
public static final String METHOD_PUT = "PUT";
public static final String METHOD_PATCH = "PATCH";
public static final String METHOD_DELETE = "DELETE";
private RestfulHttpClient(){ }
/**
* 发起请求
* @param method 请求方式:GET、POST、PUT、PATCH、DELETE
* @param url 请求url
* @param body 请求体body
* @throws IOException
*/
public static HttpResponse request(String method, String url, Object body) throws IOException {
return request(method, url, body, defaultHeaders);
}
/**
* 发起请求
* @param method 请求方式:GET、POST、PUT、PATCH、DELETE
* @param url 请求url
* @param body 请求体body
* @param headers 请求头
* @throws IOException
*/
public static HttpResponse request(String method, String url, Object body, Map<String, String> headers) throws IOException {
return getClient(url).method(method).body(body).headers(headers).request();
}
/**
* 获取请求客户端
* @param url
* @return
*/
public static HttpClient getClient(String url){
return new HttpClient(url);
}
/**
* 添加全局请求连接初始化器
* @param initializer
*/
public static void addInitializer(URLConnectionInitializer initializer){
if(initializer == null){
throw new NullPointerException("不能添加空的连接初始化器");
}
if(initializers == null){
initializers = new ArrayList<>();
}
initializers.add(initializer);
}
/**
* 请求客户端
*/
public static class HttpClient{
private Map<String, String> headers;
private int readTimeout = RestfulHttpClient.readTimeout;
private int connectTimeout = RestfulHttpClient.connectTimeout;
private String method = METHOD_GET;
private String url;
private Map<String, String> pathParams;
private Map<String, String> queryParams;
private Map<String, String> postParams;
private Object body;
private HttpResponse response;
private List<URLConnectionInitializer> initializers;
public HttpClient(String url){
if(StringUtils.isBlank(url)){
throw new IllegalArgumentException("请求地址不能为空");
}
this.url = url;
headers = new HashMap<>();
pathParams = new HashMap<>();
queryParams = new HashMap<>();
postParams = new HashMap<>();
headers.putAll(defaultHeaders);
initializers = RestfulHttpClient.initializers;
}
public HttpClient get(){
method = METHOD_GET;
return this;
}
public HttpClient post(){
method = METHOD_POST;
return this;
}
public HttpClient put(){
method = METHOD_PUT;
return this;
}
public HttpClient patch(){
method = METHOD_PATCH;
return this;
}
public HttpClient delete(){
method = METHOD_DELETE;
return this;
}
/**
* 添加请求头
*/
public HttpClient addHeader(String key, String value){
this.headers.put(key, value);
return this;
}
/**
* 批量添加请求头
*/
public HttpClient addHeaders(Map<String, String> headers){
this.headers.putAll(headers);
return this;
}
/**
* 设置读取超时时间
*/
public HttpClient readTimeout(int readTimeout){
this.readTimeout = readTimeout;
return this;
}
/**
* 设置连接超时时间
*/
public HttpClient connectTimeout(int connectTimeout){
this.connectTimeout = connectTimeout;
return this;
}
/**
* 重置请求头
*/
public HttpClient headers(Map<String, String> headers){
this.headers.clear();
this.headers.putAll(defaultHeaders);
return addHeaders(headers);
}
/**
* 设置请求方式,默认:GET
*/
public HttpClient method(String method) {
if(StringUtils.isBlank(method)){
throw new IllegalArgumentException("请求方式不能为空");
}
//请求方式不做限制
// if(!ArrayUtils.contains(new String[]{METHOD_GET, METHOD_POST, METHOD_PUT, METHOD_PATCH, METHOD_DELETE}, method.toUpperCase())){
// throw new IllegalArgumentException("请求方式设置错误,不能设置为:" + method);
// }
this.method = method.toUpperCase();
return this;
}
/**
* 设置请求地址,可带path参数
* 如:/user/{id}
*/
public HttpClient url(String url) {
this.url = url;
return this;
}
/**
* 设置请求地址参数,替换url上的参数
* 如:/user/{id} 上的{id}
*/
public HttpClient pathParams(Map<String, String> pathParams) {
this.pathParams.putAll(pathParams);
return this;
}
/**
* 添加请求地址参数,替换url上的参数
* 如:/user/{id} 上的{id}
*/
public HttpClient addPathParam(String key, String value) {
this.pathParams.put(key, value);
return this;
}
/**
* 设置url请求参数,url问题后面的参数
*/
public HttpClient queryParams(Map<String, String> queryParams) {
this.queryParams.putAll(queryParams);
return this;
}
/**
* 添加url请求参数,url问题后面的参数
*/
public HttpClient addQueryParam(String key, String value) {
this.queryParams.put(key, value);
return this;
}
/**
* 设置表单参数,与body参数冲突,只能设置其中一个,优先使用body
*/
public HttpClient postParams(Map<String, String> postParams) {
this.postParams.putAll(postParams);
return this;
}
/**
* 添加表单参数
*/
public HttpClient addPostParam(String key, String value) {
this.postParams.put(key, value);
return this;
}
/**
* 设置请求体body,与post参数冲突,只能设置其中一个
*/
public HttpClient body(Object body) {
this.body = body;
return this;
}
/**
* 获取最终的请求地址
* @return
*/
public String getRequestUrl(){
return transformUrl(this.url, pathParams, queryParams);
}
/**
* 发起请求
* @return
* @throws IOException
*/
public HttpResponse request() throws IOException {
response = RestfulHttpClient.request(this);
return response;
}
/**
* 添加请求连接初始化器
* @param initializer
*/
public HttpClient addInitializer(URLConnectionInitializer initializer){
if(initializer == null){
throw new NullPointerException("不能添加空的连接初始化器");
}
if(initializers == null){
initializers = new ArrayList<>();
}
initializers.add(initializer);
return this;
}
public Map<String, String> getHeaders() {
return headers;
}
public int getReadTimeout() {
return readTimeout;
}
public int getConnectTimeout() {
return connectTimeout;
}
public String getMethod() {
return method;
}
public String getUrl() {
return url;
}
public Map<String, String> getQueryParams() {
return queryParams;
}
public Map<String, String> getPathParams() {
return pathParams;
}
public Map<String, String> getPostParams() {
return postParams;
}
public Object getBody() {
return body;
}
public HttpResponse getResponse() {
return response;
}
}
/**
* 请求响应结果
*/
public static class HttpResponse{
private int code;
private Map<String, List<String>> headers;
private String requestUrl;
private String content;
public HttpResponse(int code, Map<String, List<String>> headers, String requestUrl, String content){
this.code = code;
this.headers = headers;
this.requestUrl = requestUrl;
this.content = content;
}
public <T>T getContent(Class<T> clz) throws IOException {
if(StringUtils.isNotBlank(content)){
return new ObjectMapper().readValue(content, clz);
}
return null;
}
/**
* 获取响应状态码
*/
public int getCode() {
return code;
}
/**
* 获取响应头
*/
public Map<String, List<String>> getHeaders() {
return headers;
}
/**
* 获取最后请求地址
*/
public String getRequestUrl() {
return requestUrl;
}
/**
* 获取响应内容
*/
public String getContent() {
return content;
}
}
/**
* 发起请求
* @throws IOException
*/
private static HttpResponse request(HttpClient client) throws IOException {
HttpURLConnection con = instance(client);
if(METHOD_GET.equalsIgnoreCase(client.getMethod())){
//GET请求,不用发请求体
return readResponse(con);
}
String requestBody = null;
if(isPrimitiveOrWrapClzOrStr(client.getBody())){
requestBody = client.getBody().toString();
}else if(client.getBody() != null){
requestBody = new ObjectMapper().writeValueAsString(client.getBody());
if(!client.getHeaders().containsKey("Content-Type")){
//设置请求媒体类型为json提交
con.setRequestProperty("Content-Type", "application/json;charset=UTF-8");
}
}else if(client.getPostParams() != null && !client.getPostParams().isEmpty()){
requestBody = toUrlParams(client.getPostParams());
}
con.setDoOutput(true);
con.setDoInput(true);
BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(con.getOutputStream(), charset));
if(requestBody != null){
//写入请求内容
bw.write(requestBody);
}
bw.close();
return readResponse(con);
}
/**
* 判断是否是字符串或基本类型或包装类型
* @param o
* @return
*/
private static boolean isPrimitiveOrWrapClzOrStr(Object o){
if(o == null){
return false;
}else if(o instanceof String){
//是字符串类型
return true;
}else if(o.getClass().isPrimitive()){
//是基本类型
return true;
}else{
try {
//是包装类型
return ((Class) o.getClass().getField("TYPE").get(null)).isPrimitive();
} catch (Exception e) {
return false;
}
}
}
/**
* 读取响应结果
* @param con
* @return
* @throws IOException
*/
private static HttpResponse readResponse(HttpURLConnection con) throws IOException {
try{
BufferedReader br = new BufferedReader(new InputStreamReader(con.getInputStream(), charset));
StringBuilder sb = new StringBuilder();
String read = null;
while ((read = br.readLine()) != null) {
sb.append(read);
sb.append("\n");
}
br.close();
return new HttpResponse(con.getResponseCode(), con.getHeaderFields(),
con.getURL().toString(), sb.toString());
} finally {
con.disconnect();
}
}
/**
* 初始化请求连接
* @param client
* @return
* @throws IOException
*/
private static HttpURLConnection instance(HttpClient client) throws IOException {
URL u = new URL(client.getRequestUrl());
HttpURLConnection con = (HttpURLConnection) u.openConnection();
con.setReadTimeout(client.getReadTimeout());
con.setConnectTimeout(client.getConnectTimeout());
con.setRequestMethod(client.getMethod());
Map<String, String> headers = client.getHeaders();
if(headers != null && !headers.isEmpty()){
for (String key : headers.keySet()) {
con.setRequestProperty(key, headers.get(key));
}
}
List<URLConnectionInitializer> initializers = client.initializers;
if(initializers != null && !initializers.isEmpty()){
for (URLConnectionInitializer initializer : initializers) {
HttpURLConnection init = initializer.init(con, client);
if(init != null){
con = init;
}
}
}
return con;
}
/**
* 处理url的参数
* 如:/user/{id},将id转成值
* @param url
* @param pathParams 地址参数
* @param queryParams 请求参数
* @return
*/
private static String transformUrl(String url, Map<String, String> pathParams, Map<String, String> queryParams){
if(pathParams != null && !pathParams.isEmpty()){
for (String key : pathParams.keySet()) {
url = url.replaceAll("\\{"+key+"\\}", pathParams.get(key));
}
}
if(queryParams != null && !queryParams.isEmpty()){
if(url.indexOf("?") > 0){
url += "&" + toUrlParams(queryParams);
}else{
url += "?" + toUrlParams(queryParams);
}
}
return url;
}
/**
* 将map参数转成url参数形式:name1=value2&name2=value2...
* @param paras
* @return
*/
public static String toUrlParams(Map<String, String> paras){
if(paras != null && !paras.isEmpty()){
StringBuffer urlParams = new StringBuffer();
for (String k : paras.keySet()) {
urlParams.append(k+"="+paras.get(k)+"&");
}
if(urlParams.length() > 0){
return urlParams.substring(0, urlParams.length() - 1);
}
}
return null;
}
/**
* 获取全局默认请求headers设置
* @return
*/
public static Map<String, String> getDefaultHeaders(){
return defaultHeaders;
}
/**
* 设置默认全局默认请求headers
* @param headers
* @return
*/
public static Map<String, String> setDefaultHeaders(Map<String, String> headers){
defaultHeaders.clear();
defaultHeaders.putAll(headers);
return getDefaultHeaders();
}
/**
* 连接初始化器
*/
public interface URLConnectionInitializer{
/**
* 初始化http请求连接对象
* @param connection
* @return
*/
HttpURLConnection init(HttpURLConnection connection, HttpClient client);
}
public static void main(String[] args) throws IOException {
String url = "http://localhost:8081/user/{id}";
RestfulHttpClient.HttpResponse response = getClient(url)
.post() //设置post请求
.addHeader("token", "tokenValue") //添加header
.addPathParam("id", "2") //设置url路径参数
.addQueryParam("test", "value") //设置url请求参数
.request(); //发起请求
System.out.println(response.getCode()); //响应状态码
System.out.println(response.getRequestUrl());//最终发起请求的地址
if(response.getCode() == 200){
//请求成功
System.out.println(response.getContent()); //响应内容
}
}
}
TrustAllHttpsInitializer.java
原始数据 复制代码
package pres.lnk.utils;
import javax.net.ssl.*;
import java.io.IOException;
import java.net.HttpURLConnection;
import java.security.cert.CertificateException;
import java.security.cert.X509Certificate;
/**
* 信任所有https的请求
* 注意:使用该初始器化之后会存在安全认证风险,慎用
* @Author lnk
* @Date 2018/5/29
*/
public class TrustAllHttpsInitializer implements RestfulHttpClient.URLConnectionInitializer {
@Override
public HttpURLConnection init(HttpURLConnection connection, RestfulHttpClient.HttpClient client) {
String protocol = connection.getURL().getProtocol();
if ("https".equalsIgnoreCase(protocol)) {
HttpsURLConnection https = (HttpsURLConnection) connection;
trustAllHosts(https);
https.setHostnameVerifier(DO_NOT_VERIFY);
return https;
}
return connection;
}
/**
* 覆盖java默认的证书验证
*/
private static final TrustManager[] trustAllCerts = new TrustManager[]{new X509TrustManager() {
@Override
public void checkClientTrusted(X509Certificate[] x509Certificates, String s)
throws CertificateException {
}
@Override
public void checkServerTrusted(X509Certificate[] x509Certificates, String s)
throws CertificateException {
}
@Override
public X509Certificate[] getAcceptedIssuers() {
return null;
}
}};
/**
* 设置不验证主机
*/
private static final HostnameVerifier DO_NOT_VERIFY = new HostnameVerifier() {
@Override
public boolean verify(String hostname, SSLSession session) {
return true;
}
};
/**
* 信任所有
* @param connection
* @return
*/
private static SSLSocketFactory trustAllHosts(HttpsURLConnection connection) {
SSLSocketFactory oldFactory = connection.getSSLSocketFactory();
try {
SSLContext sc = SSLContext.getInstance("SSL");
sc.init(null, trustAllCerts, null);
SSLSocketFactory newFactory = sc.getSocketFactory();
connection.setSSLSocketFactory(newFactory);
} catch (Exception e) {
e.printStackTrace();
}
return oldFactory;
}
public static void main(String[] args) throws IOException {
String url = "https://b2b.10086.cn/b2b/main/viewNoticeContent.html?noticeBean.id=";
RestfulHttpClient.HttpResponse response = RestfulHttpClient.getClient(url)
.addInitializer(new TrustAllHttpsInitializer())
.request(); //发起请求
System.out.println(response.getCode()); //响应状态码
System.out.println(response.getRequestUrl());//最终发起请求的地址
if(response.getCode() == 200){
//请求成功
System.out.println(response.getContent()); //响应内容
}
}
}

评论列表( 1 )

126263_skygongle
skygongle 2018-07-12 17:59

感谢分享

你可以在登录后,发表评论

11_float_left_people 11_float_left_close