fix issue #7810 (#7854) (#7870)

close #7810
pingcap · Dec 4, 2023 · d48dc34 · d48dc34
1 parent c8d1b80
commit d48dc34
Showing 1 changed file with 14 additions and 11 deletions.
diff --git a/dbms/src/Interpreters/executeQuery.cpp b/dbms/src/Interpreters/executeQuery.cpp
@@ -240,19 +240,22 @@ std::tuple<ASTPtr, BlockIO> executeQueryImpl(
         {
             if (auto * stream = dynamic_cast<IProfilingBlockInputStream *>(res.in.get()))
             {
-                stream->setProgressCallback(context.getProgressCallback());
-                stream->setProcessListElement(context.getProcessListElement());
-
-                /// Limits on the result, the quota on the result, and also callback for progress.
-                /// Limits apply only to the final result.
-                if (stage == QueryProcessingStage::Complete)
+                if (auto * dag_src = dynamic_cast<DAGQuerySource *>(&query_src); dag_src == nullptr)
                 {
-                    IProfilingBlockInputStream::LocalLimits limits;
-                    limits.mode = IProfilingBlockInputStream::LIMITS_CURRENT;
-                    limits.size_limits = SizeLimits(settings.max_result_rows, settings.max_result_bytes, settings.result_overflow_mode);
+                    stream->setProgressCallback(context.getProgressCallback());
+                    stream->setProcessListElement(context.getProcessListElement());
 
-                    stream->setLimits(limits);
-                    stream->setQuota(quota);
+                    /// Limits on the result, the quota on the result, and also callback for progress.
+                    /// Limits apply only to the final result.
+                    if (stage == QueryProcessingStage::Complete)
+                    {
+                        IProfilingBlockInputStream::LocalLimits limits;
+                        limits.mode = IProfilingBlockInputStream::LIMITS_CURRENT;
+                        limits.size_limits = SizeLimits(settings.max_result_rows, settings.max_result_bytes, settings.result_overflow_mode);
+
+                        stream->setLimits(limits);
+                        stream->setQuota(quota);
+                    }
                 }
             }
         }