feat: print more first poll

GreptimeTeam · Jul 5, 2024 · a563c3c · a563c3c
1 parent 79165f5
commit a563c3c
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/src/mito2/src/read/seq_scan.rs b/src/mito2/src/read/seq_scan.rs
@@ -254,6 +254,8 @@ impl SeqScan {
         let semaphore = self.semaphore.clone();
         let partition_ranges = self.properties.partitions[partition].clone();
         let stream = try_stream! {
+            let first_poll = stream_ctx.query_start.elapsed();
+
             for partition_range in partition_ranges {
                 let maybe_reader =
                     Self::build_merge_reader(&stream_ctx, partition_range.identifier, semaphore.clone(), &mut metrics)
@@ -287,10 +289,11 @@ impl SeqScan {
                 metrics.observe_metrics_on_finish();
 
                 debug!(
-                    "Seq scan finished, region_id: {:?}, partition: {}, metrics: {:?}",
+                    "Seq scan finished, region_id: {:?}, partition: {}, metrics: {:?}, first_poll: {:?}",
                     stream_ctx.input.mapper.metadata().region_id,
                     partition,
                     metrics,
+                    first_poll,
                 );
             }
         };

diff --git a/src/mito2/src/read/unordered_scan.rs b/src/mito2/src/read/unordered_scan.rs
@@ -146,6 +146,8 @@ impl RegionScanner for UnorderedScan {
         };
         let stream_ctx = self.stream_ctx.clone();
         let stream = try_stream! {
+            let first_poll = stream_ctx.query_start.elapsed();
+
             let mut parts = stream_ctx.parts.lock().await;
             maybe_init_parts(&mut parts, &stream_ctx.input, &mut metrics)
                 .await
@@ -196,8 +198,8 @@ impl RegionScanner for UnorderedScan {
             metrics.total_cost = query_start.elapsed();
             metrics.observe_metrics_on_finish();
             debug!(
-                "Unordered scan partition {} finished, region_id: {}, metrics: {:?}, reader_metrics: {:?}",
-                partition, mapper.metadata().region_id, metrics, reader_metrics
+                "Unordered scan partition {} finished, region_id: {}, metrics: {:?}, reader_metrics: {:?}, first_poll: {:?}",
+                partition, mapper.metadata().region_id, metrics, reader_metrics, first_poll,
             );
         };
         let stream = Box::pin(RecordBatchStreamWrapper::new(