Skip to content
Closed
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
10 changes: 7 additions & 3 deletions native/spark-expr/src/comet_scalar_funcs.rs
Original file line number Diff line number Diff line change
Expand Up @@ -22,9 +22,9 @@ use crate::math_funcs::log::spark_log;
use crate::math_funcs::modulo_expr::spark_modulo;
use crate::{
spark_ceil, spark_decimal_div, spark_decimal_integral_div, spark_floor, spark_isnan,
spark_lpad, spark_make_decimal, spark_read_side_padding, spark_round, spark_rpad, spark_unhex,
spark_unscaled_value, EvalMode, SparkContains, SparkDateDiff, SparkDateTrunc, SparkMakeDate,
SparkSizeFunc,
spark_lpad, spark_make_decimal, spark_map_sort, spark_read_side_padding, spark_round,
spark_rpad, spark_unhex, spark_unscaled_value, EvalMode, SparkContains, SparkDateDiff,
SparkDateTrunc, SparkMakeDate, SparkSizeFunc,
};
use arrow::datatypes::DataType;
use datafusion::common::{DataFusionError, Result as DataFusionResult};
Expand Down Expand Up @@ -186,6 +186,10 @@ pub fn create_comet_physical_fun_with_eval_mode(
let func = Arc::new(crate::string_funcs::spark_split);
make_comet_scalar_udf!("split", func, without data_type)
}
"map_sort" => {
let func = Arc::new(spark_map_sort);
make_comet_scalar_udf!("map_sort", func, without data_type)
}
_ => registry.udf(fun_name).map_err(|e| {
DataFusionError::Execution(format!(
"Function {fun_name} not found in the registry: {e}",
Expand Down
2 changes: 2 additions & 0 deletions native/spark-expr/src/lib.rs
Original file line number Diff line number Diff line change
Expand Up @@ -45,6 +45,7 @@ mod array_funcs;
mod comet_scalar_funcs;
pub mod hash_funcs;

mod map_funcs;
mod string_funcs;

mod datetime_funcs;
Expand All @@ -63,6 +64,7 @@ mod nondetermenistic_funcs;
pub use array_funcs::*;
pub use conditional_funcs::*;
pub use conversion_funcs::*;
pub use map_funcs::*;
pub use nondetermenistic_funcs::*;

pub use comet_scalar_funcs::{
Expand Down
Loading