lxy/redkale
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Browse Source
This commit is contained in:
Redkale
2018-05-30 19:47:21 +08:00
parent e79fdd887c
commit 7352d5b8ed
1 changed files with 35 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

67
src/org/redkale/util/Utility.java
View File

@@ -1676,41 +1676,44 @@ public final class Utility {
public static ByteArrayOutputStream remoteHttpContent(SSLContext ctx, String method, String url, int timeout, Map<String, String> headers, String body) throws IOException { public static ByteArrayOutputStream remoteHttpContent(SSLContext ctx, String method, String url, int timeout, Map<String, String> headers, String body) throws IOException {
HttpURLConnection conn = (HttpURLConnection) new URL(url).openConnection(); HttpURLConnection conn = (HttpURLConnection) new URL(url).openConnection();
conn.setConnectTimeout(timeout > 0 ? timeout : 3000); try {
conn.setReadTimeout(timeout > 0 ? timeout : 3000); conn.setConnectTimeout(timeout > 0 ? timeout : 3000);
if (conn instanceof HttpsURLConnection) { conn.setReadTimeout(timeout > 0 ? timeout : 3000);
HttpsURLConnection httpsconn = ((HttpsURLConnection) conn); if (conn instanceof HttpsURLConnection) {
httpsconn.setSSLSocketFactory((ctx == null ? DEFAULTSSL_CONTEXT : ctx).getSocketFactory()); HttpsURLConnection httpsconn = ((HttpsURLConnection) conn);
httpsconn.setHostnameVerifier(defaultVerifier); httpsconn.setSSLSocketFactory((ctx == null ? DEFAULTSSL_CONTEXT : ctx).getSocketFactory());
} httpsconn.setHostnameVerifier(defaultVerifier);
conn.setRequestMethod(method);
if (headers != null) {
for (Map.Entry<String, String> en : headers.entrySet()) { //不用forEach是为了兼容JDK 6
conn.setRequestProperty(en.getKey(), en.getValue());
} }
} conn.setRequestMethod(method);
if (body != null) { if (headers != null) {
conn.setDoInput(true); for (Map.Entry<String, String> en : headers.entrySet()) { //不用forEach是为了兼容JDK 6
conn.setDoOutput(true); conn.setRequestProperty(en.getKey(), en.getValue());
conn.getOutputStream().write(body.getBytes(UTF_8)); }
} }
conn.connect(); if (body != null) {
int rs = conn.getResponseCode(); conn.setDoInput(true);
if (rs == 301 || rs == 302) { conn.setDoOutput(true);
String newurl = conn.getHeaderField("Location"); conn.getOutputStream().write(body.getBytes(UTF_8));
}
conn.connect();
int rs = conn.getResponseCode();
if (rs == 301 || rs == 302) {
String newurl = conn.getHeaderField("Location");
conn.disconnect();
return remoteHttpContent(ctx, method, newurl, timeout, headers, body);
}
InputStream in = (rs < 400 || rs == 404) && rs != 405 ? conn.getInputStream() : conn.getErrorStream();
if ("gzip".equalsIgnoreCase(conn.getContentEncoding())) in = new GZIPInputStream(in);
ByteArrayOutputStream out = new ByteArrayOutputStream(1024);
byte[] bytes = new byte[1024];
int pos;
while ((pos = in.read(bytes)) != -1) {
out.write(bytes, 0, pos);
}
return out;
} finally {
conn.disconnect(); conn.disconnect();
return remoteHttpContent(ctx, method, newurl, timeout, headers, body);
} }
InputStream in = (rs < 400 || rs == 404) && rs != 405 ? conn.getInputStream() : conn.getErrorStream();
if ("gzip".equalsIgnoreCase(conn.getContentEncoding())) in = new GZIPInputStream(in);
ByteArrayOutputStream out = new ByteArrayOutputStream(1024);
byte[] bytes = new byte[1024];
int pos;
while ((pos = in.read(bytes)) != -1) {
out.write(bytes, 0, pos);
}
conn.disconnect();
return out;
} }
public static String read(InputStream in) throws IOException { public static String read(InputStream in) throws IOException {