This is an automated email from the ASF dual-hosted git repository.

richox pushed a commit to branch dev-v6.0.0-parallel-scan-kdev-build
in repository https://gitbox.apache.org/repos/asf/auron.git

commit 83e87e1e4f456d812e8686dac30a0a8c17b3c1b7
Author: Zhang Li <[email protected]>
AuthorDate: Tue Aug 12 19:58:24 2025 +0800

    fix execute_projected_with_key_rows_output() (#1156)
    
    Co-authored-by: zhangli20 <[email protected]>
---
 .../datafusion-ext-plans/src/common/execution_context.rs         | 4 ++--
 .../datafusion-ext-plans/src/common/row_null_checker.rs          | 9 ---------
 2 files changed, 2 insertions(+), 11 deletions(-)

diff --git a/native-engine/datafusion-ext-plans/src/common/execution_context.rs 
b/native-engine/datafusion-ext-plans/src/common/execution_context.rs
index f21fc7d3..3da6526e 100644
--- a/native-engine/datafusion-ext-plans/src/common/execution_context.rs
+++ b/native-engine/datafusion-ext-plans/src/common/execution_context.rs
@@ -316,12 +316,12 @@ impl ExecutionContext {
             );
         }
 
-        let output_schema = self.output_schema();
+        let input_schema = input.schema();
         let key_converter = RowConverter::new(
             keys.iter()
                 .map(|k| {
                     Ok(SortField::new_with_options(
-                        k.expr.data_type(&output_schema)?,
+                        k.expr.data_type(&input_schema)?,
                         k.options,
                     ))
                 })
diff --git a/native-engine/datafusion-ext-plans/src/common/row_null_checker.rs 
b/native-engine/datafusion-ext-plans/src/common/row_null_checker.rs
index 0a168635..dda5cc9a 100644
--- a/native-engine/datafusion-ext-plans/src/common/row_null_checker.rs
+++ b/native-engine/datafusion-ext-plans/src/common/row_null_checker.rs
@@ -438,15 +438,6 @@ mod tests {
 
     use super::*;
 
-    // Helper function to create a SortField for testing
-    fn create_test_sort_field(data_type: DataType, nulls_first: bool) -> 
SortField {
-        let sort_options = SortOptions {
-            descending: false,
-            nulls_first,
-        };
-        SortField::new_with_options(data_type, sort_options)
-    }
-
     #[test]
     fn test_primitive_null_detection() {
         // Create a SortField with Int32 data type

Reply via email to