package com.shirley.aTestActuator.util;

import java.io.File;
import java.io.IOException;
import java.io.InterruptedIOException;
import java.net.URL;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;

import org.apache.http.HttpEntity;
import org.apache.http.NameValuePair;
import org.apache.http.client.HttpRequestRetryHandler;
import org.apache.http.client.ServiceUnavailableRetryStrategy;
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.conn.ssl.DefaultHostnameVerifier;
import org.apache.http.conn.util.PublicSuffixMatcher;
import org.apache.http.conn.util.PublicSuffixMatcherLoader;
import org.apache.http.entity.ContentType;
import org.apache.http.entity.StringEntity;
import org.apache.http.entity.mime.MultipartEntityBuilder;
import org.apache.http.entity.mime.content.FileBody;
import org.apache.http.entity.mime.content.StringBody;
import org.apache.http.impl.client.CloseableHttpClient;
import org.apache.http.impl.client.HttpClientBuilder;
import org.apache.http.impl.client.HttpClients;
import org.apache.http.impl.conn.PoolingHttpClientConnectionManager;
import org.apache.http.message.BasicNameValuePair;
import org.apache.http.protocol.HttpContext;
import org.apache.http.util.EntityUtils;

import com.shirley.aTestActuator.entity.ResponseContent;

public class HttpClientUtil {
	private static HttpClientUtil instance = null;

	private HttpClientUtil() {
	}

	public static HttpClientUtil getInstance() {
		if (null == instance) {
			instance = new HttpClientUtil();
		}
		return instance;
	}

	public static CloseableHttpClient getHttpClient(final int executionCount, int retryInterval) {
		ServiceUnavailableRetryStrategy serviceUnavailableRetryStrategy = new MyServiceUnavailableRetryStrategy.Builder()
				.executionCount(executionCount).retryInterval(retryInterval).build();
		return HttpClientBuilder.create().setRetryHandler(new HttpRequestRetryHandler() {
			@Override
			public boolean retryRequest(IOException e, int count, HttpContext contr) {
				if (count >= executionCount) {
					// Do not retry if over max retry count
					return false;
				}
				if (e instanceof InterruptedIOException) {
					// Timeout
					return true;
				}

				return true;
			}
		}).setServiceUnavailableRetryStrategy(serviceUnavailableRetryStrategy)
				.setConnectionManager(new PoolingHttpClientConnectionManager()).build();
	}

	/**
	 * 发送 post请求
	 * 
	 * @param httpUrl
	 *            地址
	 */
	public ResponseContent sendHttpPost(String httpUrl, Map<String, String> headers, RequestConfig requestConfig,
			int executionCount, int retryInterval) {
		HttpPost httpPost = new HttpPost(httpUrl);// 创建httpPost
		return sendHttpPost(httpPost, headers, requestConfig, executionCount, retryInterval);
	}

	/**
	 * 发送 post请求
	 * 
	 * @param httpUrl
	 *            地址
	 * @param params
	 *            参数(格式:key1=value1&key2=value2)
	 */
	public ResponseContent sendHttpPost(String httpUrl, String params, Map<String, String> headers,
			RequestConfig requestConfig, int executionCount, int retryInterval) {
		HttpPost httpPost = new HttpPost(httpUrl);// 创建httpPost
		try {
			// 设置参数
			StringEntity stringEntity = new StringEntity(params, "UTF-8");
			httpPost.setEntity(stringEntity);
			return sendHttpPost(httpPost, headers, requestConfig, executionCount, retryInterval);
		} catch (Exception e) {
			return null;
		}

	}

	/**
	 * 发送 post请求
	 * 
	 * @param httpUrl
	 *            地址
	 * @param maps
	 *            参数
	 */
	public ResponseContent sendHttpPost(String httpUrl, Map<String, String> maps, Map<String, String> headers,
			RequestConfig requestConfig, int executionCount, int retryInterval) {
		HttpPost httpPost = new HttpPost(httpUrl);// 创建httpPost
		// 创建参数队列
		List<NameValuePair> nameValuePairs = new ArrayList<NameValuePair>();
		if (null != maps && maps.size() > 0) {
			for (String key : maps.keySet()) {
				nameValuePairs.add(new BasicNameValuePair(key, maps.get(key)));
			}
		}
		try {
			httpPost.setEntity(new UrlEncodedFormEntity(nameValuePairs, "UTF-8"));
			return sendHttpPost(httpPost, headers, requestConfig, executionCount, retryInterval);
		} catch (Exception e) {
			return null;
		}
	}

	/**
	 * 发送 post请求（带文件）
	 * 
	 * @param httpUrl
	 *            地址
	 * @param maps
	 *            参数
	 * @param fileLists
	 *            附件
	 */
	public ResponseContent sendHttpPost(String httpUrl, Map<String, String> maps, List<File> fileLists,
			Map<String, String> headers, RequestConfig requestConfig, int executionCount, int retryInterval) {
		HttpPost httpPost = new HttpPost(httpUrl);// 创建httpPost
		MultipartEntityBuilder meBuilder = MultipartEntityBuilder.create();
		if (null != maps && maps.size() > 0) {
			for (String key : maps.keySet()) {
				meBuilder.addPart(key, new StringBody(maps.get(key), ContentType.TEXT_PLAIN));
			}
		}
		for (File file : fileLists) {
			FileBody fileBody = new FileBody(file);
			meBuilder.addPart("files", fileBody);
		}
		HttpEntity reqEntity = meBuilder.build();
		httpPost.setEntity(reqEntity);
		return sendHttpPost(httpPost, headers, requestConfig, executionCount, retryInterval);
	}

