org.tarantool.ByteBufferStreams Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of connector Show documentation

Tarantool client for java

There is a newer version: 1.9.4

package org.tarantool;

import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.nio.ByteBuffer;

public class ByteBufferStreams {

    protected ByteBuffer buf;
    protected double resizeFactor = 1.1;
    protected OutputStream os = new OutputStream() {
        @Override
        public void write(int b) throws IOException {
            if(buf.remaining() < 1) {
                checkCapacity(buf.capacity() + 1);
            }
            buf.put((byte) b);
        }

        @Override
        public void write(byte[] b, int off, int len) throws IOException {
            if (len > buf.remaining()) {
                checkCapacity(buf.capacity() + (len - buf.remaining()));
            }
            buf.put(b, off, len);
        }
    };

    protected InputStream is = new InputStream() {
        @Override
        public int read() throws IOException {
            return 0xFF & buf.get();
        }

        @Override
        public int read(byte[] bytes, int off, int len)
                throws IOException {
            if (!buf.hasRemaining()) {
                return -1;
            }

            len = Math.min(len, buf.remaining());
            buf.get(bytes, off, len);
            return len;
        }

    };

    public ByteBufferStreams(ByteBuffer buf, double resizeFactor) {
        this.buf = buf;
        this.resizeFactor = resizeFactor;
    }


    public void checkCapacity(int size) {
        if (buf.capacity() < size) {
            ByteBuffer newBuf = ByteBuffer.allocate(size + (size)/2);
            buf.flip();
            newBuf.put(buf);
            buf = newBuf;
        }
    }





    public OutputStream asOutputStream() {
        return os;
    }

    public InputStream asInputStream() {
        return is;
    }

    public void clear() {
        buf.clear();
    }

    public ByteBuffer getBuf() {
        return buf;
    }
}