Adding alias to custom udf (#12219)
diff --git a/datafusion/proto/tests/cases/mod.rs b/datafusion/proto/tests/cases/mod.rs
index 1f837b7..fbb2cd8 100644
--- a/datafusion/proto/tests/cases/mod.rs
+++ b/datafusion/proto/tests/cases/mod.rs
@@ -34,12 +34,17 @@
signature: Signature,
// regex as original string
pattern: String,
+ aliases: Vec<String>,
}
impl MyRegexUdf {
fn new(pattern: String) -> Self {
let signature = Signature::exact(vec![DataType::Utf8], Volatility::Immutable);
- Self { signature, pattern }
+ Self {
+ signature,
+ pattern,
+ aliases: vec!["aggregate_udf_alias".to_string()],
+ }
}
}
@@ -67,6 +72,9 @@
) -> datafusion_common::Result<ColumnarValue> {
unimplemented!()
}
+ fn aliases(&self) -> &[String] {
+ &self.aliases
+ }
}
#[derive(Clone, PartialEq, ::prost::Message)]
