Mercurial > hg > Members > sugi > MessagePack-java
diff src/main/java/org/msgpack/io/StreamInput.java @ 0:cb825acd883a
first commit
author | sugi |
---|---|
date | Sat, 18 Oct 2014 15:06:15 +0900 |
parents | |
children | 769ba8da0840 |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/src/main/java/org/msgpack/io/StreamInput.java Sat Oct 18 15:06:15 2014 +0900 @@ -0,0 +1,114 @@ +// +// MessagePack for Java +// +// Copyright (C) 2009 - 2013 FURUHASHI Sadayuki +// +// Licensed under the Apache License, Version 2.0 (the "License"); +// you may not use this file except in compliance with the License. +// You may obtain a copy of the License at +// +// http://www.apache.org/licenses/LICENSE-2.0 +// +// Unless required by applicable law or agreed to in writing, software +// distributed under the License is distributed on an "AS IS" BASIS, +// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +// See the License for the specific language governing permissions and +// limitations under the License. +// +package org.msgpack.io; + +import java.io.InputStream; +import java.nio.ByteBuffer; +import java.io.IOException; +import java.io.EOFException; + +public class StreamInput extends AbstractInput { + private final InputStream in; + + private byte[] castBuffer; + private ByteBuffer castByteBuffer; + private int filled; + + public StreamInput(InputStream in) { + this.in = in; + this.castBuffer = new byte[8]; + this.castByteBuffer = ByteBuffer.wrap(castBuffer); + this.filled = 0; + } + + public int read(byte[] b, int off, int len) throws IOException { + int remain = len; + while (remain > 0) { + int n = in.read(b, off, remain); + if (n <= 0) { + throw new EOFException(); + } + incrReadByteCount(n); + remain -= n; + off += n; + } + return len; + } + + public boolean tryRefer(BufferReferer ref, int size) throws IOException { + return false; + } + + public byte readByte() throws IOException { + int n = in.read(); + if (n < 0) { + throw new EOFException(); + } + incrReadOneByteCount(); + return (byte) n; + } + + public void advance() { + incrReadByteCount(filled); + filled = 0; + } + + private void require(int len) throws IOException { + while (filled < len) { + int n = in.read(castBuffer, filled, len - filled); + if (n < 0) { + throw new EOFException(); + } + filled += n; + } + } + + public byte getByte() throws IOException { + require(1); + return castBuffer[0]; + } + + public short getShort() throws IOException { + require(2); + return castByteBuffer.getShort(0); + } + + public int getInt() throws IOException { + require(4); + return castByteBuffer.getInt(0); + } + + public long getLong() throws IOException { + require(8); + return castByteBuffer.getLong(0); + } + + public float getFloat() throws IOException { + require(4); + return castByteBuffer.getFloat(0); + } + + public double getDouble() throws IOException { + require(8); + return castByteBuffer.getDouble(0); + } + + public void close() throws IOException { + in.close(); + } +}