	/**
	 * 发送Post请求
	 * 
	 * @param httpPost
	 * @return
	 */
	private ResponseContent sendHttpPost(HttpPost httpPost, Map<String, String> headers, RequestConfig requestConfig,
			int executionCount, int retryInterval) {
		CloseableHttpClient httpClient = getHttpClient(executionCount, retryInterval);
		CloseableHttpResponse response = null;
		HttpEntity entity = null;
		ResponseContent responseContent = new ResponseContent();
		try {
			if (null == httpClient)
				// 创建默认的httpClient实例.
				httpClient = HttpClients.createDefault();
			httpPost.setConfig(requestConfig);
			if (null != headers && headers.size() > 0) {
				for (String key : headers.keySet()) {
					httpPost.addHeader(key, headers.get(key));
				}
			}

			// 执行请求
			response = httpClient.execute(httpPost);
			entity = response.getEntity();
			responseContent.setHeaders(response.getAllHeaders());
			responseContent.setResponseCode(response.getStatusLine().getStatusCode() + "");
			responseContent.setContent(EntityUtils.toString(entity, "UTF-8"));
		} catch (InterruptedIOException e) {
			responseContent.setContent("请求超时！");
		} catch (Exception e) {
			responseContent.setContent("请求报错！");
		} finally {
			try {
				// 关闭连接,释放资源
				if (null != response) {
					response.close();
				}
				if (null != httpClient) {
					httpClient.close();
				}
			} catch (IOException e) {
				e.printStackTrace();
			}
		}
		return responseContent;
	}

	/**
	 * 发送 get请求
	 * 
	 * @param httpUrl
	 */
	public ResponseContent sendHttpGet(String httpUrl, Map<String, String> headers, RequestConfig requestConfig,
			int executionCount, int retryInterval) {
		HttpGet httpGet = new HttpGet(httpUrl);
		return sendHttpGet(httpGet, headers, requestConfig, executionCount, retryInterval);

	}

	/**
	 * 发送 get请求Https
	 * 
	 * @param httpUrl
	 */
	public ResponseContent sendHttpsGet(String httpUrl, Map<String, String> headers, RequestConfig requestConfig,
			int executionCount, int retryInterval) {
		HttpGet httpGet = new HttpGet(httpUrl);// 创建get请求
		return sendHttpsGet(httpGet, headers, requestConfig, executionCount, retryInterval);
	}

	/**
	 * 发送Get请求
	 * 
	 * @param httpPost
	 * @return
	 */

	private ResponseContent sendHttpGet(HttpGet httpGet, Map<String, String> headers, RequestConfig requestConfig,
			int executionCount, int retryInterval) {
		CloseableHttpClient httpClient = getHttpClient(executionCount, retryInterval);
		CloseableHttpResponse response = null;
		HttpEntity entity = null;
		ResponseContent responseContent = new ResponseContent();
		try {
			if (null == httpClient)
				// 创建默认的httpClient实例.
				httpClient = HttpClients.createDefault();
			httpGet.setConfig(requestConfig);
			if (null != headers && headers.size() > 0) {
				for (String key : headers.keySet()) {
					httpGet.addHeader(key, headers.get(key));
				}
			}
			// 执行请求
			response = httpClient.execute(httpGet);
			entity = response.getEntity();
			responseContent.setHeaders(response.getAllHeaders());
			responseContent.setResponseCode(response.getStatusLine().getStatusCode() + "");
			responseContent.setContent(EntityUtils.toString(entity, "UTF-8"));
		} catch (InterruptedIOException e) {
			responseContent.setContent("请求超时！");
		} catch (Exception e) {
			responseContent.setContent("请求报错！");
		} finally {
			try {
				// 关闭连接,释放资源
				if (null != response) {
					response.close();
				}
				if (null != httpClient) {
					httpClient.close();
				}
			} catch (IOException e) {
				e.printStackTrace();
			}
		}
		return responseContent;
	}

	/**
	 * 发送Get请求Https
	 * 
	 * @param httpPost
	 * @return
	 */
	private ResponseContent sendHttpsGet(HttpGet httpGet, Map<String, String> headers, RequestConfig requestConfig,
			int executionCount, int retryInterval) {
		CloseableHttpClient httpClient = getHttpClient(executionCount, retryInterval);
		CloseableHttpResponse response = null;
		HttpEntity entity = null;
		ResponseContent responseContent = new ResponseContent();
		try {
			// 创建默认的httpClient实例.
			PublicSuffixMatcher publicSuffixMatcher = PublicSuffixMatcherLoader
					.load(new URL(httpGet.getURI().toString()));
			DefaultHostnameVerifier hostnameVerifier = new DefaultHostnameVerifier(publicSuffixMatcher);
			httpClient = HttpClients.custom().setSSLHostnameVerifier(hostnameVerifier).build();
			httpGet.setConfig(requestConfig);
			if (null != headers && headers.size() > 0) {
				for (String key : headers.keySet()) {
					httpGet.addHeader(key, headers.get(key));
				}
			}
			// 执行请求
			response = httpClient.execute(httpGet);
			entity = response.getEntity();
			responseContent.setHeaders(response.getAllHeaders());
			responseContent.setResponseCode(response.getStatusLine().getStatusCode() + "");
			responseContent.setContent(EntityUtils.toString(entity, "UTF-8"));
		} catch (InterruptedIOException e) {
			responseContent.setContent("请求超时！");
		} catch (Exception e) {
			responseContent.setContent("请求报错！");
		} finally {
			try {
				// 关闭连接,释放资源
				if (null != response) {
					response.close();
				}
				if (null != httpClient) {
					httpClient.close();
				}
			} catch (IOException e) {
				e.printStackTrace();
			}
		}
		return responseContent;
	}
}