mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-23 16:12:01 +00:00
63 lines
1.7 KiB
C++
63 lines
1.7 KiB
C++
#pragma once
|
|
|
|
#include <queue>
|
|
|
|
#include <Yandex/logger_useful.h>
|
|
|
|
#include <DB/Core/SortDescription.h>
|
|
|
|
#include <DB/DataStreams/IProfilingBlockInputStream.h>
|
|
|
|
|
|
namespace DB
|
|
{
|
|
|
|
/** Соединяет несколько сортированных потоков в один.
|
|
*/
|
|
class MergingSortedBlockInputStream : public IProfilingBlockInputStream
|
|
{
|
|
public:
|
|
MergingSortedBlockInputStream(BlockInputStreams inputs_, SortDescription & description_, size_t max_block_size_)
|
|
: inputs(inputs_), description(description_), max_block_size(max_block_size_), first(true),
|
|
num_columns(0), source_blocks(inputs.size()), cursors(inputs.size()), log(&Logger::get("MergingSortedBlockInputStream"))
|
|
{
|
|
children.insert(children.end(), inputs.begin(), inputs.end());
|
|
}
|
|
|
|
Block readImpl();
|
|
void readSuffix();
|
|
|
|
String getName() const { return "MergingSortedBlockInputStream"; }
|
|
|
|
BlockInputStreamPtr clone() { return new MergingSortedBlockInputStream(inputs, description, max_block_size); }
|
|
|
|
protected:
|
|
/// Инициализирует очередь и следующий блок результата.
|
|
void init(Block & merged_block, ColumnPlainPtrs & merged_columns);
|
|
|
|
/// Достаёт из источника, соответствующего current следующий блок.
|
|
void fetchNextBlock(const SortCursor & current);
|
|
|
|
|
|
BlockInputStreams inputs;
|
|
SortDescription description;
|
|
size_t max_block_size;
|
|
|
|
bool first;
|
|
|
|
/// Текущие сливаемые блоки.
|
|
size_t num_columns;
|
|
Blocks source_blocks;
|
|
|
|
typedef std::vector<SortCursorImpl> CursorImpls;
|
|
CursorImpls cursors;
|
|
|
|
typedef std::priority_queue<SortCursor> Queue;
|
|
Queue queue;
|
|
|
|
private:
|
|
Logger * log;
|
|
};
|
|
|
|
}
|