WrappedResponseInputStream.java

package com.github.dockerjava.jaxrs.util;

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

import javax.ws.rs.core.Response;

/**
 * This is a wrapper around {@link Response} that acts as a {@link InputStream}. When this {@link WrappedResponseInputStream} is closed it
 * closes the underlying {@link Response} object also to prevent blocking/hanging connections.
 *
 * @author Marcus Linke
 */
public class WrappedResponseInputStream extends InputStream {

    private Response response;

    private InputStream delegate;

    private boolean closed = false;

    public WrappedResponseInputStream(Response response) {
        this.response = response;
        this.delegate = response.readEntity(InputStream.class);
    }

    public int read() throws IOException {
        return delegate.read();
    }

    public int hashCode() {
        return delegate.hashCode();
    }

    public int read(byte[] b) throws IOException {
        return delegate.read(b);
    }

    public boolean equals(Object obj) {
        return delegate.equals(obj);
    }

    public int read(byte[] b, int off, int len) throws IOException {
        return delegate.read(b, off, len);
    }

    public long skip(long n) throws IOException {
        return delegate.skip(n);
    }

    public int available() throws IOException {
        return delegate.available();
    }

    public void close() throws IOException {
        if (closed) {
            return;
        }
        closed = true;
        delegate.close();
        response.close();
    }

    public void mark(int readlimit) {
        delegate.mark(readlimit);
    }

    public void reset() throws IOException {
        delegate.reset();
    }

    public boolean markSupported() {
        return delegate.markSupported();
    }

    public boolean isClosed() {
        return closed;
    }

}