com.qcloud.cos.model.ByteRangeCapturingInputStream Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of cos_api Show documentation
java sdk for qcloud cos
There is a newer version: 5.6.238.2
package com.qcloud.cos.model;

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

import com.qcloud.cos.internal.SdkFilterInputStream;

/**
 * FilterInputStream that captures a range of bytes from a wrapped input stream.
 */
public class ByteRangeCapturingInputStream extends SdkFilterInputStream {
    private final long startingPosition;
    private final long endingPosition;

    private long streamPosition;
    private int blockPosition = 0;
    private final byte[] block;

    private long markedStreamPosition;
    private int markedBlockPosition;

    public ByteRangeCapturingInputStream(InputStream in, long startingPosition,
            long endingPosition) {
        super(in);

        if (startingPosition >= endingPosition) {
            throw new IllegalArgumentException(
                    "Invalid byte range specified: the starting position must be less than the ending position");
        }

        this.startingPosition = startingPosition;
        this.endingPosition = endingPosition;
        int blockSize = (int) (endingPosition - startingPosition);
        block = new byte[blockSize];
    }

    public byte[] getBlock() {
        return block;
    }

    @Override
    public int read() throws IOException {
        int data = super.read();
        if (data == -1)
            return -1;

        if (streamPosition >= startingPosition && streamPosition <= endingPosition) {
            block[blockPosition++] = (byte) data;
        }

        streamPosition++;
        return data;
    }

    @Override
    public synchronized void mark(int readlimit) {
        super.mark(readlimit);
        if (markSupported()) {
            markedStreamPosition = streamPosition;
            markedBlockPosition = blockPosition;
        }
    }

    @Override
    public synchronized void reset() throws IOException {
        super.reset();
        if (markSupported()) {
            streamPosition = markedStreamPosition;
            blockPosition = markedBlockPosition;
        }
    }

    @Override
    public int read(byte[] b, int off, int len) throws IOException {
        int bytesRead = super.read(b, off, len);
        if (bytesRead == -1)
            return -1;

        if (streamPosition + bytesRead >= startingPosition && streamPosition <= endingPosition) {
            for (int i = 0; i < bytesRead; i++) {
                if (streamPosition + i >= startingPosition && streamPosition + i < endingPosition) {
                    block[blockPosition++] = b[off + i];
                }
            }
        }

        streamPosition += bytesRead;
        return bytesRead;
    }
}