ClickHouse/dbms/include/DB/IO/WriteBuffer.h

95 lines
2.7 KiB
C
Raw Normal View History

2011-08-09 15:57:33 +00:00
#pragma once
2010-06-01 13:35:09 +00:00
#include <algorithm>
#include <cstring>
2010-06-01 13:35:09 +00:00
2011-10-17 19:32:54 +00:00
#include <DB/Core/Exception.h>
#include <DB/Core/ErrorCodes.h>
#include <DB/IO/BufferBase.h>
2010-06-01 13:35:09 +00:00
namespace DB
{
/** Простой абстрактный класс для буферизованной записи данных (последовательности char) куда-нибудь.
* В отличие от std::ostream, предоставляет доступ к внутреннему буферу,
* а также позволяет вручную управлять позицией внутри буфера.
*
* Наследники должны реализовать метод nextImpl().
2010-06-01 13:35:09 +00:00
*/
class WriteBuffer : public BufferBase
2010-06-01 13:35:09 +00:00
{
public:
WriteBuffer(Position ptr, size_t size) : BufferBase(ptr, size, 0) {}
void set(Position ptr, size_t size) { BufferBase::set(ptr, size, 0); }
2010-06-01 13:35:09 +00:00
/** записать данные, находящиеся в буфере (от начала буфера до текущей позиции);
* переместить позицию в начало; кинуть исключение, если что-то не так
*/
inline void next()
{
2011-06-27 19:24:33 +00:00
if (!offset())
return;
bytes += offset();
try
{
nextImpl();
}
catch (...)
{
/** Если вызов nextImpl() был неудачным, то переместим курсор в начало,
* чтобы потом (например, при развёртке стека) не было второй попытки записать данные.
*/
pos = working_buffer.begin();
throw;
}
pos = working_buffer.begin();
}
2010-06-01 13:35:09 +00:00
/** желательно в наследниках поместить в деструктор вызов next(),
2010-06-04 18:25:25 +00:00
* чтобы последние данные записались
*/
2010-06-01 13:35:09 +00:00
virtual ~WriteBuffer() {}
inline void nextIfAtEnd()
{
if (!hasPendingData())
2010-06-01 13:35:09 +00:00
next();
}
2010-06-01 13:35:09 +00:00
void write(const char * from, size_t n)
{
size_t bytes_copied = 0;
while (bytes_copied < n)
{
nextIfAtEnd();
size_t bytes_to_copy = std::min(static_cast<size_t>(working_buffer.end() - pos), n - bytes_copied);
std::memcpy(pos, from + bytes_copied, bytes_to_copy);
pos += bytes_to_copy;
bytes_copied += bytes_to_copy;
}
2011-05-05 19:10:17 +00:00
}
inline void write(char x)
{
nextIfAtEnd();
*pos = x;
++pos;
}
2011-05-05 19:10:17 +00:00
private:
/** Записать данные, находящиеся в буфере (от начала буфера до текущей позиции).
* Кинуть исключение, если что-то не так.
*/
2011-10-17 19:32:54 +00:00
virtual void nextImpl() { throw Exception("Cannot write after end of buffer.", ErrorCodes::CANNOT_WRITE_AFTER_END_OF_BUFFER); };
2010-06-01 13:35:09 +00:00
};
}