Implement trigger() as limit() followed by collect()

Implement trigger() as limit() followed by collect()gdfg
apache · Jun 11, 2014 · 8d42d03 · 8d42d03
1 parent 0402bd7
commit 8d42d03
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/sql/core/src/main/scala/org/apache/spark/sql/SchemaRDD.scala b/sql/core/src/main/scala/org/apache/spark/sql/SchemaRDD.scala
@@ -374,6 +374,9 @@ class SchemaRDD(
 
   override def collect(): Array[Row] = queryExecution.executedPlan.executeCollect()
 
+  override def take(num: Int): Array[Row] =
+    new SchemaRDD(sqlContext, Limit(Literal(num), logicalPlan)).collect()
+
   // =======================================================================
   // Base RDD functions that do NOT change schema
   // =======================================================================