From c44875a339a300b610ffde247bbc92d02047f38b Mon Sep 17 00:00:00 2001
From: jacktengg <18241664+jacktengg@users.noreply.github.com>
Date: Tue, 23 Jul 2024 23:47:59 +0800
Subject: [PATCH] [fix](agg) fix RowsProduced is not set

---
 be/src/pipeline/exec/aggregation_source_operator.cpp | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/be/src/pipeline/exec/aggregation_source_operator.cpp b/be/src/pipeline/exec/aggregation_source_operator.cpp
index 1b7a151e2af7ea..0c05c965f1f884 100644
--- a/be/src/pipeline/exec/aggregation_source_operator.cpp
+++ b/be/src/pipeline/exec/aggregation_source_operator.cpp
@@ -460,6 +460,12 @@ void AggLocalState::do_agg_limit(vectorized::Block* block, bool* eos) {
         } else {
             reached_limit(block, eos);
         }
+    } else {
+        if (auto rows = block->rows()) {
+            _num_rows_returned += rows;
+            COUNTER_UPDATE(_blocks_returned_counter, 1);
+            COUNTER_SET(_rows_returned_counter, _num_rows_returned);
+        }
     }
 }