refactor: improve timer handling

apache · Jan 13, 2023 · 1ceff4c · 1ceff4c
1 parent b2af7b0
commit 1ceff4c
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/datafusion/core/src/physical_plan/repartition/mod.rs b/datafusion/core/src/physical_plan/repartition/mod.rs
@@ -166,7 +166,7 @@ impl BatchPartitioner {
                     num_partitions: partitions,
                     hash_buffer,
                 } => {
-                    let mut timer = self.timer.timer();
+                    let timer = self.timer.timer();
 
                     let arrays = exprs
                         .iter()
@@ -210,8 +210,8 @@ impl BatchPartitioner {
                             let batch =
                                 RecordBatch::try_new(batch.schema(), columns).unwrap();
 
-                            timer.stop();
-                            timer.restart();
+                            // bind timer so it drops w/ this iterator
+                            let _ = &timer;
 
                             Ok((partition, batch))
                         });