Skip to content

Commit

Permalink
[pdata]: Add support to MoveTo for Map, allow avoiding copies
Browse files Browse the repository at this point in the history
Signed-off-by: Bogdan Drutu <[email protected]>
  • Loading branch information
bogdandrutu committed Sep 14, 2024
1 parent 8a07009 commit 73afd69
Show file tree
Hide file tree
Showing 3 changed files with 47 additions and 1 deletion.
20 changes: 20 additions & 0 deletions .chloggen/add-map-moveto.yaml
Original file line number Diff line number Diff line change
@@ -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: pdata

# A brief description of the change. Surround your text with quotes ("") if it needs to start with a backtick (`).
note: Add support to MoveTo for Map, allow avoiding copies

# One or more tracking issues or pull requests related to the change
issues: [11175]

# 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, user]
9 changes: 9 additions & 0 deletions pdata/pcommon/map.go
Original file line number Diff line number Diff line change
Expand Up @@ -225,6 +225,15 @@ func (m Map) Range(f func(k string, v Value) bool) {
}
}

// MoveTo moves all key/values from the current map overriding the destination and
// resetting the current instance to its zero value
func (m Map) MoveTo(dest Map) {
m.getState().AssertMutable()
dest.getState().AssertMutable()
*dest.getOrig() = *m.getOrig()
*m.getOrig() = nil
}

// CopyTo copies all elements from the current map overriding the destination.
func (m Map) CopyTo(dest Map) {
dest.getState().AssertMutable()
Expand Down
19 changes: 18 additions & 1 deletion pdata/pcommon/map_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -375,11 +375,28 @@ func TestMap_FromRaw(t *testing.T) {
}, v.Map().AsRaw())
}

func TestMap_MoveTo(t *testing.T) {
dest := NewMap()
// Test MoveTo to empty
NewMap().MoveTo(dest)
assert.Equal(t, 0, dest.Len())

// Test MoveTo larger slice
src := Map(internal.GenerateTestMap())
src.MoveTo(dest)
assert.EqualValues(t, Map(internal.GenerateTestMap()), dest)
assert.Equal(t, 0, src.Len())

// Test MoveTo from empty to non-empty
NewMap().MoveTo(dest)
assert.Equal(t, 0, dest.Len())
}

func TestMap_CopyTo(t *testing.T) {
dest := NewMap()
// Test CopyTo to empty
NewMap().CopyTo(dest)
assert.EqualValues(t, 0, dest.Len())
assert.Equal(t, 0, dest.Len())

// Test CopyTo larger slice
Map(internal.GenerateTestMap()).CopyTo(dest)
Expand Down

0 comments on commit 73afd69

Please sign in to comment.