mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-16 03:12:43 +00:00
40 lines
882 B
C++
40 lines
882 B
C++
#include <IO/ReadBufferFromIStream.h>
|
|
#include <Common/Exception.h>
|
|
|
|
|
|
namespace DB
|
|
{
|
|
|
|
namespace ErrorCodes
|
|
{
|
|
extern const int CANNOT_READ_FROM_ISTREAM;
|
|
}
|
|
|
|
bool ReadBufferFromIStream::nextImpl()
|
|
{
|
|
istr.read(internal_buffer.begin(), internal_buffer.size());
|
|
size_t gcount = istr.gcount();
|
|
|
|
if (!gcount)
|
|
{
|
|
if (istr.eof())
|
|
return false;
|
|
|
|
if (istr.fail())
|
|
throw Exception("Cannot read from istream at offset " + std::to_string(count()), ErrorCodes::CANNOT_READ_FROM_ISTREAM);
|
|
|
|
throw Exception("Unexpected state of istream at offset " + std::to_string(count()), ErrorCodes::CANNOT_READ_FROM_ISTREAM);
|
|
}
|
|
else
|
|
working_buffer.resize(gcount);
|
|
|
|
return true;
|
|
}
|
|
|
|
ReadBufferFromIStream::ReadBufferFromIStream(std::istream & istr_, size_t size)
|
|
: BufferWithOwnMemory<ReadBuffer>(size), istr(istr_)
|
|
{
|
|
}
|
|
|
|
}
|