Update LimitTransform.

This commit is contained in:
Nikolai Kochetov 2020-01-28 14:50:22 +03:00
parent f94eb7a0ff
commit a19c8b3d37

View File

@ -76,7 +76,7 @@ LimitTransform::Status LimitTransform::prepare()
if (!input.hasData()) if (!input.hasData())
return Status::NeedData; return Status::NeedData;
current_chunk = input.pull(); current_chunk = input.pull(true);
has_block = true; has_block = true;
auto rows = current_chunk.getNumRows(); auto rows = current_chunk.getNumRows();
@ -95,6 +95,7 @@ LimitTransform::Status LimitTransform::prepare()
} }
/// Now, we pulled from input, and it must be empty. /// Now, we pulled from input, and it must be empty.
input.setNeeded();
return Status::NeedData; return Status::NeedData;
} }
@ -114,6 +115,7 @@ LimitTransform::Status LimitTransform::prepare()
} }
/// Now, we pulled from input, and it must be empty. /// Now, we pulled from input, and it must be empty.
input.setNeeded();
return Status::NeedData; return Status::NeedData;
} }