Implement copy

apache · Jul 27, 2014 · 64b2ee1 · 64b2ee1
1 parent 3587460
commit 64b2ee1
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/...src/main/scala/org/apache/spark/sql/catalyst/expressions/codegen/GenerateProjection.scala b/...src/main/scala/org/apache/spark/sql/catalyst/expressions/codegen/GenerateProjection.scala
@@ -188,6 +188,11 @@ object GenerateProjection extends CodeGenerator[Seq[Expression], Projection] {
         }
       """
 
+    val copyFunction =
+      q"""
+        final def copy() = new $genericRowType(this.toArray)
+      """
+
     val classBody =
       nullFunctions ++ (
         lengthDef +:
@@ -196,16 +201,12 @@ object GenerateProjection extends CodeGenerator[Seq[Expression], Projection] {
         updateFunction +:
         equalsFunction +:
         hashCodeFunction +:
+        copyFunction +:
         (tupleElements ++ specificAccessorFunctions ++ specificMutatorFunctions))
 
     val code = q"""
       final class SpecificRow(i: $rowType) extends $mutableRowType {
         ..$classBody
-
-        // Not safe!
-        final def copy() = scala.sys.error("Not implemented")
-
-        final def getStringBuilder(ordinal: Int): StringBuilder = ???
       }
 
       new $projectionType { def apply(r: $rowType) = new SpecificRow(r) }