Add specified cases.

apache · Aug 17, 2016 · 0008c3e · 0008c3e
1 parent 4094b43
commit 0008c3e
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/sql/catalyst/src/main/scala/org/apache/spark/sql/catalyst/optimizer/Optimizer.scala b/sql/catalyst/src/main/scala/org/apache/spark/sql/catalyst/optimizer/Optimizer.scala
@@ -730,15 +730,15 @@ object FoldablePropagation extends Rule[LogicalPlan] {
 
         // Operators that operate on objects should only have expressions from encoders, which
         // should never have foldable expressions.
-        case o: ObjectConsumer =>
+        case m: MapGroups =>
           stop = true
-          o
-        case o: ObjectProducer =>
+          m
+        case f: FlatMapGroupsInR =>
           stop = true
-          o
-        case a: AppendColumns =>
+          f
+        case c: CoGroup =>
           stop = true
-          a
+          c
 
         case p: LogicalPlan if !stop => p.transformExpressions {
           case a: AttributeReference if foldableMap.contains(a) =>