mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-01 12:01:58 +00:00
37 lines
817 B
C++
37 lines
817 B
C++
#include <DataStreams/SquashingBlockInputStream.h>
|
|
|
|
|
|
namespace DB
|
|
{
|
|
|
|
SquashingBlockInputStream::SquashingBlockInputStream(
|
|
const BlockInputStreamPtr & src, size_t min_block_size_rows, size_t min_block_size_bytes)
|
|
: header(src->getHeader()), transform(min_block_size_rows, min_block_size_bytes)
|
|
{
|
|
children.emplace_back(src);
|
|
}
|
|
|
|
|
|
Block SquashingBlockInputStream::readImpl()
|
|
{
|
|
if (all_read)
|
|
return {};
|
|
|
|
while (true)
|
|
{
|
|
Block block = children[0]->read();
|
|
if (!block)
|
|
all_read = true;
|
|
|
|
SquashingTransform::Result result = transform.add(block.mutateColumns());
|
|
if (result.ready)
|
|
{
|
|
if (result.columns.empty())
|
|
return {};
|
|
return header.cloneWithColumns(std::move(result.columns));
|
|
}
|
|
}
|
|
}
|
|
|
|
}
|