com.quincy.sdk.helper.HttpClientHelper Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of quincy-sdk Show documentation
The newest version!
package com.quincy.sdk.helper;

import java.io.BufferedOutputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.io.PrintWriter;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.Map.Entry;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import org.apache.http.Header;
import org.apache.http.HttpEntity;
import org.apache.http.HttpResponse;
import org.apache.http.NameValuePair;
import org.apache.http.client.ClientProtocolException;
import org.apache.http.client.HttpResponseException;
import org.apache.http.client.entity.UrlEncodedFormEntity;
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.entity.StringEntity;
import org.apache.http.impl.client.CloseableHttpClient;
import org.apache.http.impl.client.HttpClientBuilder;
import org.springframework.http.MediaType;

public class HttpClientHelper {
//	private final static String ERR_MSG = "Abnormal HTTP Status Code: %s, URI: %s";

	public static String get(String url, Header[] headers) throws IOException {
		HttpGet httpGet = null;
		try {
			httpGet = new HttpGet(url);
			return getString(httpGet, headers);
		} finally {
			if(httpGet!=null)
				httpGet.abort();
		}
	}

	public static String post(String url, Header[] headers, List nameValuePairList) throws IOException {
		return post(url, headers, new UrlEncodedFormEntity(nameValuePairList, "UTF-8"));
	}

	public static String post(String url, Header[] headers, String body) throws IOException {
		return post(url, headers, new StringEntity(body));
	}

	public static String post(String url, Header[] headers, HttpEntity entity) throws IOException {
		HttpPost httpPost = null;
		try {
			httpPost = new HttpPost(url);
			httpPost.setEntity(entity);
			return getString(httpPost, headers);
		} finally {
			if(httpPost!=null)
				httpPost.abort();
		}
	}

	/*public static byte[] getBytes(String url, Header[] headers) throws IOException {
		HttpClient client = new HttpClient();
		HttpMethodBase method = null;
		method = new GetMethod(url);
		try {
			int statusCode = client.executeMethod(method);
			if(statusCode!=200)
				throw new HttpResponseException(statusCode, String.format(ERR_MSG, statusCode, url));
			byte[] buf = method.getResponseBody();
			return buf;
		} finally {
			if(method!=null)
				method.abort();
		}
	}*/

	public static byte[] getBytes(String url, Header[] headers) throws IOException {
		HttpGet httpGet = null;
		InputStream in = null;
		try {
			httpGet = new HttpGet(url);
			in = send(httpGet, headers);
			byte[] b = CommonHelper.input2bytes(in);
			return b;
		} finally {
			if(in!=null)
				in.close();
			if(httpGet!=null)
				httpGet.abort();
		}
	}

	public static void saveAsFile(String url, Header[] headers, String path) throws IOException {
		byte[] buf = getBytes(url, headers);
		OutputStream out = null;
		try {
			out = new BufferedOutputStream(new FileOutputStream(path));
			out.write(buf);
			out.flush();
		} finally {
			if(out!=null)
				out.close();
		}
	}

	private static String getString(HttpUriRequest httpUriRequest, Header[] headers) throws IOException {
		InputStream in = null;
		try {
			in = send(httpUriRequest, headers);
			String s = CommonHelper.input2String(in);
			return s;
		} finally {
			if(in!=null)
				in.close();
		}
	}

	private static InputStream send(HttpUriRequest httpUriRequest, Header[] headers) throws ClientProtocolException, IOException {
		if(headers!=null&&headers.length>0)
			httpUriRequest.setHeaders(headers);
		CloseableHttpClient httpClient = HttpClientBuilder.create().build();
		HttpResponse response = httpClient.execute(httpUriRequest);
		int statusCode = response.getStatusLine().getStatusCode();
		if(statusCode==200) {
			HttpEntity entity = response.getEntity();
			return entity.getContent();
		} else {
			InputStream in = null;
			String msg = null;
			try {
				in = response.getEntity().getContent();
				byte[] b = new byte[in.available()];
				in.read(b);
				msg = new String(b);
			} finally {
				in.close();
			}
			throw new HttpResponseException(statusCode, "\r\nRUL: "+httpUriRequest.getURI()+"\r\nMSG: "+msg);
		}
	}

	public final static String FLAG_URI = "URI";
	public final static String FLAG_URL = "URL";

	public static String getRequestURIOrURL(HttpServletRequest request, String type) {
		StringBuffer url = FLAG_URI.equalsIgnoreCase(type)?new StringBuffer(300).append(request.getRequestURI()):request.getRequestURL();
		String s = null;
		Map map = request.getParameterMap();
		if(map!=null&&map.size()>0) {
			Set> set = map.entrySet();
			url.append("?");
			for(Entry entry:set) {
				String[] values = entry.getValue();
				url.append(entry.getKey()).append("=").append((values!=null&&values.length>0)?values[0]:"").append("&");
			}
			s = url.substring(0, url.length()-1);
		} else
			s = url.toString();
		return s;
	}

	public static void output(HttpServletResponse response, String contentType, String contentTxt) throws IOException {
		PrintWriter out = null;
		try {
			response.setContentType(contentType);
			out = response.getWriter();
			out.println(contentTxt);
			out.flush();
		} finally {
			if(out!=null)
				out.close();
		}
	}

	public static void outputJson(HttpServletResponse response, String json) throws IOException {
		output(response, MediaType.APPLICATION_JSON_UTF8_VALUE, json);
	}

	public static void main(String[] args) throws IOException {
//		List nameValuePairList = new ArrayList(4);
//		nameValuePairList.add(new BasicNameValuePair("msg_signature", "5c45ff5e21c57e6ad56bac8758b79b1d9ac89fd3"));
//		nameValuePairList.add(new BasicNameValuePair("timestamp", "1409659589"));
//		nameValuePairList.add(new BasicNameValuePair("nonce", "263014780"));
//		nameValuePairList.add(new BasicNameValuePair("echostr", "P9nAzCzyDtyTWESHep1vC5X9xho/qYX3Zpb4yKa9SKld1DsH3Iyt3tP3zNdtp+4RPcs8TgAE7OaBO+FZXvnaqQ=="));
//		System.out.println(HttpClientHelper.post("http://localhost:8080/wechat/opencallbackmode", nameValuePairList));
//		System.out.println(HttpClientHelper.post("http://www.maqiangcgq.com/wechat/opencallbackmode", nameValuePairList));
//		System.out.println(HttpClientHelper.post("http://www.maqiangcgq.com/wechat/assistant", nameValuePairList));
//		Header[] headers = new Header[]{new BasicHeader("x-requested-with", "XMLHttpRequest")};
//		System.out.println(HttpClientHelper.post("http://localhost:8080/test", null, headers));
		HttpClientHelper.saveAsFile("http://resource.jlcedu.maqiangcgq.com/upload/exam/exam-s-zhaorq-mock2-en_us.jpg", null, "D:/tmp/exam-s-zhaorq-mock2-en_us.jpg");
	}
}