Adding alias to custom udf (#12219)

apache · Aug 29, 2024 · 650dfdc · 650dfdc
1 parent 0f16849
commit 650dfdc
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/datafusion/proto/tests/cases/mod.rs b/datafusion/proto/tests/cases/mod.rs
@@ -34,12 +34,17 @@ struct MyRegexUdf {
     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 @@ impl ScalarUDFImpl for MyRegexUdf {
     ) -> datafusion_common::Result<ColumnarValue> {
         unimplemented!()
     }
+    fn aliases(&self) -> &[String] {
+        &self.aliases
+    }
 }
 
 #[derive(Clone, PartialEq, ::prost::Message)]