[pdata] Add CopyTo and MoveTo methods to primitive slices (#6044)

CHANGELOG.md

@@ -43,6 +43,7 @@
 - Add `skip-get-modules` builder flag to support isolated environment executions (#6009)
   - Skip unnecessary Go binary path validation when the builder is used with `skip-compilation` and `skip-get-modules` flags (#6026)
 - Add mapstructure hook function for confmap.Unmarshaler interface (#6029)
+- Add CopyTo and MoveTo methods to primitive slices (#6044)
 
 ### 🧰 Bug fixes 🧰
 

pdata/internal/cmd/pdatagen/internal/base_fields.go

@@ -52,14 +52,6 @@ func (ms ${structName}) Set${fieldName}(v ${returnType}) {
 	ms.getOrig().${originFieldName} = v
 }`
 
-const copyToPrimitiveSliceTestTemplate = `	if len(ms.getOrig().${originFieldName}) == 0 {	
-		dest.getOrig().${originFieldName} = nil
-	} else {
-		dest.getOrig().${originFieldName} = make(${rawType}, len(ms.getOrig().${originFieldName}))
-		copy(dest.getOrig().${originFieldName}, ms.getOrig().${originFieldName})
-	}
-`
-
 const accessorsPrimitiveSliceTemplate = `// ${fieldName} returns the ${lowerFieldName} associated with this ${structName}.
 func (ms ${structName}) ${fieldName}() ${packageName}${returnType} {
 	return ${packageName}${returnType}(internal.New${returnType}(&ms.getOrig().${originFieldName}))
@@ -516,20 +508,7 @@ func (psf *primitiveSliceField) generateSetWithTestValue(sb *strings.Builder) {
 }
 
 func (psf *primitiveSliceField) generateCopyToValue(ms baseStruct, sb *strings.Builder) {
-	sb.WriteString(os.Expand(copyToPrimitiveSliceTestTemplate, func(name string) string {
-		switch name {
-		case "structName":
-			return ms.getName()
-		case "originFieldName":
-			return psf.originFieldName
-		case "rawType":
-			return psf.rawType
-		case "returnType":
-			return psf.returnType
-		default:
-			panic(name)
-		}
-	}))
+	sb.WriteString("\tms." + psf.fieldName + "().CopyTo(dest." + psf.fieldName + "())")
 }
 
 var _ baseField = (*primitiveSliceField)(nil)

pdata/internal/cmd/pdatagen/internal/primitive_slice_structs.go

@@ -61,6 +61,17 @@ func (ms ${structName}) SetAt(i int, val ${itemType}) {
 	(*ms.getOrig())[i] = val
 }
 
+// MoveTo moves ${structName} to another instance.
+func (ms ${structName}) MoveTo(dest ${structName}) {
+	*dest.getOrig() = *ms.getOrig()
+	*ms.getOrig() = nil
+}
+
+// CopyTo copies ${structName} to another instance.
+func (ms ${structName}) CopyTo(dest ${structName}) {
+	*dest.getOrig() = copy${structName}(*ms.getOrig())
+}
+
 func copy${structName}(from []${itemType}) []${itemType} {
 	if len(from) == 0 {
 		return nil
@@ -82,6 +93,24 @@ const immutableSliceTestTemplate = `func TestNew${structName}(t *testing.T) {
 	ms.FromRaw([]${itemType}{3})
 	assert.Equal(t, 1, ms.Len())
 	assert.Equal(t, ${itemType}(3), ms.At(0))
+	
+	cp := New${structName}()
+	ms.CopyTo(cp)
+	ms.SetAt(0, ${itemType}(2))
+	assert.Equal(t, ${itemType}(2), ms.At(0))
+	assert.Equal(t, ${itemType}(3), cp.At(0))
+	ms.CopyTo(cp)
+	assert.Equal(t, ${itemType}(2), cp.At(0))
+	
+	mv := New${structName}()
+	ms.MoveTo(mv)
+	assert.Equal(t, 0, ms.Len())
+	assert.Equal(t, 1, mv.Len())
+	assert.Equal(t, ${itemType}(2), mv.At(0))
+	ms.FromRaw([]${itemType}{1, 2, 3})
+	ms.MoveTo(mv)
+	assert.Equal(t, 3, mv.Len())
+	assert.Equal(t, ${itemType}(1), mv.At(0))
 }`
 
 const primitiveSliceInternalTemplate = `