-
Notifications
You must be signed in to change notification settings - Fork 1
/
Pipe.java
80 lines (65 loc) · 1.96 KB
/
Pipe.java
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
package com.demia.io;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.io.PipedInputStream;
import java.io.PipedOutputStream;
import java.util.concurrent.CountDownLatch;
/**
* Manages a unidirectional pipe of data flowing from an OutputStream to an InputStream
*
* <p> A pipe consists of a writer and a reader.
* Once some bytes are written to the output stream they can be read
* from input stream in the order in which they were written.
*
* <p> This pipe implementations will buffer up to 1024 bytes by default, however one can specify
* a custom buffer size as well </p>
*
*
* @author Ranjit Aneesh
*
*/
public abstract class Pipe {
PipedOutputStream out;
PipedInputStream in;
Thread readerThread;
Thread writerThread;
CountDownLatch awaitReaderToDie = new CountDownLatch(1);
public Pipe() {
this(1024);
}
public Pipe(int bufferSize) {
in = new PipedInputStream();
try {
out = new PipedOutputStream(in);
} catch (IOException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
}
public void start() {
readerThread = new Thread(new Runnable() {
@Override
public void run() {
reader(in);
awaitReaderToDie.countDown();
}
});
writerThread = new Thread(new Runnable() {
@Override
public void run() {
writer(out);
try {
awaitReaderToDie.await();
} catch (InterruptedException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
}
});
readerThread.start();
writerThread.start();
}
protected abstract void writer(OutputStream out);
protected abstract void reader(InputStream in);
}