Fix the problematic case.

apache · Nov 28, 2017 · 8c7f749 · 8c7f749
1 parent 6368702
commit 8c7f749
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/sql/catalyst/src/main/scala/org/apache/spark/sql/catalyst/expressions/Expression.scala b/sql/catalyst/src/main/scala/org/apache/spark/sql/catalyst/expressions/Expression.scala
@@ -126,9 +126,9 @@ abstract class Expression extends TreeNode[Expression] {
       eval.inputRow = ctx.INPUT_ROW
     }
     if (ctx.currentVars != null) {
-      val boundRefs = children.flatMap(_.collect {
+      val boundRefs = this.collect {
         case b @ BoundReference(ordinal, _, _) if ctx.currentVars(ordinal) != null => (ordinal, b)
-      }).toMap
+      }.toMap
 
       ctx.currentVars.zipWithIndex.filter(_._1 != null).foreach { case (currentVar, idx) =>
         if (boundRefs.contains(idx)) {