Add check for named expressions without associated ExecChecks

Signed-off-by: Andy Grove <andygrove@nvidia.com>
NVIDIA · Aug 11, 2021 · 2b46a96 · 2b46a96
1 parent 92bd4b6
commit 2b46a96
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/sql-plugin/src/main/scala/com/nvidia/spark/rapids/TypeChecks.scala b/sql-plugin/src/main/scala/com/nvidia/spark/rapids/TypeChecks.scala
@@ -812,9 +812,17 @@ class ExecChecks private(
       meta.childPlans.flatMap(_.outputAttributes.map(toStructField)),
       "unsupported data types in input: %s")
 
+    val namedChildExprs = meta.namedChildExprs
+
+    val missing = namedChildExprs.keys.filterNot(extendedChecks.contains)
+    if (missing.nonEmpty) {
+      throw new IllegalStateException(s"${meta.getClass.getSimpleName} " +
+        s"is missing ExecChecks for ${missing.mkString(",")}")
+    }
+
     extendedChecks.foreach {
       case (fieldName, typeSig) =>
-        val fieldMeta = meta.namedChildExprs(fieldName)
+        val fieldMeta = namedChildExprs(fieldName)
           .flatMap(_.typeMeta.dataType)
           .zipWithIndex
           .map(t => StructField(s"c${t._2}", t._1))