Fixes compilation error

apache · May 12, 2015 · fb5a607 · fb5a607
1 parent 9d17607
commit fb5a607
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 6 deletions.
diff --git a/sql/core/src/main/scala/org/apache/spark/sql/sources/interfaces.scala b/sql/core/src/main/scala/org/apache/spark/sql/sources/interfaces.scala
@@ -314,7 +314,7 @@ abstract class FSBasedRelation extends BaseRelation {
   def dataSchema: StructType
 
   /**
-   * Builds an `RDD[Row` containing all rows within this relation.
+   * Builds an `RDD[Row]` containing all rows within this relation.
    *
    * @param inputPaths Data files to be read. If the underlying relation is partitioned, only data
    *        files within required partition directories are included.
@@ -325,7 +325,7 @@ abstract class FSBasedRelation extends BaseRelation {
   }
 
   /**
-   * Builds an `RDD[Row` containing all rows within this relation.
+   * Builds an `RDD[Row]` containing all rows within this relation.
    *
    * @param requiredColumns Required columns.
    * @param inputPaths Data files to be read. If the underlying relation is partitioned, only data

diff --git a/sql/core/src/test/scala/org/apache/spark/sql/sources/FSBasedRelationSuite.scala b/sql/core/src/test/scala/org/apache/spark/sql/sources/FSBasedRelationSuite.scala
@@ -41,15 +41,15 @@ case class SimpleFSBasedRelation
     (val sqlContext: SQLContext)
   extends FSBasedRelation {
 
+  class SimpleOutputWriter extends OutputWriter {
+    override def write(row: Row): Unit = TestResult.writtenRows += row
+  }
+
   override val path = parameter("path")
 
   override def dataSchema: StructType =
     DataType.fromJson(parameter("schema")).asInstanceOf[StructType]
 
-  override def newOutputWriter(path: String): OutputWriter = new OutputWriter {
-    override def write(row: Row): Unit = TestResult.writtenRows += row
-  }
-
   override def buildScan(
       requiredColumns: Array[String],
       filters: Array[Filter],
@@ -67,6 +67,8 @@ case class SimpleFSBasedRelation
 
     Option(TestResult.rowsToRead).getOrElse(sqlContext.emptyResult)
   }
+
+  override def outputWriterClass: Class[_ <: OutputWriter] = classOf[SimpleOutputWriter]
 }
 
 object TestResult {