diff --git a/rust/lance-index/src/scalar/btree.rs b/rust/lance-index/src/scalar/btree.rs index f85d92d3f3e..18c0794f17b 100644 --- a/rust/lance-index/src/scalar/btree.rs +++ b/rust/lance-index/src/scalar/btree.rs @@ -62,7 +62,7 @@ use rangemap::RangeInclusiveMap; use roaring::RoaringBitmap; use serde::{Deserialize, Serialize, Serializer}; use snafu::location; -use tracing::info; +use tracing::{info, instrument}; mod flat; @@ -1044,6 +1044,7 @@ impl BTreeIndex { .await } + #[instrument(level = "debug", skip_all)] async fn read_page( &self, page_number: u32, @@ -1090,6 +1091,7 @@ impl BTreeIndex { } } + #[instrument(level = "debug", skip_all)] fn try_from_serialized( data: RecordBatch, store: Arc, diff --git a/rust/lance-index/src/scalar/btree/flat.rs b/rust/lance-index/src/scalar/btree/flat.rs index 1da09425528..4cd029ea717 100644 --- a/rust/lance-index/src/scalar/btree/flat.rs +++ b/rust/lance-index/src/scalar/btree/flat.rs @@ -18,6 +18,7 @@ use lance_core::utils::address::RowAddress; use lance_core::utils::mask::{NullableRowAddrSet, RowAddrTreeMap}; use lance_core::Result; use roaring::RoaringBitmap; +use tracing::instrument; use crate::metrics::MetricsCollector; use crate::scalar::btree::BTREE_VALUES_COLUMN; @@ -46,6 +47,7 @@ impl DeepSizeOf for FlatIndex { } impl FlatIndex { + #[instrument(name = "FlatIndex::try_new", level = "debug", skip_all)] pub fn try_new(data: RecordBatch) -> Result { // Sort by row id to make bitmap construction more efficient let data = data.sort_by_column(IDS_COL_IDX, None)?; diff --git a/rust/lance/src/index.rs b/rust/lance/src/index.rs index 589c7c26b93..1431d5687a8 100644 --- a/rust/lance/src/index.rs +++ b/rust/lance/src/index.rs @@ -1268,6 +1268,7 @@ impl DatasetIndexInternalExt for Dataset { } } + #[instrument(level = "debug", skip_all)] async fn open_scalar_index( &self, column: &str,