ClickHouse/dbms/include/DB/Storages/MergeTree/MergeTreeSharder.h

#pragma once

#include <DB/IO/WriteBufferFromFile.h>
#include <DB/IO/CompressedWriteBuffer.h>
#include <DB/Columns/ColumnsNumber.h>
#include <DB/Interpreters/sortBlock.h>
#include <DB/Storages/MergeTree/MergeTreeData.h>
#include <DB/Core/Block.h>

namespace DB
{

struct ShardedBlockWithDateInterval final
{
	ShardedBlockWithDateInterval(const Block & block_, size_t shard_no_, UInt16 min_date_, UInt16 max_date_);
	ShardedBlockWithDateInterval(const ShardedBlockWithDateInterval &) = delete;
	ShardedBlockWithDateInterval & operator=(const ShardedBlockWithDateInterval &) = delete;

	Block block;
	size_t shard_no;
	UInt16 min_date;
	UInt16 max_date;
};

using ShardedBlocksWithDateIntervals = std::list<ShardedBlockWithDateInterval>;

class ReshardingJob;

/** Создаёт новые шардированные куски с данными.
  */
class MergeTreeSharder final
{
public:
	MergeTreeSharder(MergeTreeData & data_, const ReshardingJob & job_);
	MergeTreeSharder(const MergeTreeSharder &) = delete;
	MergeTreeSharder & operator=(const MergeTreeSharder &) = delete;

	/** Разбивает блок на блоки по ключу шардирования, каждый из которых
	  * нужно записать в отдельный кусок. Работает детерминированно: если
	  * отдать на вход такой же блок, на выходе получатся такие же блоки в
	  * таком же порядке.
	  */
	ShardedBlocksWithDateIntervals shardBlock(const Block & block);

	/** Все строки должны относиться к одному месяцу.
	  * temp_index - значение left и right для нового куска. Можно будет изменить при переименовании.
	  * Возвращает временный кусок с именем, начинающимся с tmp_.
	  */
	MergeTreeData::MutableDataPartPtr writeTempPart(ShardedBlockWithDateInterval & sharded_block_with_dates, Int64 temp_index);

private:
	std::vector<IColumn::Filter> createFilters(Block block);

private:
	MergeTreeData & data;
	const ReshardingJob & job;
	Logger * log;
	std::vector<size_t> slots;
};

}