From 8137b8ff3decc6ddf3fa43196c7805c2eb7dbe5d Mon Sep 17 00:00:00 2001
From: evenyag <realevenyag@gmail.com>
Date: Wed, 6 Nov 2024 15:25:49 +0800
Subject: [PATCH] chore: more logs

---
 src/query/src/part_sort.rs | 30 ++++++++++++++++++++++++++++++
 1 file changed, 30 insertions(+)

diff --git a/src/query/src/part_sort.rs b/src/query/src/part_sort.rs
index 74facfdeaa..313f7d454a 100644
--- a/src/query/src/part_sort.rs
+++ b/src/query/src/part_sort.rs
@@ -394,6 +394,15 @@ impl PartSortStream {
         if self.buffer.is_empty() {
             return Ok(DfRecordBatch::new_empty(self.schema.clone()));
         }
+
+        common_telemetry::info!(
+            "[PartSortStream] Region {} Partition {} part index {} sort {} rows",
+            self.region_id,
+            self.partition,
+            self.cur_part_idx,
+            self.buffer.iter().map(|b| b.len()).sum::<usize>(),
+        );
+
         let mut sort_columns = Vec::with_capacity(self.buffer.len());
         let mut opt = None;
         for batch in self.buffer.iter() {
@@ -492,6 +501,13 @@ impl PartSortStream {
 
         let next_range_idx = self.try_find_next_range(&sort_column)?;
         let Some(idx) = next_range_idx else {
+            common_telemetry::info!(
+                "[PartSortStream] Region {} Partition {} part index {} push {} rows",
+                self.region_id,
+                self.partition,
+                self.cur_part_idx,
+                batch.num_rows(),
+            );
             self.buffer.push(batch);
             // keep polling input for next batch
             return Ok(None);
@@ -500,6 +516,13 @@ impl PartSortStream {
         let this_range = batch.slice(0, idx);
         let remaining_range = batch.slice(idx, batch.num_rows() - idx);
         if this_range.num_rows() != 0 {
+            common_telemetry::info!(
+                "[PartSortStream] Region {} Partition {} part index {} push {} rows",
+                self.region_id,
+                self.partition,
+                self.cur_part_idx,
+                batch.num_rows(),
+            );
             self.buffer.push(this_range);
         }
         // mark end of current PartitionRange
@@ -515,6 +538,13 @@ impl PartSortStream {
             // remaining batch is within the current partition range
             // push to the buffer and continue polling
             if remaining_range.num_rows() != 0 {
+                common_telemetry::info!(
+                    "[PartSortStream] Region {} Partition {} part index {} push {} rows",
+                    self.region_id,
+                    self.partition,
+                    self.cur_part_idx,
+                    batch.num_rows(),
+                );
                 self.buffer.push(remaining_range);
             }
         }