feat: add optional instrumentation to readers (#1431)

Add optional instrumentation using the `tracing` crate for `vortex-io`
and `vortex-file`. Disabled by default.

Part of some work to make it easier to collect deep profiles of
performance on object storage. Example output when hooked up to a
subscriber:


<img width="1265" alt="image"
src="https://github.com/user-attachments/assets/352564c6-9c99-408b-bb63-4e6c5bed8efd">

Cargo.toml

@@ -135,6 +135,7 @@ tar = "0.4"
 tempfile = "3"
 thiserror = "2.0.0"
 tokio = "1.37.0"
+tracing = "0.1"
 uninit = "0.6.2"
 url = "2"
 uuid = "1.8.0"

vortex-file/Cargo.toml

@@ -29,6 +29,7 @@ tokio = { workspace = true, features = [
     "fs",
     "rt-multi-thread",
 ] }
+tracing = { workspace = true, optional = true }
 vortex-array = { workspace = true }
 vortex-buffer = { workspace = true }
 vortex-dtype = { workspace = true, features = ["flatbuffers"] }
@@ -57,3 +58,4 @@ object_store = [
     "vortex-error/object_store",
     "vortex-io/object_store",
 ]
+tracing = ["dep:tracing", "vortex-io/tracing"]

vortex-file/src/read/stream.rs

@@ -298,6 +298,7 @@ impl<R: VortexReadAt + Unpin> VortexFileArrayStream<R> {
     }
 }
 
+#[cfg_attr(feature = "tracing", tracing::instrument(skip_all))]
 async fn read_ranges<R: VortexReadAt>(
     reader: R,
     ranges: Vec<MessageLocator>,

vortex-io/Cargo.toml

@@ -17,6 +17,7 @@ categories.workspace = true
 bytes = { workspace = true }
 compio = { workspace = true, features = ["bytes", "macros"], optional = true }
 tokio = { workspace = true, features = ["fs"], optional = true }
+tracing = { workspace = true, optional = true }
 futures = { workspace = true, features = ["std"] }
 futures-util = { workspace = true }
 flume = { workspace = true }
@@ -31,8 +32,6 @@ tokio = { workspace = true, features = ["full"] }
 
 [features]
 futures = ["futures-util/io"]
-compio = ["dep:compio"]
-tokio = ["dep:tokio"]
 object_store = ["dep:object_store", "vortex-error/object_store"]
 
 [lints]

vortex-io/src/object_store.rs

@@ -65,6 +65,7 @@ impl ObjectStoreReadAt {
 }
 
 impl VortexReadAt for ObjectStoreReadAt {
+    #[cfg_attr(feature = "tracing", tracing::instrument(skip(self)))]
     fn read_byte_range(
         &self,
         pos: u64,
@@ -82,6 +83,7 @@ impl VortexReadAt for ObjectStoreReadAt {
         })
     }
 
+    #[cfg_attr(feature = "tracing", tracing::instrument(skip(self)))]
     fn size(&self) -> impl Future<Output = u64> + 'static {
         let object_store = self.object_store.clone();
         let location = self.location.clone();

vortex-io/src/tokio.rs

@@ -62,6 +62,7 @@ impl Deref for TokioFile {
 }
 
 impl VortexReadAt for TokioFile {
+    #[cfg_attr(feature = "tracing", tracing::instrument(skip(self)))]
     fn read_byte_range(
         &self,
         pos: u64,
@@ -79,6 +80,7 @@ impl VortexReadAt for TokioFile {
         }
     }
 
+    #[cfg_attr(feature = "tracing", tracing::instrument(skip(self)))]
     fn size(&self) -> impl Future<Output = u64> + 'static {
         let this = self.clone();