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

Author: Reynold Xin <[email protected]> Closes #349 from rxin/rand and squashes the following commits: fd11322 [Reynold Xin] Added eval for Rand (without any support for user-defined seed).
apache · Apr 8, 2014 · 31e6fff · 31e6fff
1 parent 55dfd5d
commit 31e6fff
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()"
 }