org.rx.io.HybridStream Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of rxlib Show documentation
A set of utilities for Java
There is a newer version: 3.0.0
package org.rx.io;

import lombok.Setter;
import lombok.extern.slf4j.Slf4j;

import java.io.*;

@Slf4j
public final class HybridStream extends IOStream implements Serializable {
    private static final long serialVersionUID = 2137331266386948293L;
    private static final int DEFAULT_MEMORY_SIZE = 1024 * 1024 * 4;
    private final int maxMemorySize;
    private MemoryStream memoryStream = new MemoryStream();
    private FileStream fileStream;
    @Setter
    private String name;

    private synchronized IOStream getStream() {
        if (fileStream != null) {
            return fileStream;
        }
        if (memoryStream.getLength() >= maxMemorySize) {
            log.debug("MemorySize: {}, switch to FileStream", maxMemorySize);
            memoryStream.copyTo(fileStream = new FileStream());
            fileStream.setPosition(memoryStream.getPosition());
            memoryStream = null;
            return fileStream;
        }
        return memoryStream;
    }

    @Override
    public String getName() {
        if (name == null) {
            return getStream().getName();
        }
        return name;
    }

    @Override
    public InputStream getReader() {
        return getStream().getReader();
    }

    @Override
    public OutputStream getWriter() {
        return getStream().getWriter();
    }

    @Override
    public boolean canSeek() {
        return getStream().canSeek();
    }

    @Override
    public long getPosition() {
        return getStream().getPosition();
    }

    @Override
    public void setPosition(long position) {
        getStream().setPosition(position);
    }

    @Override
    public long getLength() {
        return getStream().getLength();
    }

    public HybridStream() {
        this(DEFAULT_MEMORY_SIZE);
    }

    public HybridStream(int maxMemorySize) {
        super(null, null);
        this.maxMemorySize = maxMemorySize;
    }

    @Override
    protected void freeObjects() {
        getStream().close();
    }

    @Override
    public long available() {
        return getStream().available();
    }

    @Override
    public int read() {
        return getStream().read();
    }

    @Override
    public int read(byte[] buffer, int offset, int count) {
        return getStream().read(buffer, offset, count);
    }

    @Override
    public void write(int b) {
        getStream().write(b);
    }

    @Override
    public void write(byte[] buffer, int offset, int count) {
        getStream().write(buffer, offset, count);
    }

    @Override
    public void flush() {
        getStream().flush();
    }

    @Override
    public void copyTo(IOStream out) {
        getStream().copyTo(out);
    }

    @Override
    public void copyTo(OutputStream out) {
        getStream().copyTo(out);
    }
}