package cn.xxx.util;

import java.io.IOException;
import java.nio.charset.Charset;
import java.util.Map;

import org.apache.http.HttpEntity;
import org.apache.http.client.config.RequestConfig;
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.entity.ContentType;
import org.apache.http.entity.StringEntity;
import org.apache.http.impl.client.CloseableHttpClient;
import org.apache.http.impl.client.HttpClients;
import org.apache.http.util.EntityUtils;

/**
 * HTTP请求工具类.
 * 
 */
public class HttpUtil {

    /**
     * Get方式取得URL的内容.
     * 
     * @param url
     *            访问的网址
     * @return
     */
    public static String getUrlContent(String url) {

        CloseableHttpClient httpclient = HttpClients.createDefault();
        HttpGet httpGet = new HttpGet(url);
        CloseableHttpResponse response = null;
        try {

            // 执行请求
            response = httpclient.execute(httpGet);

            // 转换结果
            HttpEntity entity1 = response.getEntity();
            String html = EntityUtils.toString(entity1);

            // 消费掉内容
            EntityUtils.consume(entity1);
            return html;
        } catch (IOException e) {

            return "";
        } finally {
            if (response != null) {
                try {
                    response.close();
                } catch (IOException e) {
                }
            }
            try {
                httpclient.close();
            } catch (IOException e) {
            }
        }
    }

    /**
     * Post方式取得URL的内容，默认为"application/x-www-form-urlencoded"格式，charset为UTF-8.
     * 
     * @param url
     *            访问的网址
     * @param content
     *            提交的数据
     * @return
     */
    public static String postToUrl(String url, String content) {
        return postToUrl(url, content, "application/x-www-form-urlencoded",
                "UTF-8");
    }

    /**
     * json字符串形式请求
     * 
     * @param url
     * @param content
     * @return
     */
    public static String postForJson(String url, String content) {
        return postToUrl(url, content, "application/json", "UTF-8");
    }

    /**
     * Post方式取得URL的内容.
     * 
     * @param url
     *            访问的网址
     * @param content
     *            提交的数据
     * @param contentType
     *            数据类型
     * @param charset
     *            字符集
     * @return
     */
    public static String postToUrl(String url, String content,
            String contentType, String charset) {
        return postToUrl(url, content, contentType, charset, null);
    }

    /**
     * Post方式取得URL的内容.
     * 
     * @param url
     *            访问的网址
     * @param content
     *            提交的数据
     * @param contentType
     *            数据类型
     * @param charset
     *            字符集
     * @param headerMap
     *            要增加的header信息
     * @return
     */
    public static String postToUrl(String url, String content,
            String contentType, String charset, Map<String, String> headerMap) {
        CloseableHttpClient httpclient = HttpClients.createDefault();
        HttpPost httpPost = new HttpPost(url);

        // 设置内容
        ContentType type = ContentType.create(contentType,
                Charset.forName(charset));
        StringEntity reqEntity = new StringEntity(content, type);
        httpPost.setEntity(reqEntity);

        // Header
        if (headerMap != null && headerMap.size() > 0) {
            for (String key : headerMap.keySet()) {
                httpPost.addHeader(key, headerMap.get(key));
            }
        }

        // 这是超时
        RequestConfig requestConfig = RequestConfig.custom()
                .setSocketTimeout(30000).setConnectTimeout(60000).build();// 设置请求和传输超时时间
        httpPost.setConfig(requestConfig);

        CloseableHttpResponse response = null;
        try {
            // 执行请求
            response = httpclient.execute(httpPost);

            // 转换结果
            HttpEntity entity1 = response.getEntity();
            String html = EntityUtils.toString(entity1, charset);

            // 消费掉内容
            EntityUtils.consume(entity1);
            return html;
        } catch (IOException ex) {
            return "";
        } finally {
            if (response != null) {
                try {
                    response.close();
                } catch (IOException e) {
                }
            }
            try {
                httpclient.close();
            } catch (IOException e) {
            }
        }
    }

    public static void main(String[] args) {

        String html = postForJson("http://xxx.com/",
                "{\"frm\":\"frm\",\"ipcc\":\"ipcc\"}");
        System.out.println(html);

    }
}