2018-07-09 20:36:58 +00:00
|
|
|
#include <IO/WriteBufferFromOStream.h>
|
|
|
|
#include <Common/Exception.h>
|
|
|
|
|
|
|
|
|
|
|
|
namespace DB
|
|
|
|
{
|
|
|
|
|
|
|
|
namespace ErrorCodes
|
|
|
|
{
|
|
|
|
extern const int CANNOT_WRITE_TO_OSTREAM;
|
|
|
|
}
|
|
|
|
|
|
|
|
void WriteBufferFromOStream::nextImpl()
|
|
|
|
{
|
|
|
|
if (!offset())
|
|
|
|
return;
|
|
|
|
|
|
|
|
ostr->write(working_buffer.begin(), offset());
|
|
|
|
ostr->flush();
|
|
|
|
|
|
|
|
if (!ostr->good())
|
2018-07-10 20:45:27 +00:00
|
|
|
throw Exception("Cannot write to ostream at offset " + std::to_string(count()),
|
2018-07-10 19:30:58 +00:00
|
|
|
ErrorCodes::CANNOT_WRITE_TO_OSTREAM);
|
2018-07-09 20:36:58 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
WriteBufferFromOStream::WriteBufferFromOStream(
|
|
|
|
size_t size,
|
|
|
|
char * existing_memory,
|
|
|
|
size_t alignment)
|
|
|
|
: BufferWithOwnMemory<WriteBuffer>(size, existing_memory, alignment)
|
|
|
|
{
|
|
|
|
}
|
|
|
|
|
|
|
|
WriteBufferFromOStream::WriteBufferFromOStream(
|
|
|
|
std::ostream & ostr_,
|
|
|
|
size_t size,
|
|
|
|
char * existing_memory,
|
|
|
|
size_t alignment)
|
|
|
|
: BufferWithOwnMemory<WriteBuffer>(size, existing_memory, alignment), ostr(&ostr_)
|
|
|
|
{
|
|
|
|
}
|
|
|
|
|
|
|
|
WriteBufferFromOStream::~WriteBufferFromOStream()
|
|
|
|
{
|
|
|
|
try
|
|
|
|
{
|
|
|
|
next();
|
|
|
|
}
|
|
|
|
catch (...)
|
|
|
|
{
|
|
|
|
tryLogCurrentException(__PRETTY_FUNCTION__);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|