diff --git a/coral-hive/src/main/java/com/linkedin/coral/hive/hive2rel/functions/StaticHiveFunctionRegistry.java b/coral-hive/src/main/java/com/linkedin/coral/hive/hive2rel/functions/StaticHiveFunctionRegistry.java index 95c7d12f8..9ec18e7db 100644 --- a/coral-hive/src/main/java/com/linkedin/coral/hive/hive2rel/functions/StaticHiveFunctionRegistry.java +++ b/coral-hive/src/main/java/com/linkedin/coral/hive/hive2rel/functions/StaticHiveFunctionRegistry.java @@ -358,6 +358,8 @@ public RelDataType inferReturnType(SqlOperatorBinding opBinding) { family(SqlTypeFamily.NUMERIC, SqlTypeFamily.STRING, SqlTypeFamily.STRING)); createAddUserDefinedFunction("com.linkedin.dali.udf.date.hive.EpochToEpochMilliseconds", BIGINT_NULLABLE, NUMERIC); createAddUserDefinedFunction("com.linkedin.dali.udf.sanitize.hive.Sanitize", HiveReturnTypes.STRING, STRING); + createAddUserDefinedFunction("org.apache.hadoop.hive.ql.udf.generic.GenericProject", ARG0, + family(SqlTypeFamily.ANY, SqlTypeFamily.STRING)); // LIHADOOP-48502: The following UDFs are already defined using Transport UDF. // The class name is the corresponding Hive UDF. @@ -393,6 +395,8 @@ public RelDataType inferReturnType(SqlOperatorBinding opBinding) { family(Collections.nCopies(2, SqlTypeFamily.STRING))); createAddUserDefinedFunction("com.linkedin.stdudfs.parsing.hive.Ip2Str", HiveReturnTypes.STRING, or(family(SqlTypeFamily.STRING, SqlTypeFamily.NUMERIC, SqlTypeFamily.NUMERIC), family(SqlTypeFamily.STRING))); + createAddUserDefinedFunction("com.linkedin.stdudfs.daliudfs.hive.IsTestMemberId", ReturnTypes.BOOLEAN, + family(SqlTypeFamily.NUMERIC, SqlTypeFamily.STRING)); // This is a Hive Custom UDF which is a simplified version of 'date-converter' package. // This UDF is not converted to a transport UDF.