[arrow-datafusion] branch master updated: Fix CoalesceBatches elasped_compute metric (#4664)

[dh...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

dheres pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/arrow-datafusion.git


The following commit(s) were added to refs/heads/master by this push:
     new 9f6387ec2 Fix CoalesceBatches elasped_compute metric (#4664)
9f6387ec2 is described below

commit 9f6387ec25a37f48ee15c46d721efb893e263524
Author: Jeffrey <22...@users.noreply.github.com>
AuthorDate: Mon Dec 19 00:08:48 2022 +1100

    Fix CoalesceBatches elasped_compute metric (#4664)
---
 datafusion/core/src/physical_plan/coalesce_batches.rs | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/datafusion/core/src/physical_plan/coalesce_batches.rs b/datafusion/core/src/physical_plan/coalesce_batches.rs
index 762bd9092..4bec6d73d 100644
--- a/datafusion/core/src/physical_plan/coalesce_batches.rs
+++ b/datafusion/core/src/physical_plan/coalesce_batches.rs
@@ -197,14 +197,14 @@ impl CoalesceBatchesStream {
     ) -> Poll<Option<ArrowResult<RecordBatch>>> {
         // Get a clone (uses same underlying atomic) as self gets borrowed below
         let cloned_time = self.baseline_metrics.elapsed_compute().clone();
-        // records time on drop
-        let _timer = cloned_time.timer();
 
         if self.is_closed {
             return Poll::Ready(None);
         }
         loop {
             let input_batch = self.input.poll_next_unpin(cx);
+            // records time on drop
+            let _timer = cloned_time.timer();
             match input_batch {
                 Poll::Ready(x) => match x {
                     Some(Ok(ref batch)) => {