fix bug

alteryx · Apr 12, 2015 · 00a4430 · 00a4430
1 parent 078d69b
commit 00a4430
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/sql/core/src/main/scala/org/apache/spark/sql/execution/Exchange.scala b/sql/core/src/main/scala/org/apache/spark/sql/execution/Exchange.scala
@@ -174,12 +174,14 @@ private[sql] case class AddExchange(sqlContext: SQLContext) extends Rule[SparkPl
       def addExchangeIfNecessary(
           partitioning: Partitioning,
           child: SparkPlan,
-          rowOrdering: Option[Ordering[Row]] = None): SparkPlan =
-        if (child.outputPartitioning != partitioning) {
-          Exchange(partitioning, child, sort = child.outputOrdering != rowOrdering)
+          rowOrdering: Option[Ordering[Row]] = None): SparkPlan = {
+        val needSort = child.outputOrdering != rowOrdering
+        if (child.outputPartitioning != partitioning || needSort) {
+          Exchange(partitioning, child, sort = needSort)
         } else {
           child
         }
+      }
 
       if (meetsRequirements && compatible) {
         operator