chore: add traces

src/mito2/src/read/scan_region.rs

@@ -651,6 +651,7 @@ impl ScanInput {
     }
 
     /// Prunes a file to scan and returns the builder to build readers.
+    #[tracing::instrument(level = tracing::Level::DEBUG, skip_all)]
     async fn prune_file(
         &self,
         file_index: usize,
@@ -809,6 +810,7 @@ impl StreamContext {
     }
 
     /// Creates file ranges to scan.
+    #[tracing::instrument(level = tracing::Level::DEBUG, skip_all)]
     pub(crate) async fn build_file_ranges(
         &self,
         index: RowGroupIndex,
@@ -883,6 +885,7 @@ impl RangeBuilderList {
     }
 
     /// Builds file ranges to read the row group at `index`.
+    #[tracing::instrument(level = tracing::Level::DEBUG, skip_all)]
     async fn build_file_ranges(
         &self,
         input: &ScanInput,

src/mito2/src/read/scan_util.rs

@@ -18,7 +18,7 @@ use std::sync::{Arc, Mutex};
 use std::time::{Duration, Instant};
 
 use async_stream::try_stream;
-use common_telemetry::debug;
+use common_telemetry::{debug, tracing};
 use futures::Stream;
 use store_api::storage::RegionId;
 
@@ -148,6 +148,7 @@ pub(crate) fn scan_mem_ranges(
 }
 
 /// Scans file ranges at `index`.
+#[tracing::instrument(level = tracing::Level::DEBUG, skip_all)]
 pub(crate) fn scan_file_ranges(
     stream_ctx: Arc<StreamContext>,
     part_metrics: PartitionMetrics,

src/mito2/src/sst/parquet/file_range.rs

@@ -19,7 +19,7 @@ use std::ops::BitAnd;
 use std::sync::Arc;
 
 use api::v1::{OpType, SemanticType};
-use common_telemetry::error;
+use common_telemetry::{error, tracing};
 use datatypes::arrow::array::BooleanArray;
 use datatypes::arrow::buffer::BooleanBuffer;
 use parquet::arrow::arrow_reader::RowSelection;
@@ -80,6 +80,7 @@ impl FileRange {
     }
 
     /// Returns a reader to read the [FileRange].
+    #[tracing::instrument(level = tracing::Level::DEBUG, skip_all)]
     pub(crate) async fn reader(
         &self,
         selector: Option<TimeSeriesRowSelector>,

src/mito2/src/sst/parquet/reader.rs

@@ -22,7 +22,7 @@ use std::time::{Duration, Instant};
 use api::v1::SemanticType;
 use async_trait::async_trait;
 use common_recordbatch::filter::SimpleFilterEvaluator;
-use common_telemetry::{debug, warn};
+use common_telemetry::{debug, tracing, warn};
 use common_time::range::TimestampRange;
 use common_time::timestamp::TimeUnit;
 use common_time::Timestamp;
@@ -183,6 +183,7 @@ impl ParquetReaderBuilder {
     /// Builds a [FileRangeContext] and collects row groups to read.
     ///
     /// This needs to perform IO operation.
+    #[tracing::instrument(level = tracing::Level::DEBUG, skip_all)]
     pub(crate) async fn build_reader_input(
         &self,
         metrics: &mut ReaderMetrics,
@@ -301,6 +302,7 @@ impl ParquetReaderBuilder {
     }
 
     /// Reads parquet metadata of specific file.
+    #[tracing::instrument(level = tracing::Level::DEBUG, skip_all)]
     async fn read_parquet_metadata(
         &self,
         file_path: &str,
@@ -336,6 +338,7 @@ impl ParquetReaderBuilder {
     }
 
     /// Computes row groups to read, along with their respective row selections.
+    #[tracing::instrument(level = tracing::Level::DEBUG, skip_all)]
     async fn row_groups_to_read(
         &self,
         read_format: &ReadFormat,
@@ -381,6 +384,7 @@ impl ParquetReaderBuilder {
     }
 
     /// Prunes row groups by fulltext index. Returns `true` if the row groups are pruned.
+    #[tracing::instrument(level = tracing::Level::DEBUG, skip_all)]
     async fn prune_row_groups_by_fulltext_index(
         &self,
         row_group_size: usize,
@@ -461,6 +465,7 @@ impl ParquetReaderBuilder {
     /// TODO(zhongzc): Devise a mechanism to enforce the non-use of indices
     /// as an escape route in case of index issues, and it can be used to test
     /// the correctness of the index.
+    #[tracing::instrument(level = tracing::Level::DEBUG, skip_all)]
     async fn prune_row_groups_by_inverted_index(
         &self,
         row_group_size: usize,
@@ -528,6 +533,7 @@ impl ParquetReaderBuilder {
     }
 
     /// Prunes row groups by min-max index.
+    #[tracing::instrument(level = tracing::Level::DEBUG, skip_all)]
     fn prune_row_groups_by_minmax(
         &self,
         read_format: &ReadFormat,
@@ -869,6 +875,7 @@ impl RowGroupReaderBuilder {
     }
 
     /// Builds a [ParquetRecordBatchReader] to read the row group at `row_group_idx`.
+    #[tracing::instrument(level = tracing::Level::DEBUG, skip_all)]
     pub(crate) async fn build(
         &self,
         row_group_idx: usize,

src/mito2/src/sst/parquet/row_group.rs

@@ -18,6 +18,7 @@ use std::ops::Range;
 use std::sync::Arc;
 
 use bytes::{Buf, Bytes};
+use common_telemetry::tracing;
 use object_store::ObjectStore;
 use parquet::arrow::arrow_reader::{RowGroups, RowSelection};
 use parquet::arrow::ProjectionMask;
@@ -97,6 +98,7 @@ impl<'a> InMemoryRowGroup<'a> {
     }
 
     /// Fetches the necessary column data into memory
+    #[tracing::instrument(level = tracing::Level::DEBUG, skip_all)]
     pub async fn fetch(
         &mut self,
         projection: &ProjectionMask,