Add debug print statement to DataFrame.collect for improved traceability

kosiew · Jan 13, 2025 · 52c57ce · 52c57ce
1 parent a8d4077
commit 52c57ce
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/datafusion/core/src/dataframe/mod.rs b/datafusion/core/src/dataframe/mod.rs
@@ -1208,6 +1208,7 @@ impl DataFrame {
     /// # }
     /// ```
     pub async fn collect(self) -> Result<Vec<RecordBatch>> {
+        println!("==> Datafrrame.collect");
         let task_ctx = Arc::new(self.task_ctx());
         let plan = self.create_physical_plan().await?;
         collect(plan, task_ctx).await

diff --git a/datafusion/core/src/dataframe/parquet.rs b/datafusion/core/src/dataframe/parquet.rs
@@ -103,7 +103,7 @@ impl DataFrame {
         )?
         .build()?;
 
-        println!("==> Built copy_to logical plan");
+        println!("==> ==> Built copy_to logical plan");
 
         let result = DataFrame {
             session_state: self.session_state,