chore(query): reduce-functions-code-expand (#17114)

* chore(query): reduce-functions-code-expand

* chore(query): reduce-functions-code-expand

* chore(query): update

* chore(query): fix

Author: sundy-li

src/query/codegen/src/writes/register.rs

@@ -175,6 +175,11 @@ impl DataBlock {
         DataBlock::new(vec![], 0)
     }
 
+    #[inline]
+    pub fn empty_with_rows(rows: usize) -> Self {
+        DataBlock::new(vec![], rows)
+    }
+
     #[inline]
     pub fn empty_with_schema(schema: DataSchemaRef) -> Self {
         let columns = schema

src/query/expression/src/block.rs

@@ -1910,7 +1910,7 @@ impl<'a, Index: ColumnIndex> ConstantFolder<'a, Index> {
                 };
 
                 if inner_expr.as_constant().is_some() {
-                    let block = DataBlock::empty();
+                    let block = DataBlock::empty_with_rows(1);
                     let evaluator = Evaluator::new(&block, self.func_ctx, self.fn_registry);
                     // Since we know the expression is constant, it'll be safe to change its column index type.
                     let cast_expr = cast_expr.project_column_ref(|_| unreachable!());
@@ -2041,7 +2041,7 @@ impl<'a, Index: ColumnIndex> ConstantFolder<'a, Index> {
                 };
 
                 if all_args_is_scalar {
-                    let block = DataBlock::empty();
+                    let block = DataBlock::empty_with_rows(1);
                     let evaluator = Evaluator::new(&block, self.func_ctx, self.fn_registry);
                     // Since we know the expression is constant, it'll be safe to change its column index type.
                     let func_expr = func_expr.project_column_ref(|_| unreachable!());
@@ -2120,7 +2120,7 @@ impl<'a, Index: ColumnIndex> ConstantFolder<'a, Index> {
                 }
 
                 if all_args_is_scalar {
-                    let block = DataBlock::empty();
+                    let block = DataBlock::empty_with_rows(1);
                     let evaluator = Evaluator::new(&block, self.func_ctx, self.fn_registry);
                     // Since we know the expression is constant, it'll be safe to change its column index type.
                     let func_expr = func_expr.project_column_ref(|_| unreachable!());
@@ -2167,7 +2167,7 @@ impl<'a, Index: ColumnIndex> ConstantFolder<'a, Index> {
                 };
 
                 if all_args_is_scalar {
-                    let block = DataBlock::empty();
+                    let block = DataBlock::empty_with_rows(1);
                     let evaluator = Evaluator::new(&block, self.func_ctx, self.fn_registry);
                     // Since we know the expression is constant, it'll be safe to change its column index type.
                     let func_expr = func_expr.project_column_ref(|_| unreachable!());

src/query/expression/src/evaluator.rs

@@ -55,6 +55,9 @@ mod kernels;
 mod property;
 mod register;
 mod register_comparison;
+#[allow(dead_code)]
+mod register_vectorize;
+
 pub mod row;
 pub mod sampler;
 pub mod schema;
@@ -75,8 +78,8 @@ pub use crate::hilbert::*;
 pub use crate::input_columns::*;
 pub use crate::kernels::*;
 pub use crate::property::*;
-pub use crate::register::*;
 pub use crate::register_comparison::*;
+pub use crate::register_vectorize::*;
 pub use crate::row::*;
 pub use crate::schema::*;
 pub use crate::utils::block_thresholds::BlockThresholds;