Implement copy

marmbrus · marmbrus · commit 64b2ee19f3e6 · 2014-07-27T14:58:57.000-07:00
diff --git a/sql/catalyst/src/main/scala/org/apache/spark/sql/catalyst/expressions/codegen/GenerateProjection.scala b/sql/catalyst/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) }
