com.helger.commons.io.stream.NonBlockingBufferedReader Maven / Gradle / Ivy
Show all versions of ph-commons Show documentation
/*
* Copyright (C) 2014-2022 Philip Helger (www.helger.com)
* philip[at]helger[dot]com
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.helger.commons.io.stream;
import java.io.IOException;
import java.io.Reader;
import java.io.UncheckedIOException;
import java.util.Iterator;
import java.util.NoSuchElementException;
import java.util.Spliterator;
import java.util.Spliterators;
import java.util.stream.Stream;
import java.util.stream.StreamSupport;
import javax.annotation.Nonnegative;
import javax.annotation.Nonnull;
import javax.annotation.Nullable;
import com.helger.commons.ValueEnforcer;
import com.helger.commons.string.StringHelper;
/**
* A non-synchronized copy of the class {@link java.io.BufferedReader}.
*
* @author Philip Helger
* @see java.io.BufferedReader
*/
public class NonBlockingBufferedReader extends Reader
{
private static final int INVALIDATED = -2;
private static final int UNMARKED = -1;
private static final int DEFAULT_CHAR_BUFFER_SIZE = 8192;
private static final int DEFAULT_EXPECTED_LINE_LENGTH = 80;
private Reader m_aReader;
private char [] m_aBuf;
private int m_nChars;
private int m_nNextCharIndex;
private int m_nMarkedChar = UNMARKED;
/** Valid only when markedChar > 0 */
private int m_nReadAheadLimit = 0;
/** If the next character is a line feed, skip it */
private boolean m_bSkipLF = false;
/** The skipLF flag when the mark was set */
private boolean m_bMarkedSkipLF = false;
/**
* Creates a buffering character-input stream that uses a default-sized input
* buffer.
*
* @param aReader
* A Reader
*/
public NonBlockingBufferedReader (@Nonnull final Reader aReader)
{
this (aReader, DEFAULT_CHAR_BUFFER_SIZE);
}
/**
* Creates a buffering character-input stream that uses an input buffer of the
* specified size.
*
* @param aReader
* A Reader
* @param nSize
* Input-buffer size
* @exception IllegalArgumentException
* If size is ≤ 0
*/
public NonBlockingBufferedReader (@Nonnull final Reader aReader, @Nonnegative final int nSize)
{
super (aReader);
ValueEnforcer.isGT0 (nSize, "Size");
m_aReader = aReader;
m_aBuf = new char [nSize];
}
/**
* Checks to make sure that the stream has not been closed
*
* @throws IOException
* If the reader is not open
*/
private void _ensureOpen () throws IOException
{
if (m_aReader == null)
throw new IOException ("Stream closed");
}
/**
* Fills the input buffer, taking the mark into account if it is valid.
*
* @throws IOException
* In case reading fails
*/
private void _fill () throws IOException
{
int nDstOfs = 0;
if (m_nMarkedChar > UNMARKED)
{
// Marked
final int nDelta = m_nNextCharIndex - m_nMarkedChar;
if (nDelta >= m_nReadAheadLimit)
{
// Gone past read-ahead limit: Invalidate mark
m_nMarkedChar = INVALIDATED;
m_nReadAheadLimit = 0;
}
else
{
if (m_nReadAheadLimit <= m_aBuf.length)
{
// Shuffle in the current buffer
System.arraycopy (m_aBuf, m_nMarkedChar, m_aBuf, 0, nDelta);
m_nMarkedChar = 0;
nDstOfs = nDelta;
}
else
{
// Reallocate buffer to accommodate read-ahead limit
final char [] aNewBuf = new char [m_nReadAheadLimit];
System.arraycopy (m_aBuf, m_nMarkedChar, aNewBuf, 0, nDelta);
m_aBuf = aNewBuf;
m_nMarkedChar = 0;
nDstOfs = nDelta;
}
m_nNextCharIndex = nDelta;
m_nChars = nDelta;
}
}
int nBytesRead;
do
{
nBytesRead = m_aReader.read (m_aBuf, nDstOfs, m_aBuf.length - nDstOfs);
} while (nBytesRead == 0);
if (nBytesRead > 0)
{
m_nChars = nDstOfs + nBytesRead;
m_nNextCharIndex = nDstOfs;
}
}
/**
* Reads a single character.
*
* @return The character read, as an integer in the range 0 to 65535 (
* 0x00-0xffff), or -1 if the end of the stream has been
* reached
* @exception IOException
* If an I/O error occurs
*/
@Override
public int read () throws IOException
{
_ensureOpen ();
while (true)
{
if (m_nNextCharIndex >= m_nChars)
{
_fill ();
if (m_nNextCharIndex >= m_nChars)
return -1;
}
if (m_bSkipLF)
{
m_bSkipLF = false;
if (m_aBuf[m_nNextCharIndex] == '\n')
{
m_nNextCharIndex++;
continue;
}
}
return m_aBuf[m_nNextCharIndex++];
}
}
/**
* Reads characters into a portion of an array, reading from the underlying
* stream if necessary.
*
* @param aBuf
* The buffer to be filled
* @param nOfs
* The offset to start reading
* @param nLen
* The number of bytes to read
* @return The number of bytes read
* @throws IOException
* in case reading fails
*/
private int _internalRead (final char [] aBuf, final int nOfs, final int nLen) throws IOException
{
if (m_nNextCharIndex >= m_nChars)
{
/*
* If the requested length is at least as large as the buffer, and if
* there is no mark/reset activity, and if line feeds are not being
* skipped, do not bother to copy the characters into the local buffer. In
* this way buffered streams will cascade harmlessly.
*/
if (nLen >= m_aBuf.length && m_nMarkedChar <= UNMARKED && !m_bSkipLF)
return m_aReader.read (aBuf, nOfs, nLen);
_fill ();
}
if (m_nNextCharIndex >= m_nChars)
return -1;
if (m_bSkipLF)
{
m_bSkipLF = false;
if (m_aBuf[m_nNextCharIndex] == '\n')
{
m_nNextCharIndex++;
if (m_nNextCharIndex >= m_nChars)
_fill ();
if (m_nNextCharIndex >= m_nChars)
return -1;
}
}
final int nBytesRead = Math.min (nLen, m_nChars - m_nNextCharIndex);
System.arraycopy (m_aBuf, m_nNextCharIndex, aBuf, nOfs, nBytesRead);
m_nNextCharIndex += nBytesRead;
return nBytesRead;
}
/**
* Reads characters into a portion of an array.
*
* This method implements the general contract of the corresponding
* {@link Reader#read(char[], int, int) read} method of the
* {@link Reader} class. As an additional convenience, it
* attempts to read as many characters as possible by repeatedly invoking the
* read method of the underlying stream. This iterated
* read continues until one of the following conditions becomes
* true:
*
* - The specified number of characters have been read,
*
- The
read method of the underlying stream returns
* -1, indicating end-of-file, or
* - The
ready method of the underlying stream returns
* false, indicating that further input requests would block.
*
* If the first read on the underlying stream returns
* -1 to indicate end-of-file then this method returns
* -1. Otherwise this method returns the number of characters
* actually read.
*
* Subclasses of this class are encouraged, but not required, to attempt to
* read as many characters as possible in the same fashion.
*
* Ordinarily this method takes characters from this stream's character
* buffer, filling it from the underlying stream as necessary. If, however,
* the buffer is empty, the mark is not valid, and the requested length is at
* least as large as the buffer, then this method will read characters
* directly from the underlying stream into the given array. Thus redundant
* NonBlockingBufferedReaders will not copy data unnecessarily.
*
* @param cbuf
* Destination buffer
* @param nOfs
* Offset at which to start storing characters
* @param nLen
* Maximum number of characters to read
* @return The number of characters read, or -1 if the end of the stream has
* been reached
* @exception IOException
* If an I/O error occurs
*/
@Override
public int read (final char [] cbuf, final int nOfs, final int nLen) throws IOException
{
_ensureOpen ();
ValueEnforcer.isArrayOfsLen (cbuf, nOfs, nLen);
if (nLen == 0)
return 0;
// Main read
int n = _internalRead (cbuf, nOfs, nLen);
if (n <= 0)
return n;
while (n < nLen && m_aReader.ready ())
{
final int n1 = _internalRead (cbuf, nOfs + n, nLen - n);
if (n1 <= 0)
break;
n += n1;
}
return n;
}
/**
* Reads a line of text. A line is considered to be terminated by any one of a
* line feed ('\n'), a carriage return ('\r'), or a carriage return followed
* immediately by a linefeed.
*
* @return A {@link String} containing the contents of the line, not including
* any line-termination characters, or null if the end of
* the stream has been reached
* @exception IOException
* If an I/O error occurs
*/
@Nullable
public String readLine () throws IOException
{
StringBuilder aSB = null;
int nStartChar;
_ensureOpen ();
boolean bOmitLF = m_bSkipLF;
while (true)
{
if (m_nNextCharIndex >= m_nChars)
_fill ();
if (m_nNextCharIndex >= m_nChars)
{
/* EOF */
if (StringHelper.hasText (aSB))
return aSB.toString ();
return null;
}
boolean bEOL = false;
char cLast = 0;
int nIndex;
/* Skip a leftover '\n', if necessary */
if (bOmitLF && m_aBuf[m_nNextCharIndex] == '\n')
m_nNextCharIndex++;
m_bSkipLF = false;
bOmitLF = false;
for (nIndex = m_nNextCharIndex; nIndex < m_nChars; nIndex++)
{
cLast = m_aBuf[nIndex];
if (cLast == '\n' || cLast == '\r')
{
bEOL = true;
break;
}
}
nStartChar = m_nNextCharIndex;
m_nNextCharIndex = nIndex;
if (bEOL)
{
String sRet;
if (aSB == null)
sRet = new String (m_aBuf, nStartChar, nIndex - nStartChar);
else
{
aSB.append (m_aBuf, nStartChar, nIndex - nStartChar);
sRet = aSB.toString ();
}
m_nNextCharIndex++;
if (cLast == '\r')
m_bSkipLF = true;
return sRet;
}
if (aSB == null)
aSB = new StringBuilder (DEFAULT_EXPECTED_LINE_LENGTH);
aSB.append (m_aBuf, nStartChar, nIndex - nStartChar);
}
}
/**
* Skips characters.
*
* @param nBytes
* The number of characters to skip
* @return The number of characters actually skipped
* @exception IllegalArgumentException
* If n is negative.
* @exception IOException
* If an I/O error occurs
*/
@Override
public long skip (final long nBytes) throws IOException
{
ValueEnforcer.isGE0 (nBytes, "Bytes");
_ensureOpen ();
long nRest = nBytes;
while (nRest > 0)
{
if (m_nNextCharIndex >= m_nChars)
_fill ();
if (m_nNextCharIndex >= m_nChars)
{
// EOF
break;
}
if (m_bSkipLF)
{
m_bSkipLF = false;
if (m_aBuf[m_nNextCharIndex] == '\n')
m_nNextCharIndex++;
}
final int d = m_nChars - m_nNextCharIndex;
if (nRest <= d)
{
m_nNextCharIndex += (int) nRest;
nRest = 0;
break;
}
nRest -= d;
m_nNextCharIndex = m_nChars;
}
return nBytes - nRest;
}
/**
* Tells whether this stream is ready to be read. A buffered character stream
* is ready if the buffer is not empty, or if the underlying character stream
* is ready.
*
* @return true if the reader is ready
* @exception IOException
* If an I/O error occurs
*/
@Override
public boolean ready () throws IOException
{
_ensureOpen ();
/*
* If newline needs to be skipped and the next char to be read is a newline
* character, then just skip it right away.
*/
if (m_bSkipLF)
{
/*
* Note that in.ready() will return true if and only if the next read on
* the stream will not block.
*/
if (m_nNextCharIndex >= m_nChars && m_aReader.ready ())
_fill ();
if (m_nNextCharIndex < m_nChars)
{
if (m_aBuf[m_nNextCharIndex] == '\n')
m_nNextCharIndex++;
m_bSkipLF = false;
}
}
return m_nNextCharIndex < m_nChars || m_aReader.ready ();
}
/**
* Tells whether this stream supports the mark() operation, which it does.
*
* @return Always true
*/
@Override
public boolean markSupported ()
{
return true;
}
/**
* Marks the present position in the stream. Subsequent calls to reset() will
* attempt to reposition the stream to this point.
*
* @param nReadAheadLimit
* Limit on the number of characters that may be read while still
* preserving the mark. An attempt to reset the stream after reading
* characters up to this limit or beyond may fail. A limit value larger
* than the size of the input buffer will cause a new buffer to be
* allocated whose size is no smaller than limit. Therefore large
* values should be used with care.
* @exception IllegalArgumentException
* If readAheadLimit is %lt; 0
* @exception IOException
* If an I/O error occurs
*/
@Override
public void mark (@Nonnegative final int nReadAheadLimit) throws IOException
{
ValueEnforcer.isGE0 (nReadAheadLimit, "ReadAheadLimit");
_ensureOpen ();
m_nReadAheadLimit = nReadAheadLimit;
m_nMarkedChar = m_nNextCharIndex;
m_bMarkedSkipLF = m_bSkipLF;
}
/**
* Resets the stream to the most recent mark.
*
* @exception IOException
* If the stream has never been marked, or if the mark has been
* invalidated
*/
@Override
public void reset () throws IOException
{
_ensureOpen ();
if (m_nMarkedChar < 0)
throw new IOException (m_nMarkedChar == INVALIDATED ? "Mark invalid" : "Stream not marked");
m_nNextCharIndex = m_nMarkedChar;
m_bSkipLF = m_bMarkedSkipLF;
}
@Override
public void close () throws IOException
{
if (m_aReader != null)
{
m_aReader.close ();
m_aReader = null;
m_aBuf = null;
}
}
/**
* Returns a {@code Stream}, the elements of which are lines read from this
* {@code BufferedReader}. The {@link Stream} is lazily populated, i.e., read
* only occurs during the
* terminal stream
* operation.
*
* The reader must not be operated on during the execution of the terminal
* stream operation. Otherwise, the result of the terminal stream operation is
* undefined.
*
* After execution of the terminal stream operation there are no guarantees
* that the reader will be at a specific position from which to read the next
* character or line.
*
* If an {@link IOException} is thrown when accessing the underlying
* {@code BufferedReader}, it is wrapped in an {@link UncheckedIOException}
* which will be thrown from the {@code Stream} method that caused the read to
* take place. This method will return a Stream if invoked on a BufferedReader
* that is closed. Any operation on that stream that requires reading from the
* BufferedReader after it is closed, will cause an UncheckedIOException to be
* thrown.
*
* @return a {@code Stream} providing the lines of text described by
* this {@code BufferedReader}
* @since 1.8
*/
@Nonnull
public Stream lines ()
{
final Iterator aIter = new Iterator ()
{
private String m_sNextLine;
@Override
public boolean hasNext ()
{
if (m_sNextLine != null)
return true;
try
{
m_sNextLine = readLine ();
return m_sNextLine != null;
}
catch (final IOException e)
{
throw new UncheckedIOException (e);
}
}
@Override
public String next ()
{
if (m_sNextLine != null || hasNext ())
{
final String sLine = m_sNextLine;
m_sNextLine = null;
return sLine;
}
throw new NoSuchElementException ();
}
};
return StreamSupport.stream (Spliterators.spliteratorUnknownSize (aIter, Spliterator.ORDERED | Spliterator.NONNULL), false);
}
}