This is an automated email from the ASF dual-hosted git repository.
xudong963 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 1fedb4e000 Counting elapsed_compute in BoundedWindowAggExec (#14869)
1fedb4e000 is described below
commit 1fedb4e000293e3997b477d87d575f3a5453171e
Author: Yongting You <2010you...@gmail.com>
AuthorDate: Tue Feb 25 21:49:18 2025 +0800
Counting elapsed_compute in BoundedWindowAggExec (#14869)
---
datafusion/physical-plan/src/windows/bounded_window_agg_exec.rs | 7 +++++++
1 file changed, 7 insertions(+)
diff --git a/datafusion/physical-plan/src/windows/bounded_window_agg_exec.rs
b/datafusion/physical-plan/src/windows/bounded_window_agg_exec.rs
index c78c870ff3..0d9c58b3bf 100644
--- a/datafusion/physical-plan/src/windows/bounded_window_agg_exec.rs
+++ b/datafusion/physical-plan/src/windows/bounded_window_agg_exec.rs
@@ -998,8 +998,13 @@ impl BoundedWindowAggStream {
return Poll::Ready(None);
}
+ let elapsed_compute = self.baseline_metrics.elapsed_compute().clone();
match ready!(self.input.poll_next_unpin(cx)) {
Some(Ok(batch)) => {
+ // Start the timer for compute time within this operator. It
will be
+ // stopped when dropped.
+ let _timer = elapsed_compute.timer();
+
self.search_mode.update_partition_batch(
&mut self.input_buffer,
batch,
@@ -1013,6 +1018,8 @@ impl BoundedWindowAggStream {
}
Some(Err(e)) => Poll::Ready(Some(Err(e))),
None => {
+ let _timer = elapsed_compute.timer();
+
self.finished = true;
for (_, partition_batch_state) in
self.partition_buffers.iter_mut() {
partition_batch_state.is_end = true;
---------------------------------------------------------------------
To unsubscribe, e-mail: commits-unsubscr...@datafusion.apache.org
For additional commands, e-mail: commits-h...@datafusion.apache.org
