ClickHouse/dbms/include/DB/DataStreams/NativeBlockOutputStream.h

48 lines
2.0 KiB
C++
Raw Normal View History

2011-08-19 19:18:15 +00:00
#pragma once
#include <DB/DataStreams/IBlockOutputStream.h>
#include <DB/Core/Types.h>
#include <DB/DataTypes/IDataType.h>
2011-08-19 19:18:15 +00:00
namespace DB
{
class WriteBuffer;
class CompressedWriteBuffer;
2011-08-19 19:18:15 +00:00
/** Сериализует поток блоков в родном бинарном формате (с именами и типами столбцов).
* Предназначено для взаимодействия между серверами.
*
* Может быть указан поток для записи индекса. Индекс содержит смещения до каждого кусочка каждого столбца.
* Если делается append в уже существующий файл, и нужно записать индекс, то укажите initial_size_of_file.
2011-08-19 19:18:15 +00:00
*/
class NativeBlockOutputStream : public IBlockOutputStream
{
public:
/** В случае указания ненулевой client_revision, может записываться дополнительная информация о блоке,
* в зависимости от поддерживаемой для указанной ревизии.
*/
NativeBlockOutputStream(
WriteBuffer & ostr_, UInt64 client_revision_ = 0,
WriteBuffer * index_ostr_ = nullptr, size_t initial_size_of_file_ = 0);
2011-08-19 19:18:15 +00:00
void write(const Block & block) override;
2016-08-13 01:57:35 +00:00
void flush() override;
static void writeData(const IDataType & type, const ColumnPtr & column, WriteBuffer & ostr, size_t offset, size_t limit);
2015-10-29 20:38:37 +00:00
String getContentType() const override { return "application/octet-stream"; }
2011-08-19 19:18:15 +00:00
private:
WriteBuffer & ostr;
UInt64 client_revision;
WriteBuffer * index_ostr;
size_t initial_size_of_file; /// Начальный размер файла с данными, если делается append. Используется для индекса.
/// Если требуется записывать индекс, то ostr обязан быть CompressedWriteBuffer.
CompressedWriteBuffer * ostr_concrete = nullptr;
2011-08-19 19:18:15 +00:00
};
}