fix accounting

alamb · Oct 10, 2024 · 8e876bf · 8e876bf
1 parent c7aa11f
commit 8e876bf
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/datafusion/functions-aggregate/src/min_max/min_max_bytes.rs b/datafusion/functions-aggregate/src/min_max/min_max_bytes.rs
@@ -497,14 +497,13 @@ impl MinMaxBytesState {
                 )
             }
             EmitTo::First(n) => {
-                let min_maxes = self.min_max.drain(..n).collect();
-                let data_capacity: usize = self
-                    .min_max
+                let first_min_maxes: Vec<_> = self.min_max.drain(..n).collect();
+                let first_data_capacity: usize = first_min_maxes
                     .iter()
                     .map(|opt| opt.as_ref().map(|s| s.len()).unwrap_or(0))
                     .sum();
-                self.total_data_bytes -= data_capacity;
-                (data_capacity, min_maxes)
+                self.total_data_bytes -= first_data_capacity;
+                (first_data_capacity, first_min_maxes)
             }
         }
     }