mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-10-19 06:50:50 +00:00
88 lines
2.9 KiB
C++
88 lines
2.9 KiB
C++
#include <Storages/MergeTree/IMergeTreeDataPartWriter.h>
|
|
|
|
namespace DB
|
|
{
|
|
|
|
class MergeTreeDataPartWriterWide : public IMergeTreeDataPartWriter
|
|
{
|
|
public:
|
|
|
|
using ColumnToSize = std::map<std::string, UInt64>;
|
|
|
|
MergeTreeDataPartWriterWide(
|
|
const String & part_path,
|
|
const MergeTreeData & storage,
|
|
const NamesAndTypesList & columns_list,
|
|
const String & marks_file_extension,
|
|
const CompressionCodecPtr & default_codec,
|
|
const WriterSettings & settings,
|
|
const ColumnToSize & merged_column_to_size = {});
|
|
|
|
MarkWithOffset write(const Block & block, const IColumn::Permutation * permutation,
|
|
size_t from_mark, size_t index_offset,
|
|
const MergeTreeIndexGranularity & index_granularity,
|
|
const Block & primary_key_block, const Block & skip_indexes_block) override;
|
|
|
|
void finalize(IMergeTreeDataPart::Checksums & checksums, bool write_final_mark, bool sync = false) override;
|
|
|
|
IDataType::OutputStreamGetter createStreamGetter(const String & name, WrittenOffsetColumns & offset_columns, bool skip_offsets);
|
|
|
|
/// Write data of one column.
|
|
/// Return how many marks were written and
|
|
/// how many rows were written for last mark
|
|
MarkWithOffset writeColumn(
|
|
const String & name,
|
|
const IDataType & type,
|
|
const IColumn & column,
|
|
const MergeTreeIndexGranularity & index_granularity,
|
|
WrittenOffsetColumns & offset_columns,
|
|
bool skip_offsets,
|
|
IDataType::SerializeBinaryBulkStatePtr & serialization_state,
|
|
size_t from_mark,
|
|
size_t index_offset);
|
|
|
|
private:
|
|
/// Write single granule of one column (rows between 2 marks)
|
|
size_t writeSingleGranule(
|
|
const String & name,
|
|
const IDataType & type,
|
|
const IColumn & column,
|
|
WrittenOffsetColumns & offset_columns,
|
|
bool skip_offsets,
|
|
IDataType::SerializeBinaryBulkStatePtr & serialization_state,
|
|
IDataType::SerializeBinaryBulkSettings & serialize_settings,
|
|
size_t from_row,
|
|
size_t number_of_rows,
|
|
bool write_marks);
|
|
|
|
/// Write mark for column
|
|
void writeSingleMark(
|
|
const String & name,
|
|
const IDataType & type,
|
|
WrittenOffsetColumns & offset_columns,
|
|
bool skip_offsets,
|
|
size_t number_of_rows,
|
|
DB::IDataType::SubstreamPath & path);
|
|
|
|
void writeFinalMark(
|
|
const std::string & column_name,
|
|
const DataTypePtr column_type,
|
|
WrittenOffsetColumns & offset_columns,
|
|
bool skip_offsets,
|
|
DB::IDataType::SubstreamPath & path);
|
|
|
|
void addStreams(
|
|
const String & name,
|
|
const IDataType & type,
|
|
const CompressionCodecPtr & effective_codec,
|
|
size_t estimated_size,
|
|
bool skip_offsets);
|
|
|
|
SerializationStates serialization_states;
|
|
bool can_use_adaptive_granularity;
|
|
ColumnStreams column_streams;
|
|
|
|
};
|
|
|
|
}
|