package org.apache.commons.compress.compressors.zstandard;
import java.io.IOException;
import java.io.InputStream;
import com.github.luben.zstd.ZstdInputStream;
import org.apache.commons.compress.compressors.CompressorInputStream;
import org.apache.commons.compress.utils.CountingInputStream;
import org.apache.commons.compress.utils.IOUtils;
import org.apache.commons.compress.utils.InputStreamStatistics;
public class ZstdCompressorInputStream extends CompressorInputStream
implements InputStreamStatistics {
private final CountingInputStream countingStream;
private final ZstdInputStream decIS;
public ZstdCompressorInputStream(final InputStream in) throws IOException {
this.decIS = new ZstdInputStream(countingStream = new CountingInputStream(in));
}
@Override
public int available() throws IOException {
return decIS.available();
}
@Override
public void close() throws IOException {
decIS.close();
}
@Override
public int read(final byte[] b) throws IOException {
return read(b, 0, b.length);
}
@Override
public long skip(final long n) throws IOException {
return IOUtils.skip(decIS, n);
}
@Override
public synchronized void mark(final int readlimit) {
decIS.mark(readlimit);
}
@Override
public boolean markSupported() {
return decIS.markSupported();
}
@Override
public int read() throws IOException {
final int ret = decIS.read();
count(ret == -1 ? 0 : 1);
return ret;
}
@Override
public int read(final byte[] buf, final int off, final int len) throws IOException {
if (len == 0) {
return 0;
}
final int ret = decIS.read(buf, off, len);
count(ret);
return ret;
}
@Override
public String toString() {
return decIS.toString();
}
@Override
public synchronized void reset() throws IOException {
decIS.reset();
}
@Override
public long getCompressedCount() {
return countingStream.getBytesRead();
}
}