[Coral-Trino] Map some UDFs to Trino UDF (#524)

linkedin · Aug 21, 2024 · 9e79488 · 9e79488
1 parent aa086d1
commit 9e79488
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/...-trino/src/main/java/com/linkedin/coral/trino/rel2trino/CoralToTrinoSqlCallConverter.java b/...-trino/src/main/java/com/linkedin/coral/trino/rel2trino/CoralToTrinoSqlCallConverter.java
@@ -122,6 +122,15 @@ protected SqlCall transform(SqlCall sqlCall) {
             "com.linkedin.stdudfs.urnextractor.hive.UrnExtractorFunctionWrapper", 1, "urn_extractor"),
         new CoralRegistryOperatorRenameSqlCallTransformer(
             "com.linkedin.stdudfs.hive.daliudfs.UrnExtractorFunctionWrapper", 1, "urn_extractor"),
+        new CoralRegistryOperatorRenameSqlCallTransformer("com.linkedin.groot.runtime.udf.spark.HasMemberConsentUDF", 3,
+            "has_member_consent"),
+        new CoralRegistryOperatorRenameSqlCallTransformer("com.linkedin.groot.runtime.udf.spark.RedactFieldIfUDF", 4,
+            "urn_extractor"),
+        new CoralRegistryOperatorRenameSqlCallTransformer(
+            "com.linkedin.groot.runtime.udf.spark.RedactSecondarySchemaFieldIfUDF", 5,
+            "redact_secondary_schema_field_if"),
+        new CoralRegistryOperatorRenameSqlCallTransformer("com.linkedin.groot.runtime.udf.spark.GetMappedValueUDF", 2,
+            "get_mapped_value"),
         new GenericCoralRegistryOperatorRenameSqlCallTransformer(),
 
         new ReturnTypeAdjustmentTransformer(configs), new UnnestOperatorTransformer(), new AsOperatorTransformer(),