Generate .copy and .copyMetas function for generated classes (#53)

Author: alancai98

pig-runtime/src/org/partiql/pig/runtime/DomainNode.kt

@@ -24,6 +24,9 @@ interface DomainNode {
     /** This node's collection of metadata. */
     val metas: MetaContainer
 
+    /** Creates a copy of the current node with [newMetas] as the new metas. */
+    fun copyMetas(newMetas : MetaContainer): DomainNode
+
     /** Creates a copy of the current node with the specified [key] and [value] as metadata.*/
     fun withMeta(metaKey: String, metaValue: Any): DomainNode
 

pig-runtime/src/org/partiql/pig/runtime/LongPrimitive.kt

@@ -31,6 +31,10 @@ class LongPrimitive(val value: Long, override val metas: MetaContainer) : Domain
     fun copy(value: Long = this.value, metas: MetaContainer = this.metas): LongPrimitive =
         LongPrimitive(value, metas)
 
+    /** Creates a copy of the current [LongPrimitive] with [newMetas] as the new metas. */
+    override fun copyMetas(newMetas: MetaContainer): LongPrimitive =
+        LongPrimitive(value, newMetas)
+
     /** Creates a copy of the current [LongPrimitive] with the specified additional meta. */
     override fun withMeta(metaKey: String, metaValue: Any): LongPrimitive =
         LongPrimitive(this.value, metas + metaContainerOf(metaKey to metaValue))

pig-runtime/src/org/partiql/pig/runtime/SymbolPrimitive.kt

@@ -31,6 +31,10 @@ class SymbolPrimitive(val text: String, override val metas: MetaContainer) : Dom
     fun copy(text: String = this.text, metas: MetaContainer = this.metas): SymbolPrimitive =
         SymbolPrimitive(text, metas)
 
+    /** Creates a copy of the current [SymbolPrimitive] with [newMetas] as the new metas. */
+    override fun copyMetas(newMetas: MetaContainer): SymbolPrimitive =
+        SymbolPrimitive(text, newMetas)
+
     /** Creates a copy of the current [SymbolPrimitive] with the specified additional meta. */
     override fun withMeta(metaKey: String, metaValue: Any): SymbolPrimitive =
         SymbolPrimitive(text, metas + metaContainerOf(metaKey to metaValue))

pig-runtime/test/org/partiql/pig/runtime/IonElementTransformerBaseTests.kt

@@ -31,6 +31,10 @@ class  IonElementTransformerBaseTests {
     abstract class DummyDomainNode : DomainNode
 
     data class CorrectDomainNode(val someString: String, override val metas: MetaContainer = emptyMetaContainer()): DummyDomainNode() {
+        override fun copyMetas(newMetas: MetaContainer): DomainNode {
+            error("does not need to be implemented for tests")
+        }
+
         override fun withMeta(metaKey: String, metaValue: Any): DomainNode {
             error("does not need to be implemented for tests")
         }
@@ -40,6 +44,10 @@ class  IonElementTransformerBaseTests {
     }
 
     data class IncorrectDomainNode(val someString: String): DummyDomainNode() {
+        override fun copyMetas(newMetas: MetaContainer): DomainNode {
+            error("does not need to be implemented for tests")
+        }
+
         override fun withMeta(metaKey: String, metaValue: Any): DomainNode {
             error("does not need to be implemented for tests")
         }