2019-02-26 18:40:08 +00:00
|
|
|
#include <Processors/Transforms/PartialSortingTransform.h>
|
|
|
|
#include <Interpreters/sortBlock.h>
|
|
|
|
|
|
|
|
namespace DB
|
|
|
|
{
|
|
|
|
|
2019-02-27 11:24:14 +00:00
|
|
|
PartialSortingTransform::PartialSortingTransform(const Block & header, SortDescription & description, UInt64 limit)
|
2019-02-26 18:40:08 +00:00
|
|
|
: ISimpleTransform(header, header, false)
|
|
|
|
, description(description), limit(limit)
|
|
|
|
{
|
|
|
|
}
|
|
|
|
|
|
|
|
void PartialSortingTransform::transform(Chunk & chunk)
|
|
|
|
{
|
|
|
|
auto num_rows = chunk.getNumRows();
|
|
|
|
auto block = getInputPort().getHeader().cloneWithColumns(chunk.detachColumns());
|
|
|
|
chunk.clear();
|
|
|
|
sortBlock(block, description, limit);
|
|
|
|
chunk.setColumns(block.getColumns(), num_rows);
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|