You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@datafusion.apache.org by al...@apache.org on 2025/05/14 20:55:24 UTC
(datafusion) branch main updated: fix: track coalescer's consumption (#16048)
This is an automated email from the ASF dual-hosted git repository.
alamb pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/datafusion.git
The following commit(s) were added to refs/heads/main by this push:
new d5ecaca33c fix: track coalescer's consumption (#16048)
d5ecaca33c is described below
commit d5ecaca33c8472a6520e081ac3c9ad3dac098caf
Author: Ruihang Xia <wa...@gmail.com>
AuthorDate: Thu May 15 04:55:20 2025 +0800
fix: track coalescer's consumption (#16048)
Signed-off-by: Ruihang Xia <wa...@gmail.com>
---
datafusion/physical-plan/src/coalesce_batches.rs | 2 ++
1 file changed, 2 insertions(+)
diff --git a/datafusion/physical-plan/src/coalesce_batches.rs b/datafusion/physical-plan/src/coalesce_batches.rs
index a0dd7371b4..f35231fb6a 100644
--- a/datafusion/physical-plan/src/coalesce_batches.rs
+++ b/datafusion/physical-plan/src/coalesce_batches.rs
@@ -354,6 +354,7 @@ impl CoalesceBatchesStream {
}
}
CoalesceBatchesStreamState::ReturnBuffer => {
+ let _timer = cloned_time.timer();
// Combine buffered batches into one batch and return it.
let batch = self.coalescer.finish_batch()?;
// Set to pull state for the next iteration.
@@ -366,6 +367,7 @@ impl CoalesceBatchesStream {
// If buffer is empty, return None indicating the stream is fully consumed.
Poll::Ready(None)
} else {
+ let _timer = cloned_time.timer();
// If the buffer still contains batches, prepare to return them.
let batch = self.coalescer.finish_batch()?;
Poll::Ready(Some(Ok(batch)))
---------------------------------------------------------------------
To unsubscribe, e-mail: commits-unsubscribe@datafusion.apache.org
For additional commands, e-mail: commits-help@datafusion.apache.org