working on BufferedInputStreamWrapper
This commit is contained in:
parent
a05a3aabfa
commit
1aec5ea929
3 changed files with 80 additions and 3 deletions
|
@ -0,0 +1,77 @@
|
||||||
|
package org.capnproto;
|
||||||
|
|
||||||
|
import java.io.IOException;
|
||||||
|
import java.nio.ByteBuffer;
|
||||||
|
import java.nio.channels.ReadableByteChannel;
|
||||||
|
|
||||||
|
public final class BufferedInputStreamWrapper implements BufferedInputStream {
|
||||||
|
|
||||||
|
private final ReadableByteChannel inner;
|
||||||
|
private final ByteBuffer buf;
|
||||||
|
private int cap = 0;
|
||||||
|
|
||||||
|
public BufferedInputStreamWrapper(ReadableByteChannel chan) {
|
||||||
|
this.inner = chan;
|
||||||
|
this.buf = ByteBuffer.allocate(8192);
|
||||||
|
}
|
||||||
|
|
||||||
|
public final int read(ByteBuffer dst) throws IOException {
|
||||||
|
int numBytes = dst.remaining();
|
||||||
|
if (numBytes < cap - this.buf.position()) {
|
||||||
|
//# Serve from the current buffer.
|
||||||
|
ByteBuffer slice = this.buf.slice();
|
||||||
|
slice.limit(numBytes);
|
||||||
|
dst.put(slice);
|
||||||
|
this.buf.position(this.buf.position() + numBytes);
|
||||||
|
return numBytes;
|
||||||
|
} else {
|
||||||
|
//# Copy current available into destination.
|
||||||
|
ByteBuffer slice = this.buf.slice();
|
||||||
|
int fromFirstBuffer = cap - this.buf.position();
|
||||||
|
slice.limit(fromFirstBuffer);
|
||||||
|
dst.put(slice);
|
||||||
|
|
||||||
|
numBytes -= fromFirstBuffer;
|
||||||
|
if (numBytes <= this.buf.capacity()) {
|
||||||
|
//# Read the next buffer-full.
|
||||||
|
this.buf.rewind();
|
||||||
|
int n = readAtLeast(this.inner, this.buf, numBytes);
|
||||||
|
|
||||||
|
// ...
|
||||||
|
//ByteBuffer slice =
|
||||||
|
//dst.put(
|
||||||
|
|
||||||
|
this.cap = n;
|
||||||
|
this.buf.position(numBytes);
|
||||||
|
return fromFirstBuffer + numBytes;
|
||||||
|
} else {
|
||||||
|
//# Forward large read to the underlying stream.
|
||||||
|
}
|
||||||
|
}
|
||||||
|
throw new Error("unimplemented");
|
||||||
|
}
|
||||||
|
|
||||||
|
public final ByteBuffer getReadBuffer() {
|
||||||
|
return this.buf;
|
||||||
|
}
|
||||||
|
|
||||||
|
public final void close() throws IOException {
|
||||||
|
this.inner.close();
|
||||||
|
}
|
||||||
|
|
||||||
|
public final boolean isOpen() {
|
||||||
|
return this.inner.isOpen();
|
||||||
|
}
|
||||||
|
|
||||||
|
public static int readAtLeast(ReadableByteChannel reader, ByteBuffer buf, int minBytes) throws IOException {
|
||||||
|
int numRead = 0;
|
||||||
|
while (numRead < minBytes) {
|
||||||
|
int res = reader.read(buf);
|
||||||
|
if (res < 0) {
|
||||||
|
throw new Error("premature EOF");
|
||||||
|
}
|
||||||
|
numRead += res;
|
||||||
|
}
|
||||||
|
return numRead;
|
||||||
|
}
|
||||||
|
}
|
|
@ -6,8 +6,8 @@ import java.nio.channels.WritableByteChannel;
|
||||||
|
|
||||||
public final class BufferedOutputStreamWrapper implements BufferedOutputStream {
|
public final class BufferedOutputStreamWrapper implements BufferedOutputStream {
|
||||||
|
|
||||||
public final WritableByteChannel inner;
|
private final WritableByteChannel inner;
|
||||||
public final ByteBuffer buf;
|
private final ByteBuffer buf;
|
||||||
|
|
||||||
public BufferedOutputStreamWrapper(WritableByteChannel w) {
|
public BufferedOutputStreamWrapper(WritableByteChannel w) {
|
||||||
this.inner = w;
|
this.inner = w;
|
||||||
|
|
|
@ -15,7 +15,7 @@ public final class ByteChannelMessageReader {
|
||||||
return result;
|
return result;
|
||||||
}
|
}
|
||||||
|
|
||||||
static void fillBuffer(ByteBuffer buffer, ReadableByteChannel bc) throws IOException {
|
public static void fillBuffer(ByteBuffer buffer, ReadableByteChannel bc) throws IOException {
|
||||||
while(buffer.hasRemaining()) {
|
while(buffer.hasRemaining()) {
|
||||||
int r = bc.read(buffer);
|
int r = bc.read(buffer);
|
||||||
if (r < 0) {
|
if (r < 0) {
|
||||||
|
|
Loading…
Reference in a new issue