2016-07-01 21:02:13 +00:00
|
|
|
#include <DB/DataStreams/SquashingBlockInputStream.h>
|
|
|
|
|
|
|
|
|
|
|
|
namespace DB
|
|
|
|
{
|
|
|
|
|
|
|
|
SquashingBlockInputStream::SquashingBlockInputStream(BlockInputStreamPtr & src, size_t min_block_size_rows, size_t min_block_size_bytes)
|
2016-07-06 21:48:11 +00:00
|
|
|
: transform(min_block_size_rows, min_block_size_bytes)
|
2016-07-01 21:02:13 +00:00
|
|
|
{
|
|
|
|
children.emplace_back(src);
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
Block SquashingBlockInputStream::readImpl()
|
|
|
|
{
|
2016-07-07 01:57:48 +00:00
|
|
|
if (all_read)
|
|
|
|
return {};
|
|
|
|
|
2016-07-06 21:48:11 +00:00
|
|
|
while (true)
|
2016-07-01 21:02:13 +00:00
|
|
|
{
|
2016-07-07 01:57:48 +00:00
|
|
|
Block block = children[0]->read();
|
|
|
|
if (!block)
|
|
|
|
all_read = true;
|
|
|
|
|
|
|
|
SquashingTransform::Result result = transform.add(std::move(block));
|
2016-07-06 21:48:11 +00:00
|
|
|
if (result.ready)
|
|
|
|
return result.block;
|
2016-07-01 21:02:13 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|