2011-08-09 15:57:33 +00:00
|
|
|
|
#pragma once
|
2010-06-01 13:35:09 +00:00
|
|
|
|
|
|
|
|
|
#include <algorithm>
|
2011-06-27 18:22:14 +00:00
|
|
|
|
#include <cstring>
|
2010-06-01 13:35:09 +00:00
|
|
|
|
|
2015-10-05 01:35:28 +00:00
|
|
|
|
#include <DB/Common/Exception.h>
|
2011-06-27 18:22:14 +00:00
|
|
|
|
#include <DB/IO/BufferBase.h>
|
2010-06-01 13:35:09 +00:00
|
|
|
|
|
|
|
|
|
|
|
|
|
|
namespace DB
|
|
|
|
|
{
|
|
|
|
|
|
2016-01-11 21:46:36 +00:00
|
|
|
|
namespace ErrorCodes
|
|
|
|
|
{
|
|
|
|
|
extern const int CANNOT_WRITE_AFTER_END_OF_BUFFER;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
2010-06-01 13:35:09 +00:00
|
|
|
|
/** Простой абстрактный класс для буферизованной записи данных (последовательности char) куда-нибудь.
|
|
|
|
|
* В отличие от std::ostream, предоставляет доступ к внутреннему буферу,
|
|
|
|
|
* а также позволяет вручную управлять позицией внутри буфера.
|
|
|
|
|
*
|
2011-06-24 21:08:26 +00:00
|
|
|
|
* Наследники должны реализовать метод nextImpl().
|
2010-06-01 13:35:09 +00:00
|
|
|
|
*/
|
2011-06-27 18:22:14 +00:00
|
|
|
|
class WriteBuffer : public BufferBase
|
2010-06-01 13:35:09 +00:00
|
|
|
|
{
|
|
|
|
|
public:
|
2011-06-27 18:22:14 +00:00
|
|
|
|
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
|
|
|
|
|
|
|
|
|
/** записать данные, находящиеся в буфере (от начала буфера до текущей позиции);
|
|
|
|
|
* переместить позицию в начало; кинуть исключение, если что-то не так
|
|
|
|
|
*/
|
2011-06-24 21:08:26 +00:00
|
|
|
|
inline void next()
|
|
|
|
|
{
|
2011-06-27 19:24:33 +00:00
|
|
|
|
if (!offset())
|
|
|
|
|
return;
|
2011-06-27 18:22:14 +00:00
|
|
|
|
bytes += offset();
|
2011-12-12 01:06:13 +00:00
|
|
|
|
|
|
|
|
|
try
|
|
|
|
|
{
|
|
|
|
|
nextImpl();
|
|
|
|
|
}
|
|
|
|
|
catch (...)
|
|
|
|
|
{
|
|
|
|
|
/** Если вызов nextImpl() был неудачным, то переместим курсор в начало,
|
|
|
|
|
* чтобы потом (например, при развёртке стека) не было второй попытки записать данные.
|
|
|
|
|
*/
|
|
|
|
|
pos = working_buffer.begin();
|
|
|
|
|
throw;
|
|
|
|
|
}
|
2015-02-07 23:13:04 +00:00
|
|
|
|
|
2011-06-24 21:08:26 +00:00
|
|
|
|
pos = working_buffer.begin();
|
|
|
|
|
}
|
2010-06-01 13:35:09 +00:00
|
|
|
|
|
2011-12-12 01:06:13 +00:00
|
|
|
|
/** желательно в наследниках поместить в деструктор вызов next(),
|
2010-06-04 18:25:25 +00:00
|
|
|
|
* чтобы последние данные записались
|
|
|
|
|
*/
|
2010-06-01 13:35:09 +00:00
|
|
|
|
virtual ~WriteBuffer() {}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
inline void nextIfAtEnd()
|
|
|
|
|
{
|
2015-02-07 23:13:04 +00:00
|
|
|
|
if (!hasPendingData())
|
2010-06-01 13:35:09 +00:00
|
|
|
|
next();
|
|
|
|
|
}
|
|
|
|
|
|
2015-02-07 23:13:04 +00:00
|
|
|
|
|
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
|
|
|
|
}
|
|
|
|
|
|
2011-05-10 17:23:29 +00:00
|
|
|
|
|
|
|
|
|
inline void write(char x)
|
|
|
|
|
{
|
|
|
|
|
nextIfAtEnd();
|
|
|
|
|
*pos = x;
|
|
|
|
|
++pos;
|
|
|
|
|
}
|
2015-02-07 23:13:04 +00:00
|
|
|
|
|
2011-05-05 19:10:17 +00:00
|
|
|
|
private:
|
2011-06-24 21:08:26 +00:00
|
|
|
|
/** Записать данные, находящиеся в буфере (от начала буфера до текущей позиции).
|
|
|
|
|
* Кинуть исключение, если что-то не так.
|
|
|
|
|
*/
|
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
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
}
|