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 { 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); ByteArrayOutputStream out = new ByteArrayOutputStream(1024);
byte[] bytes = new byte[1024]; byte[] bytes = new byte[1024];
int pos; int pos;
while ((pos = in.read(bytes)) != -1) { while ((pos = in.read(bytes)) != -1) {
out.write(bytes, 0, pos); out.write(bytes, 0, pos);
} }
if (close) in.close();
return out; return out;
} }
@@ -1182,17 +1191,6 @@ public final class Utility {
return readStream(in).toByteArray(); 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 { public static byte[] readBytesThenClose(InputStream in) throws IOException {
return readStreamThenClose(in).toByteArray(); return readStreamThenClose(in).toByteArray();
} }