org.apache.nifi.stream.io

Class LimitingInputStream

java.lang.Object

java.io.InputStream

org.apache.nifi.stream.io.LimitingInputStream

All Implemented Interfaces:

Closeable, AutoCloseable

public class LimitingInputStream
extends InputStream

Field Summary

Fields

Modifier and Type	Field and Description
private long	bytesRead
private InputStream	in
private long	limit
private boolean	limitReached
private long	markOffset

Constructor Summary

Constructors

Constructor and Description
LimitingInputStream(InputStream in, long limit) Constructs a limited input stream whereby if the limit is reached all subsequent calls to read will return a -1 and hasLimitReached() will indicate true.

Method Summary

Modifier and Type	Method and Description
int	available()
void	close()
long	getLimit()
boolean	hasReachedLimit()
void	mark(int readlimit)
private int	markLimitReached()
boolean	markSupported()
int	read()
int	read(byte[] b)
int	read(byte[] b, int off, int len)
void	reset()
long	skip(long n)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Field Detail

in

private final InputStream in

limit

private final long limit

bytesRead

private long bytesRead

limitReached

private volatile boolean limitReached

markOffset

private long markOffset

Constructor Detail

LimitingInputStream

public LimitingInputStream(InputStream in,
                           long limit)

Constructs a limited input stream whereby if the limit is reached all subsequent calls to read will return a -1 and hasLimitReached() will indicate true. The limit is inclusive so if all 100 bytes of a 100 byte stream are read it will be true, otherwise false.

Parameters:

in - the underlying input stream

limit - maximum length of bytes to read from underlying input stream

Method Detail

hasReachedLimit

public boolean hasReachedLimit()
                        throws IOException

Throws:

IOException

markLimitReached

private int markLimitReached()

read

public int read()
         throws IOException

Specified by:

read in class InputStream

Throws:

IOException

read

public int read(byte[] b)
         throws IOException

Overrides:

read in class InputStream

Throws:

IOException

read

public int read(byte[] b,
                int off,
                int len)
         throws IOException

Overrides:

read in class InputStream

Throws:

IOException

skip

public long skip(long n)
          throws IOException

Overrides:

skip in class InputStream

Throws:

IOException

available

public int available()
              throws IOException

Overrides:

available in class InputStream

Throws:

IOException

close

public void close()
           throws IOException

Specified by:

close in interface Closeable

Specified by:

close in interface AutoCloseable

Overrides:

close in class InputStream

Throws:

IOException

mark

public void mark(int readlimit)

Overrides:

mark in class InputStream

markSupported

public boolean markSupported()

Overrides:

markSupported in class InputStream

reset

public void reset()
           throws IOException

Overrides:

reset in class InputStream

Throws:

IOException

getLimit

public long getLimit()