Skip to content

Commit 02fc169

Browse files
committed
add tracing to file-source
1 parent a42207c commit 02fc169

File tree

1 file changed

+18
-9
lines changed

1 file changed

+18
-9
lines changed

datafusion_iceberg/src/table.rs

Lines changed: 18 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -565,15 +565,22 @@ async fn table_scan(
565565
statistics
566566
};
567567

568-
let file_source = Arc::new(
569-
if let Some(physical_predicate) = physical_predicate.clone() {
570-
ParquetSource::default()
571-
.with_predicate(physical_predicate)
572-
.with_pushdown_filters(true)
573-
} else {
574-
ParquetSource::default()
575-
},
576-
);
568+
let file_source = {
569+
let physical_predicate = physical_predicate.clone();
570+
async move {
571+
Arc::new(
572+
if let Some(physical_predicate) = physical_predicate.clone() {
573+
ParquetSource::default()
574+
.with_predicate(physical_predicate)
575+
.with_pushdown_filters(true)
576+
} else {
577+
ParquetSource::default()
578+
},
579+
)
580+
}
581+
.instrument(tracing::debug_span!("datafusion_iceberg::file_source"))
582+
.await
583+
};
577584

578585
// Create plan for every partition with delete files
579586
let mut plans = stream::iter(equality_delete_file_groups.into_iter())
@@ -1613,6 +1620,8 @@ mod tests {
16131620
if let Tabular::Table(table) = table.tabular.read().await.deref() {
16141621
assert_eq!(table.manifests(None, None).await.unwrap().len(), 2);
16151622
};
1623+
1624+
panic!()
16161625
}
16171626

16181627
#[tokio::test]

0 commit comments

Comments
 (0)