nbbrd.io.http.ext.TeeInputStream Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of java-io-http Show documentation
Common IO utilities - http
The newest version!
package nbbrd.io.http.ext;

import nbbrd.io.Resource;

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

@lombok.AllArgsConstructor
final class TeeInputStream extends InputStream {

    @lombok.NonNull
    private final InputStream input;

    @lombok.NonNull
    private final OutputStream output;

    @Override
    public int read() throws IOException {
        int result = input.read();
        if (isNotEndOfStream(result)) {
            output.write(result);
        }
        return result;
    }

    @Override
    public int read(byte[] b) throws IOException {
        int result = input.read(b);
        if (isNotEndOfStream(result)) {
            output.write(b, 0, result);
        }
        return result;
    }

    @Override
    public int read(byte[] b, int off, int len) throws IOException {
        int result = input.read(b, off, len);
        if (isNotEndOfStream(result)) {
            output.write(b, off, result);
        }
        return result;
    }

    @Override
    public long skip(long n) throws IOException {
        return super.skip(n);
    }

    @Override
    public int available() throws IOException {
        return input.available();
    }

    @Override
    public void close() throws IOException {
        try {
            copyRemaining(input, output);
        } finally {
            Resource.closeBoth(input, output);
        }
    }

    @Override
    public synchronized void mark(int readlimit) {
    }

    @Override
    public synchronized void reset() throws IOException {
        throw new IOException("mark/reset not supported");
    }

    @Override
    public boolean markSupported() {
        return false;
    }

    private static void copyRemaining(InputStream source, OutputStream target) {
        byte[] buf = new byte[8192];
        int length;
        try {
            while ((length = source.read(buf)) > 0) {
                target.write(buf, 0, length);
            }
        } catch (IOException ex) {
            // do nothing
        }
    }

    private static boolean isNotEndOfStream(int result) {
        return result != -1;
    }
}