commit ea14805a0588ac3cd7ffbef0e23463962bf9246b
author Aditya Toomula <atoomula@linkedin.com> Tue Mar 23 15:27:20 2021 -0700
committer GitHub <noreply@github.com> Tue Mar 23 15:27:20 2021 -0700
tree 00b5f3bb4715c0950a4388c53299f01595c5c4be
parent 4e392013a321a55eaccc03f75780cbea48668f4e

SAMZA-2639: Samza-sql: Fix ANY type validation for the parsed args (#1479)

samza-sql/src/main/java/org/apache/samza/sql/planner/Checker.java

diff --git a/samza-sql/src/main/java/org/apache/samza/sql/planner/Checker.java b/samza-sql/src/main/java/org/apache/samza/sql/planner/Checker.java
index 4a3c657..28ba187 100644
--- a/samza-sql/src/main/java/org/apache/samza/sql/planner/Checker.java
+++ b/samza-sql/src/main/java/org/apache/samza/sql/planner/Checker.java
@@ -98,7 +98,9 @@
         if (parsedSqlArgType.getSqlTypeName() == SqlTypeName.CHAR && udfArgumentAsSqlType == SqlTypeName.VARCHAR) {
           return true;
         } else if (!Objects.equals(parsedSqlArgType.getSqlTypeName(), udfArgumentAsSqlType)
-                && !ANY_SQL_TYPE_NAMES.contains(udfArgumentAsSqlType) && hasOneUdfMethod(udfMetadata)) {
+                && !ANY_SQL_TYPE_NAMES.contains(udfArgumentAsSqlType)
+                && !ANY_SQL_TYPE_NAMES.contains(parsedSqlArgType.getSqlTypeName())
+                && hasOneUdfMethod(udfMetadata)) {
           // 3(b). Throw up and fail on mismatch between the SamzaSqlType and CalciteType for any argument.
           String msg = String.format("Type mismatch in udf class: %s at argument index: %d." +
                           "Expected type: %s, actual type: %s.", udfMetadata.getName(),