Feature: add values and keys methods to sync.Map

Author: aopoltorzhicky

pkg/sync/map.go

@@ -67,3 +67,23 @@ func (m *Map[K, V]) Len() int {
 	defer m.mx.RUnlock()
 	return len(m.m)
 }
+
+func (m *Map[K, V]) Values() []V {
+	arr := make([]V, 0)
+	m.mx.RLock()
+	for _, v := range m.m {
+		arr = append(arr, v)
+	}
+	m.mx.RUnlock()
+	return arr
+}
+
+func (m *Map[K, V]) Keys() []K {
+	arr := make([]K, 0)
+	m.mx.RLock()
+	for k := range m.m {
+		arr = append(arr, k)
+	}
+	m.mx.RUnlock()
+	return arr
+}

pkg/sync/map_test.go

@@ -5,6 +5,8 @@ import (
 	"runtime"
 	"sync"
 	"testing"
+
+	"github.com/stretchr/testify/require"
 )
 
 func TestMap_Get(t *testing.T) {
@@ -170,3 +172,29 @@ func TestMap_Len(t *testing.T) {
 		t.Fatalf("unexpected map size, got %v want %v", length, 3)
 	}
 }
+
+func TestMap_Values(t *testing.T) {
+	m := NewMap[int, string]()
+	for i, v := range [3]string{"len", "sync", "map"} {
+		m.Set(i, v)
+	}
+
+	values := m.Values()
+	require.Len(t, values, 3)
+	require.Contains(t, values, "len")
+	require.Contains(t, values, "sync")
+	require.Contains(t, values, "map")
+}
+
+func TestMap_Keys(t *testing.T) {
+	m := NewMap[int, string]()
+	for i, v := range [3]string{"len", "sync", "map"} {
+		m.Set(i, v)
+	}
+
+	keys := m.Keys()
+	require.Len(t, keys, 3)
+	require.Contains(t, keys, 0)
+	require.Contains(t, keys, 1)
+	require.Contains(t, keys, 2)
+}