This commit is contained in:
Redkale
2017-05-11 13:58:33 +08:00
parent 5e2be5e926
commit fa2513d934

View File

@@ -1169,12 +1169,21 @@ public final class Utility {
}
public static ByteArrayOutputStream readStream(InputStream in) throws IOException {
return readStream(in, false);
}
public static ByteArrayOutputStream readStreamThenClose(InputStream in) throws IOException {
return readStream(in, true);
}
private static ByteArrayOutputStream readStream(InputStream in, boolean close) throws IOException {
ByteArrayOutputStream out = new ByteArrayOutputStream(1024);
byte[] bytes = new byte[1024];
int pos;
while ((pos = in.read(bytes)) != -1) {
out.write(bytes, 0, pos);
}
if (close) in.close();
return out;
}
@@ -1182,17 +1191,6 @@ public final class Utility {
return readStream(in).toByteArray();
}
public static ByteArrayOutputStream readStreamThenClose(InputStream in) throws IOException {
ByteArrayOutputStream out = new ByteArrayOutputStream(1024);
byte[] bytes = new byte[1024];
int pos;
while ((pos = in.read(bytes)) != -1) {
out.write(bytes, 0, pos);
}
in.close();
return out;
}
public static byte[] readBytesThenClose(InputStream in) throws IOException {
return readStreamThenClose(in).toByteArray();
}