jayzhan211 commented on code in PR #14440: URL: https://github.com/apache/datafusion/pull/14440#discussion_r1961784030
########## datafusion/expr-common/src/signature.rs: ########## @@ -466,6 +551,186 @@ fn get_data_types(native_type: &NativeType) -> Vec<DataType> { } } +/// Represents type coercion rules for function arguments, specifying both the desired type +/// and optional implicit coercion rules for source types. +/// +/// # Examples +/// +/// ``` +/// use datafusion_expr_common::signature::{Coercion, TypeSignatureClass}; +/// use datafusion_common::types::{NativeType, logical_binary, logical_string}; +/// +/// // Exact coercion that only accepts timestamp types +/// let exact = Coercion::new_exact(TypeSignatureClass::Timestamp); +/// +/// // Implicit coercion that accepts string types but can coerce from binary types +/// let implicit = Coercion::new_implicit( +/// TypeSignatureClass::Native(logical_string()), +/// vec![TypeSignatureClass::Native(logical_binary())], +/// NativeType::String +/// ); +/// ``` +/// +/// There are two variants: +/// +/// * `Exact` - Only accepts arguments that exactly match the desired type +/// * `Implicit` - Accepts the desired type and can coerce from specified source types +#[derive(Debug, Clone, Eq, PartialOrd)] +pub enum Coercion { + /// Coercion that only accepts arguments exactly matching the desired type. + Exact { + /// The required type for the argument + desired_type: TypeSignatureClass, + }, + + /// Coercion that accepts the desired type and can implicitly coerce from other types. + Implicit { + /// The primary desired type for the argument + desired_type: TypeSignatureClass, + /// Rules for implicit coercion from other types + implicit_coercion: ImplicitCoercion, + }, Review Comment: https://github.com/apache/datafusion/issues/14296 If we start to bring user-defined coercion rule to others operations, not only function, the user need to define `coercion rule` for other ops like binary comparison as well. What you are suggesting is limited the extensibility of coercion rule executed in DataFusion -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: github-unsubscr...@datafusion.apache.org For queries about this service, please contact Infrastructure at: us...@infra.apache.org --------------------------------------------------------------------- To unsubscribe, e-mail: github-unsubscr...@datafusion.apache.org For additional commands, e-mail: github-h...@datafusion.apache.org