当前位置: 首页 > 工具软件 > httpkit > 使用案例 >

扩展JFinal的HttpKit

司空通
2023-12-01

对HttpKit做了些扩展,增加了PUT、DELETE方法,可传入自定义SSLSocketFactory,代码如下:

public class HttpKit {

	private HttpKit() {
	}

	/**
	 * https 域名校验
	 */
	private class TrustAnyHostnameVerifier implements HostnameVerifier {
		public boolean verify(String hostname, SSLSession session) {
			return true;
		}
	}

	/**
	 * https 证书管理
	 */
	private class TrustAnyTrustManager implements X509TrustManager {
		public X509Certificate[] getAcceptedIssuers() {
			return null;
		}
		public void checkClientTrusted(X509Certificate[] chain, String authType) throws CertificateException {
		}
		public void checkServerTrusted(X509Certificate[] chain, String authType) throws CertificateException {
		}
	}

	private static final String GET = "GET";
	private static final String POST = "POST";
	private static final String PUT = "PUT";
	private static final String DELETE = "DELETE";
	private static final String CHARSET = "UTF-8";

	private static final TrustAnyHostnameVerifier trustAnyHostnameVerifier = new HttpKit().new TrustAnyHostnameVerifier();

	private static SSLSocketFactory initSSLSocketFactory() {
		try {
			TrustManager[] tm = {new HttpKit().new TrustAnyTrustManager()};
			SSLContext sslContext = SSLContext.getInstance("TLS", "SunJSSE");
			sslContext.init(null, tm, new java.security.SecureRandom());
			return sslContext.getSocketFactory();
		} catch (Exception e) {
			throw new RuntimeException(e);
		}
	}

	private static HttpURLConnection getHttpConnection(String url, SSLSocketFactory sslSocketFactory, String method, Map<String, String> headers) throws IOException, NoSuchAlgorithmException, NoSuchProviderException, KeyManagementException {
		URL _url = new URL(url);
		HttpURLConnection conn = (HttpURLConnection) _url.openConnection();
		if (conn instanceof HttpsURLConnection) {
			if (sslSocketFactory != null) {
				((HttpsURLConnection) conn).setSSLSocketFactory(sslSocketFactory);
			} else {
				((HttpsURLConnection) conn).setSSLSocketFactory(initSSLSocketFactory());
			}
			((HttpsURLConnection) conn).setHostnameVerifier(trustAnyHostnameVerifier);
		}

		conn.setRequestMethod(method);
		conn.setDoOutput(true);
		conn.setDoInput(true);

		conn.setConnectTimeout(19000);
		conn.setReadTimeout(19000);

		conn.setRequestProperty("Content-Type", "application/x-www-form-urlencoded");
		conn.setRequestProperty("User-Agent", "Mozilla/5.0 (Windows NT 6.3; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/33.0.1750.146 Safari/537.36");

		if (headers != null && !headers.isEmpty())
			for (Entry<String, String> entry : headers.entrySet())
				conn.setRequestProperty(entry.getKey(), entry.getValue());

		return conn;
	}

	private static String read(String url, SSLSocketFactory sslSocketFactory, String method, Map<String, String> queryParas, Map<String, String> headers) {
		HttpURLConnection conn = null;
		try {
			conn = getHttpConnection(buildUrlWithQueryString(url, queryParas), sslSocketFactory, method, headers);
			conn.connect();
			return readResponseString(conn);
		} catch (Exception e) {
			throw new RuntimeException(e);
		} finally {
			if (conn != null) {
				conn.disconnect();
			}
		}
	}

	/**
	 * Send GET request
	 */
	public static String get(String url, SSLSocketFactory sslSocketFactory, Map<String, String> queryParas, Map<String, String> headers) {
		return read(url, sslSocketFactory, GET, queryParas, headers);
	}

	public static String get(String url, Map<String, String> queryParas, Map<String, String> headers) {
		return get(url, null, queryParas, headers);
	}

	public static String get(String url, Map<String, String> queryParas) {
		return get(url, queryParas, null);
	}

	public static String get(String url) {
		return get(url, null);
	}

	public static byte[] getBytes(String url, SSLSocketFactory sslSocketFactory, Map<String, String> queryParas, Map<String, String> headers) {
		HttpURLConnection conn = null;
		try {
			conn = getHttpConnection(buildUrlWithQueryString(url, queryParas), sslSocketFactory, GET, headers);
			conn.connect();
			return readResponseByte(conn);
		} catch (Exception e) {
			throw new RuntimeException(e);
		} finally {
			if (conn != null) {
				conn.disconnect();
			}
		}
	}

	public static byte[] getBytes(String url, Map<String, String> queryParas) {
		return getBytes(url, null, queryParas, null);
	}

	public static byte[] getBytes(String url) {
		return getBytes(url, null);
	}

	/**
	 * Send DELETE request
	 */
	public static String delete(String url, SSLSocketFactory sslSocketFactory, Map<String, String> queryParas, Map<String, String> headers) {
		return read(url, sslSocketFactory, DELETE, queryParas, headers);
	}

	public static String delete(String url, Map<String, String> queryParas) {
		return delete(url, null, queryParas, null);
	}

