From c310aee032a6ac4b885cc5a65b6323e32b41ee0f Mon Sep 17 00:00:00 2001
From: Will Jones <willjones127@gmail.com>
Date: Fri, 13 Dec 2024 11:46:07 -0800
Subject: [PATCH] feat: enable tracing for object storage (#3244)

We have this on by default for local fs and in-memory. We also want this
for object storage.
---
 rust/lance-io/src/object_store.rs | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/rust/lance-io/src/object_store.rs b/rust/lance-io/src/object_store.rs
index 80bfea8726..358ff4bf5d 100644
--- a/rust/lance-io/src/object_store.rs
+++ b/rust/lance-io/src/object_store.rs
@@ -886,7 +886,7 @@ async fn configure_store(
             let store = builder.build()?;
 
             Ok(ObjectStore {
-                inner: Arc::new(store),
+                inner: Arc::new(store).traced(),
                 scheme: String::from(url.scheme()),
                 block_size: 64 * 1024,
                 use_constant_size_upload_parts,
@@ -902,7 +902,7 @@ async fn configure_store(
                 builder = builder.with_config(key, value);
             }
             let store = builder.build()?;
-            let store = Arc::new(store);
+            let store = Arc::new(store).traced();
 
             Ok(ObjectStore {
                 inner: store,
@@ -917,7 +917,7 @@ async fn configure_store(
         "az" => {
             storage_options.with_env_azure();
             let (store, _) = parse_url_opts(&url, storage_options.as_azure_options())?;
-            let store = Arc::new(store);
+            let store = Arc::new(store).traced();
 
             Ok(ObjectStore {
                 inner: store,