blob: 4d55e2f175c9a771a387f487117dde3db43f7415 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
|
package com.koushikdutta.async.util;
import java.io.ByteArrayOutputStream;
import java.io.Closeable;
import java.io.DataInputStream;
import java.io.DataOutputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.nio.ByteBuffer;
import java.nio.channels.Channels;
import java.nio.channels.ReadableByteChannel;
import java.nio.channels.WritableByteChannel;
public class StreamUtility {
public static void fastChannelCopy(final ReadableByteChannel src, final WritableByteChannel dest) throws IOException {
final ByteBuffer buffer = ByteBuffer.allocateDirect(16 * 1024);
while (src.read(buffer) != -1) {
// prepare the buffer to be drained
buffer.flip();
// write to the channel, may block
dest.write(buffer);
// If partial transfer, shift remainder down
// If buffer is empty, same as doing recycle()
buffer.compact();
}
// EOF will leave buffer in fill state
buffer.flip();
// make sure the buffer is fully drained.
while (buffer.hasRemaining()) {
dest.write(buffer);
}
}
public static void copyStream(InputStream input, OutputStream output) throws IOException
{
final ReadableByteChannel inputChannel = Channels.newChannel(input);
final WritableByteChannel outputChannel = Channels.newChannel(output);
// copy the channels
fastChannelCopy(inputChannel, outputChannel);
}
public static byte[] readToEndAsArray(InputStream input) throws IOException
{
DataInputStream dis = new DataInputStream(input);
byte[] stuff = new byte[1024];
ByteArrayOutputStream buff = new ByteArrayOutputStream();
int read = 0;
while ((read = dis.read(stuff)) != -1)
{
buff.write(stuff, 0, read);
}
dis.close();
return buff.toByteArray();
}
public static String readToEnd(InputStream input) throws IOException
{
return new String(readToEndAsArray(input));
}
static public String readFile(String filename) throws IOException {
return readFile(new File(filename));
}
static public String readFile(File file) throws IOException {
byte[] buffer = new byte[(int) file.length()];
DataInputStream input = null;
try {
input = new DataInputStream(new FileInputStream(file));
input.readFully(buffer);
} finally {
closeQuietly(input);
}
return new String(buffer);
}
public static void writeFile(File file, String string) throws IOException {
file.getParentFile().mkdirs();
DataOutputStream dout = new DataOutputStream(new FileOutputStream(file));
dout.write(string.getBytes());
dout.close();
}
public static void writeFile(String file, String string) throws IOException {
writeFile(new File(file), string);
}
public static void closeQuietly(Closeable... closeables) {
if (closeables == null)
return;
for (Closeable closeable : closeables) {
if (closeable != null) {
try {
closeable.close();
} catch (IOException e) {
// http://stackoverflow.com/a/156525/9636
}
}
}
}
public static void eat(InputStream input) throws IOException {
byte[] stuff = new byte[1024];
while (input.read(stuff) != -1);
}
}
|