	public static String delete(String url) {
		return delete(url, null, null, null);
	}

	private static String write(String url, SSLSocketFactory sslSocketFactory, String method, Map<String, String> queryParas, String data, Map<String, String> headers) {
		HttpURLConnection conn = null;
		try {
			conn = getHttpConnection(buildUrlWithQueryString(url, queryParas), sslSocketFactory, method, headers);
			conn.connect();

			OutputStream out = conn.getOutputStream();
			out.write(data.getBytes(CHARSET));
			out.flush();
			out.close();

			return readResponseString(conn);
		} catch (Exception e) {
			throw new RuntimeException(e);
		} finally {
			if (conn != null) {
				conn.disconnect();
			}
		}
	}

	/**
	 * Send POST request
	 */
	public static String post(String url, SSLSocketFactory sslSocketFactory, Map<String, String> queryParas, String data, Map<String, String> headers) {
		return write(url, sslSocketFactory, POST, queryParas, data, headers);
	}

	public static String post(String url, Map<String, String> queryParas, String data, Map<String, String> headers) {
		return post(url, null, queryParas, data, headers);
	}

	public static String post(String url, Map<String, String> queryParas, String data) {
		return post(url, queryParas, data, null);
	}

	public static String post(String url, String data, Map<String, String> headers) {
		return post(url, null, data, headers);
	}

	public static String post(String url, String data) {
		return post(url, null, data, null);
	}

	/**
	 * Send PUT request
	 */
	public static String put(String url, SSLSocketFactory sslSocketFactory, Map<String, String> queryParas, String data, Map<String, String> headers) {
		return write(url, sslSocketFactory, PUT, queryParas, data, headers);
	}

	public static String put(String url, Map<String, String> queryParas, String data, Map<String, String> headers) {
		return put(url, null, queryParas, data, headers);
	}

	public static String put(String url, Map<String, String> queryParas, String data) {
		return put(url, queryParas, data, null);
	}

	public static String put(String url, String data, Map<String, String> headers) {
		return put(url, null, data, headers);
	}

	public static String put(String url, String data) {
		return put(url, null, data, null);
	}

	private static String readResponseString(HttpURLConnection conn) {
		StringBuilder sb = new StringBuilder();
		InputStream inputStream = null;
		try {
			inputStream = conn.getInputStream();
			BufferedReader reader = new BufferedReader(new InputStreamReader(inputStream, CHARSET));
			String line = null;
			while ((line = reader.readLine()) != null) {
				sb.append(line).append("\n");
			}
			return sb.toString();
		} catch (Exception e) {
			throw new RuntimeException(e);
		} finally {
			if (inputStream != null) {
				try {
					inputStream.close();
				} catch (IOException e) {
					e.printStackTrace();
				}
			}
		}
	}

	private static byte[] readResponseByte(HttpURLConnection conn) {
		InputStream inputStream = null;
		ByteArrayOutputStream outStream = null;
		try {
			inputStream = conn.getInputStream();
			outStream = new ByteArrayOutputStream();
			byte[] buffer = new byte[1024];
			int len = 0;
			while ((len = inputStream.read(buffer)) != -1) {
				outStream.write(buffer, 0, len);
				outStream.flush();
			}
			return outStream.toByteArray();
		} catch (Exception e) {
			throw new RuntimeException(e);
		} finally {
			if (inputStream != null) {
				try {
					inputStream.close();
				} catch (IOException e) {
					e.printStackTrace();
				}
			}
			if (outStream != null) {
				try {
					outStream.close();
				} catch (IOException e) {
					e.printStackTrace();
				}
			}
		}
	}

	/**
	 * Build queryString of the url
	 */
	private static String buildUrlWithQueryString(String url, Map<String, String> queryParas) {
		if (queryParas == null || queryParas.isEmpty())
			return url;

		StringBuilder sb = new StringBuilder(url);
		boolean isFirst;
		if (url.indexOf("?") == -1) {
			isFirst = true;
			sb.append("?");
		} else {
			isFirst = false;
		}

		for (Entry<String, String> entry : queryParas.entrySet()) {
			if (isFirst)
				isFirst = false;
			else
				sb.append("&");

			String key = entry.getKey();
			String value = entry.getValue();
			if (StrKit.notBlank(value))
				try {
					value = URLEncoder.encode(value, CHARSET);
				} catch (UnsupportedEncodingException e) {
					throw new RuntimeException(e);
				}
			sb.append(key).append("=").append(value);
		}
		return sb.toString();
	}

	public static String readIncommingRequestData(HttpServletRequest request) {
		BufferedReader br = null;
		try {
			StringBuilder result = new StringBuilder();
			br = request.getReader();
			for (String line = null; (line = br.readLine()) != null;) {
				result.append(line).append("\n");
			}

			return result.toString();
		} catch (IOException e) {
			throw new RuntimeException(e);
		} finally {
			if (br != null)
				try {
					br.close();
				} catch (IOException e) {
					e.printStackTrace();
				}
		}
	}

}


转载于:https://my.oschina.net/u/203952/blog/499563

 类似资料: