Github user viirya commented on a diff in the pull request:
https://github.com/apache/spark/pull/21231#discussion_r186672482
--- Diff:
sql/catalyst/src/main/scala/org/apache/spark/sql/catalyst/expressions/SortOrder.scala
---
@@ -147,7 +148,44 @@ case class SortPrefix(child: SortOrder) extends
UnaryExpression {
(!child.isAscending && child.nullOrdering == NullsLast)
}
- override def eval(input: InternalRow): Any = throw new
UnsupportedOperationException
+ private lazy val calcPrefix: Any => Long = child.child.dataType match {
+ case BooleanType => (raw) =>
+ if (raw.asInstanceOf[Boolean]) 1 else 0
+ case _: IntegralType => (raw) =>
+ raw.asInstanceOf[java.lang.Number].longValue()
+ case DateType | TimestampType =>
+ _.asInstanceOf[java.lang.Number].longValue()
--- End diff --
Aren't the above two cases being the same thing?
---
---------------------------------------------------------------------
To unsubscribe, e-mail: reviews-unsubscr...@spark.apache.org
For additional commands, e-mail: reviews-h...@spark.apache.org
