fix

Author: forsaken628

src/query/service/src/pipelines/processors/transforms/sort/bounds.rs

@@ -48,11 +48,8 @@ impl Bounds {
         Ok(Bounds(vec![block.get_last_column().clone()]))
     }
 
-    pub fn merge<R: Rows>(vector: Vec<Bounds>, batch_rows: usize) -> Result<Self> {
-        let mut vector: Vec<_> = vector
-            .into_iter()
-            .filter(|bounds| !bounds.is_empty())
-            .collect();
+    pub fn merge<R: Rows>(mut vector: Vec<Bounds>, batch_rows: usize) -> Result<Self> {
+        debug_assert!(vector.iter().all(|bounds| !bounds.is_empty()));
         match vector.len() {
             0 => Ok(Bounds(vec![])),
             1 => Ok(vector.pop().unwrap()),

src/query/service/src/pipelines/processors/transforms/sort/sort_broadcast.rs

@@ -138,7 +138,9 @@ impl<R: Rows + 'static> HookTransform for TransformSortBoundBroadcast<R> {
         let bounds = Bounds::merge::<R>(
             self.input_data
                 .iter_mut()
-                .map(|meta| std::mem::take(&mut meta.bounds))
+                .filter_map(|meta| {
+                    (!meta.bounds.is_empty()).then(|| std::mem::take(&mut meta.bounds))
+                })
                 .collect(),
             self.state.batch_rows,
         )?;
@@ -169,7 +171,10 @@ impl<R: Rows + 'static> HookTransform for TransformSortBoundBroadcast<R> {
             }))
             .await?;
 
-        let bounds_vec = global.into_iter().map(|meta| meta.bounds).collect();
+        let bounds_vec = global
+            .into_iter()
+            .filter_map(|meta| (!meta.bounds.is_empty()).then_some(meta.bounds))
+            .collect();
         self.output_data = Some(SortCollectedMeta {
             bounds: Bounds::merge::<R>(bounds_vec, self.state.batch_rows)?.dedup::<R>(),
             ..local

tests/sqllogictests/suites/ee/02_ee_aggregating_index/02_0000_async_agg_index_base.test

@@ -414,7 +414,7 @@ Logout 2 2 5.0
 query T
 explain select event_name, user_id, max(user_id), avg(id) from t where user_id > 1 group by event_name, user_id order by event_name, user_id desc
 ----
-Sort
+Sort(Single)
 ├── output columns: [max(user_id) (#3), t.event_name (#2), t.user_id (#1), sum(id) / if(count(id) = 0, 1, count(id)) (#6)]
 ├── sort keys: [event_name ASC NULLS LAST, user_id DESC NULLS LAST]
 ├── estimated rows: 6.00