From bcf3534e355645b664ec77caa801c560ea5ba6c3 Mon Sep 17 00:00:00 2001
From: Nikolai Kochetov <nik-kochetov@yandex-team.ru>
Date: Fri, 12 Apr 2019 18:20:24 +0300
Subject: [PATCH] Fix get rows before limit at least.

---
 dbms/src/Processors/QueryPipeline.cpp | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/dbms/src/Processors/QueryPipeline.cpp b/dbms/src/Processors/QueryPipeline.cpp
index 57965335f92..42ac10a6518 100644
--- a/dbms/src/Processors/QueryPipeline.cpp
+++ b/dbms/src/Processors/QueryPipeline.cpp
@@ -503,6 +503,16 @@ void QueryPipeline::calcRowsBeforeLimit()
             has_limit = true;
             rows_before_limit_at_least += limit->getRowsBeforeLimitAtLeast();
         }
+
+        if (auto * source = typeid_cast<SourceFromInputStream *>(processor.get()))
+        {
+            auto & info = source->getStream().getProfileInfo();
+            if (info.hasAppliedLimit())
+            {
+                has_limit = true;
+                rows_before_limit_at_least += info.getRowsBeforeLimit();
+            }
+        }
     }
 
     UInt64 rows_before_limit = 0;