fix

Signed-off-by: coldWater <forsaken628@gmail.com>

Author: forsaken628

src/query/service/src/pipelines/builders/builder_sort.rs

@@ -258,7 +258,8 @@ impl SortPipelineBuilder {
             Ok(ProcessorPtr::create(builder.build_collect(input, output)?))
         })?;
 
-        let state = SortSampleState::new(inputs, max_threads, self.schema.clone(), max_block_size);
+        let state =
+            SortSampleState::new(inputs, max_threads, builder.inner_schema(), max_block_size);
 
         builder.add_shuffle(pipeline, state.clone())?;
 

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

@@ -218,7 +218,7 @@ impl TransformSortBuilder {
     }
 
     fn new_base(&self) -> Base {
-        let schema = add_order_field(self.schema.clone(), &self.sort_desc);
+        let schema = self.inner_schema();
         let sort_row_offset = schema.fields().len() - 1;
         Base {
             sort_row_offset,
@@ -228,6 +228,10 @@ impl TransformSortBuilder {
         }
     }
 
+    pub fn inner_schema(&self) -> DataSchemaRef {
+        add_order_field(self.schema.clone(), &self.sort_desc)
+    }
+
     pub fn add_shuffle(&self, pipeline: &mut Pipeline, state: Arc<SortSampleState>) -> Result<()> {
         use std::sync::atomic;
         let i = atomic::AtomicUsize::new(0);

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

@@ -251,7 +251,6 @@ impl SortSampleState {
 struct StateInner {
     // target partitions
     partitions: usize,
-    // schema for bounds DataBlock
     schema: DataSchemaRef,
     partial: Vec<Option<Bounds>>,
     bounds: Option<Bounds>,