Added eval for Rand (without any support for user-defined seed).

witgo · Apr 7, 2014 · fd11322 · fd11322
1 parent 0307db0
commit fd11322
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/sql/catalyst/src/main/scala/org/apache/spark/sql/catalyst/expressions/Rand.scala b/sql/catalyst/src/main/scala/org/apache/spark/sql/catalyst/expressions/Rand.scala
@@ -17,11 +17,18 @@
 
 package org.apache.spark.sql.catalyst.expressions
 
+import java.util.Random
 import org.apache.spark.sql.catalyst.types.DoubleType
 
+
 case object Rand extends LeafExpression {
-  def dataType = DoubleType
-  def nullable = false
-  def references = Set.empty
+  override def dataType = DoubleType
+  override def nullable = false
+  override def references = Set.empty
+
+  private[this] lazy val rand = new Random
+
+  override def eval(input: Row = null) = rand.nextDouble().asInstanceOf[EvaluatedType]
+
   override def toString = "RAND()"
 }