fix sort eval bug

alteryx · Apr 3, 2015 · 57baa40 · 57baa40
1 parent 303b6da
commit 57baa40
Showing 1 changed file with 4 additions and 1 deletion.
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
@@ -85,7 +85,10 @@ case class Exchange(newPartitioning: Partitioning, child: SparkPlan) extends Una
             iter.map(r => mutablePair.update(hashExpressions(r), r))
           }
         }
-        val sortingExpressions = expressions.map(s => new SortOrder(s, Ascending))
+        val sortingExpressions = expressions.zipWithIndex.map {
+          case (exp, index) =>
+            new SortOrder(BoundReference(index, exp.dataType, exp.nullable), Ascending)
+        }
         val ordering = new RowOrdering(sortingExpressions, child.output)
         val part = new HashPartitioner(numPartitions)
         val shuffled = new ShuffledRDD[Row, Row, Row](rdd, part).setKeyOrdering(ordering)