[pdata/pcommon] Add RemoveIf method to primitive slice types (#14028)

Add `RemoveIf` method to primitive slice types (StringSlice, Int64Slice,
UInt64Slice, Float64Slice, Int32Slice, ByteSlice) implemented
consistently with other slices

Author: dmitryax

.chloggen/pdata-primitive-slice-removeif.yaml

@@ -0,0 +1,20 @@
+# Use this changelog template to create an entry for release notes.
+
+# One of 'breaking', 'deprecation', 'new_component', 'enhancement', 'bug_fix'
+change_type: enhancement
+
+# The name of the component, or a single word describing the area of concern, (e.g. otlpreceiver)
+component: pkg/pdata
+
+# A brief description of the change.  Surround your text with quotes ("") if it needs to start with a backtick (`).
+note: "Add `RemoveIf` method to primitive slice types (StringSlice, Int64Slice, UInt64Slice, Float64Slice, Int32Slice, ByteSlice)"
+
+# One or more tracking issues or pull requests related to the change
+issues: [14027]
+
+# Optional: The change log or logs in which this entry should be included.
+# e.g. '[user]' or '[user, api]'
+# Include 'user' if the change is relevant to end users.
+# Include 'api' if there is a change to a library API.
+# Default: '[user]'
+change_logs: [api]

internal/cmd/pdatagen/internal/pdata/templates/primitive_slice.go.tmpl

@@ -126,6 +126,28 @@ func (ms {{ .structName }}) MoveAndAppendTo(dest {{ .structName }}) {
   *ms.getOrig() = nil
 }
 
+// RemoveIf calls f sequentially for each element present in the slice.
+// If f returns true, the element is removed from the slice.
+func (ms {{ .structName }}) RemoveIf(f func({{ .itemType }}) bool) {
+	ms.getState().AssertMutable()
+	newLen := 0
+	for i := 0; i < len(*ms.getOrig()); i++ {
+		if f((*ms.getOrig())[i]) {
+			continue
+		}
+		if newLen == i {
+			// Nothing to move, element is at the right place.
+			newLen++
+			continue
+		}
+		(*ms.getOrig())[newLen] = (*ms.getOrig())[i]
+		var zero {{ .itemType }}
+		(*ms.getOrig())[i] = zero
+		newLen++
+	}
+	*ms.getOrig() = (*ms.getOrig())[:newLen]
+}
+
 // CopyTo copies all elements from the current slice overriding the destination.
 func (ms {{ .structName }}) CopyTo(dest {{ .structName }}) {
 	dest.getState().AssertMutable()

internal/cmd/pdatagen/internal/pdata/templates/primitive_slice_test.go.tmpl

@@ -153,6 +153,32 @@ func Test{{ .structName }}MoveAndAppendTo(t *testing.T) {
   assert.Equal(t, ms2.Len(), 6)
 }
 
+func Test{{ .structName }}RemoveIf(t *testing.T) {
+	emptySlice := New{{ .structName }}()
+	emptySlice.RemoveIf(func(el {{ .itemType }}) bool {
+		t.Fail()
+		return false
+	})
+
+	ms := New{{ .structName }}()
+	ms.FromRaw([]{{ .itemType }}{ {{ .testOrigVal }} })
+	pos := 0
+	ms.RemoveIf(func(el {{ .itemType }}) bool {
+		pos++
+		return pos%2 == 1
+	})
+	assert.Equal(t, pos/2, ms.Len())
+}
+
+func Test{{ .structName }}RemoveIfAll(t *testing.T) {
+	ms := New{{ .structName }}()
+	ms.FromRaw([]{{ .itemType }}{ {{ .testOrigVal }} })
+	ms.RemoveIf(func(el {{ .itemType }}) bool {
+		return true
+	})
+	assert.Equal(t, 0, ms.Len())
+}
+
 func Test{{ .structName }}Equal(t *testing.T) {
 	ms := New{{ .structName }}()
 	ms2 := New{{ .structName }}()