Skip to content

Commit

Permalink
Merge ce059b2 into dd668c0
Browse files Browse the repository at this point in the history
  • Loading branch information
ivanmorozov333 authored Jan 11, 2025
2 parents dd668c0 + ce059b2 commit 86a7403
Showing 1 changed file with 3 additions and 4 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@ void TScanHead::OnSourceReady(const std::shared_ptr<IDataSource>& source, std::s
Context->GetCommonContext()->GetCounters().OnSourceFinished(
source->GetRecordsCount(), source->GetUsedRawBytes(), tableExt ? tableExt->num_rows() : 0);

if ((!tableExt || !tableExt->num_rows()) && Context->GetCommonContext()->GetReadMetadata()->HasLimit() && InFlightLimit < MaxInFlight) {
if (/*(!tableExt || !tableExt->num_rows()) &&*/ Context->GetCommonContext()->GetReadMetadata()->HasLimit() && InFlightLimit < MaxInFlight) {
InFlightLimit = 2 * InFlightLimit;
}
source->MutableStageResult().SetResultChunk(std::move(tableExt), startIndex, recordsCount);
Expand Down Expand Up @@ -59,10 +59,9 @@ void TScanHead::OnSourceReady(const std::shared_ptr<IDataSource>& source, std::s
AFL_VERIFY(FetchingSourcesByIdx.erase(frontSource->GetSourceIdx()));
FetchingSources.pop_front();
frontSource->ClearResult();
if (Context->GetCommonContext()->GetReadMetadata()->HasLimit() && FetchingSources.size() && frontSource->GetResultRecordsCount()) {
if (Context->GetCommonContext()->GetReadMetadata()->HasLimit() && SortedSources.size() && frontSource->GetResultRecordsCount()) {
FinishedSources.emplace(frontSource);
while (FinishedSources.size() && (*FinishedSources.begin())->GetFinish() < FetchingSources.front()->GetStart()) {
auto fetchingSource = FetchingSources.front();
while (FinishedSources.size() && (*FinishedSources.begin())->GetFinish() < SortedSources.front()->GetStart()) {
auto finishedSource = *FinishedSources.begin();
FetchedCount += finishedSource->GetResultRecordsCount();
FinishedSources.erase(FinishedSources.begin());
Expand Down

0 comments on commit 86a7403

Please sign in to comment.