com.github.becausetesting.http.HttpClientUtils Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of commons Show documentation
Show all versions of commons Show documentation
A common libraries used for testing framework.
/**
* Project Name:commons
* File Name:HttpClientUtils.java
* Package Name:com.github.becausetesting.http
* Date:Apr 17, 20166:29:02 PM
* Copyright (c) 2016, [email protected] All Rights Reserved.
*
*/
package com.github.becausetesting.http;
import java.io.IOException;
import java.io.InputStream;
import java.io.InterruptedIOException;
import java.net.URL;
import java.net.UnknownHostException;
import java.util.List;
import java.util.Map;
import javax.net.ssl.SSLException;
import org.apache.http.Consts;
import org.apache.http.HeaderElement;
import org.apache.http.HeaderElementIterator;
import org.apache.http.HeaderIterator;
import org.apache.http.HttpEntity;
import org.apache.http.HttpEntityEnclosingRequest;
import org.apache.http.HttpRequest;
import org.apache.http.HttpResponse;
import org.apache.http.NameValuePair;
import org.apache.http.ParseException;
import org.apache.http.ProtocolVersion;
import org.apache.http.StatusLine;
import org.apache.http.client.ClientProtocolException;
import org.apache.http.client.HttpRequestRetryHandler;
import org.apache.http.client.config.RequestConfig;
import org.apache.http.client.entity.UrlEncodedFormEntity;
import org.apache.http.client.methods.CloseableHttpResponse;
import org.apache.http.client.methods.HttpGet;
import org.apache.http.client.methods.HttpPost;
import org.apache.http.client.methods.HttpUriRequest;
import org.apache.http.client.protocol.HttpClientContext;
import org.apache.http.conn.ConnectTimeoutException;
import org.apache.http.impl.client.CloseableHttpClient;
import org.apache.http.impl.client.HttpClientBuilder;
import org.apache.http.message.BasicHeaderElementIterator;
import org.apache.http.protocol.HttpContext;
import org.apache.http.util.EntityUtils;
import org.apache.log4j.Logger;
import com.github.becausetesting.application.Environment;
/**
* ClassName:HttpClientUtils Function: TODO ADD FUNCTION. Reason: TODO ADD
* REASON. Date: Apr 17, 2016 6:29:02 PM
*
* @author Administrator
* @version 1.0.0
* @since JDK 1.8
*/
public class HttpClientUtils {
private static Logger logger = Logger.getLogger(HttpClientUtils.class);
private static CloseableHttpClient httpClient=null;
private static HttpResponse httpResponse=null;
//default http request settings
private static int DEFAULT_REQUEST_TIMEOUT=1000;
private static int DEFAULT_RETRY_TIMES=0;
enum Type {
GET, POST;
}
public String getRequest(String url) {
String responseContent = sendRequestContent(url, Type.GET,null);
return responseContent;
}
public String postRequest(String url,List postdata) {
String responseContent = sendRequestContent(url, Type.POST,postdata);
return responseContent;
}
private String sendRequestContent(String url, Type type,List postdata) {
String responseContent = null;
HttpResponse httpResponse = getHttpResponse(url, type,postdata);
HttpEntity httpEntity = httpResponse.getEntity();
try {
responseContent = EntityUtils.toString(httpEntity);
} catch (ParseException | IOException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
return responseContent;
}
public InputStream sendRequestStream(String url, Type type,List postdata) {
InputStream inputStream = null;
HttpResponse httpResponse = getHttpResponse(url, type,postdata);
HttpEntity responseEntity = httpResponse.getEntity();
try {
inputStream = responseEntity.getContent();
} catch (UnsupportedOperationException e) {
// TODO Auto-generated catch block
e.printStackTrace();
} catch (IOException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
return inputStream;
}
public static RequestConfig setDefaultHttpHeaders(){
RequestConfig requestConfig = RequestConfig.custom()
.setSocketTimeout(DEFAULT_REQUEST_TIMEOUT)
.setConnectTimeout(DEFAULT_REQUEST_TIMEOUT)
.build();
return requestConfig;
}
private HttpRequestRetryHandler setRetryTimes(){
HttpRequestRetryHandler httpRequestRetryHandler=new HttpRequestRetryHandler() {
public boolean retryRequest(IOException exception, int executionCount, HttpContext context) {
if (executionCount >DEFAULT_RETRY_TIMES) {
return false;
}
if (exception instanceof InterruptedIOException) {
// Timeout
return false;
}
if (exception instanceof UnknownHostException) {
// Unknown host
return false;
}
if (exception instanceof ConnectTimeoutException) {
// Connection refused
return false;
}
if (exception instanceof SSLException) {
// SSL handshake exception
return false;
}
HttpClientContext clientContext = HttpClientContext.adapt(context);
HttpRequest request = clientContext.getRequest();
boolean idempotent = !(request instanceof HttpEntityEnclosingRequest);
if (idempotent) {
// Retry if the request is considered idempotent
return true;
}
return false;
}
};
return httpRequestRetryHandler;
}
/**
* getCookies: refer http://blog.csdn.net/alibert/article/details/42788189
* @author [email protected]
* @since JDK 1.8
*/
public static void getCookies(){
HeaderElementIterator headerIterator = new BasicHeaderElementIterator(httpResponse.headerIterator("Set-Cookie"));
while(headerIterator.hasNext()){
HeaderElement cookiesElement = headerIterator.nextElement();
logger.info(cookiesElement.getName() + " = " + cookiesElement.getValue());
NameValuePair[] params = cookiesElement.getParameters();
for (int i = 0; i < params.length; i++) {
logger.info(" " + params[i]);
}
}
}
public void getHttpClient(){
if (httpClient != null) {
HttpRequestRetryHandler httpRequestRetryHandler = setRetryTimes();
httpClient = HttpClientBuilder.create().setRetryHandler(httpRequestRetryHandler).build();
}
}
private HttpResponse getHttpResponse(String url, Type type,List postdata) {
HttpGet httpGet = null;
HttpPost httpPost = null;
RequestConfig defaultHttpHeaders =setDefaultHttpHeaders();
try {
switch (type) {
case GET:
httpGet = new HttpGet(url);
httpGet.setConfig(defaultHttpHeaders);
httpResponse = httpClient.execute(httpGet);
break;
case POST:
httpPost = new HttpPost(url);
httpPost.setConfig(defaultHttpHeaders);
UrlEncodedFormEntity entity = new UrlEncodedFormEntity(postdata, Consts.UTF_8);
httpPost.setEntity(entity);
httpResponse = httpClient.execute(httpGet);
default:
httpGet = new HttpGet(url);
httpGet.setConfig(defaultHttpHeaders);
httpResponse = httpClient.execute(httpGet);
break;
}
StatusLine statusLine = httpResponse.getStatusLine();
/*
* ProtocolVersion httpProtocolVersion =
* statusLine.getProtocolVersion(); String reasonPhrase =
* statusLine.getReasonPhrase(); int statusCode =
* statusLine.getStatusCode();
*/
String allstatus = statusLine.toString();
logger.info("Reponse http status: " + allstatus);
} catch (ClientProtocolException e) {
// TODO Auto-generated catch block
e.printStackTrace();
} catch (IOException e) {
// TODO Auto-generated catch block
e.printStackTrace();
} finally {
try {
httpClient.close();
} catch (IOException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
}
return httpResponse;
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy