thevar1able/ClickHouse
1
0
Fork 0
mirror of https://github.com/ClickHouse/ClickHouse.git synced 2024-12-01 12:01:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix limit transfrom prepare.

Browse Source
This commit is contained in:
Nikolai Kochetov 2019-04-12 17:43:28 +03:00
parent a8fdddfc88
commit 5aa3d98ea2
1 changed files with 29 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

35
dbms/src/Processors/LimitTransform.cpp
View File

@ -18,30 +18,35 @@ LimitTransform::LimitTransform(
LimitTransform::Status LimitTransform::prepare()
{
/// Check if we are done with pushing.
bool pushing_is_finished = rows_read >= offset + limit;
/// Check can output.
if (output.isFinished())
{
input.close();
return Status::Finished;
pushing_is_finished = true;
if (!always_read_till_end)
{
input.close();
return Status::Finished;
}
}
if (!output.canPush())
if (!pushing_is_finished && !output.canPush())
{
input.setNotNeeded();
return Status::PortFull;
}
/// Push block if can.
if (has_block && block_processed)
if (!pushing_is_finished && has_block && block_processed)
{
output.push(std::move(current_chunk));
has_block = false;
block_processed = false;
}
/// Check if we are done with pushing.
bool pushing_is_finished = rows_read >= offset + limit;
if (pushing_is_finished)
{
if (!always_read_till_end)
@ -77,6 +82,12 @@ LimitTransform::Status LimitTransform::prepare()
current_chunk.clear();
has_block = false;
if (input.isFinished())
{
output.finish();
return Status::Finished;
}
/// Now, we pulled from input, and it must be empty.
return Status::NeedData;
}
@ -90,6 +101,12 @@ LimitTransform::Status LimitTransform::prepare()
current_chunk.clear();
has_block = false;
if (input.isFinished())
{
output.finish();
return Status::Finished;
}
/// Now, we pulled from input, and it must be empty.
return Status::NeedData;
}
@ -103,6 +120,12 @@ LimitTransform::Status LimitTransform::prepare()
output.push(std::move(current_chunk));
has_block = false;
if (input.isFinished())
{
output.finish();
return Status::Finished;
}
return Status::NeedData;
}