mapex.Keys(), mapex.Values()

Author: guonaihong

README.md

@@ -126,10 +126,29 @@ s2 := s.Clone()
 assert.True(t, s.Equal(s2))
 ```
 
-## 九`ifop`
+## 九、`ifop`
 ifop是弥补下golang没有三目运算符，使用函数模拟
 
 ```go
 // 如果该值不为0, 返回原来的值，否则默认值
 val = IfElse(len(val) != 0, val, "default")
 ```
+## 十、`mapex`
+薄薄一层包装，增加标准库map的接口
+* mapex.Keys()
+```go
+m := make(map[string]string)
+m["a"] = "1"
+m["b"] = "2"
+m["c"] = "3"
+get := mapex.Keys(m)// 返回map的所有key
+
+```
+* mapex.Values()
+```go
+m := make(map[string]string)
+m["a"] = "1"
+m["b"] = "2"
+m["c"] = "3"
+get := Values(m)
+```

go.mod

@@ -1,14 +1,14 @@
 module github.com/guonaihong/gstl
 
-go 1.18
+go 1.19
 
 require (
+	github.com/cespare/xxhash/v2 v2.1.2
 	github.com/stretchr/testify v1.7.1
 	golang.org/x/exp v0.0.0-20220328175248-053ad81199eb
 )
 
 require (
-	github.com/cespare/xxhash/v2 v2.1.2 // indirect
 	github.com/davecgh/go-spew v1.1.0 // indirect
 	github.com/pmezard/go-difflib v1.0.0 // indirect
 	gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c // indirect

mapex/mapex.go

@@ -0,0 +1,27 @@
+package mapex
+
+type Map[K comparable, V any] map[K]V
+
+func Keys[K comparable, V any](m map[K]V) (keys []K) {
+	return Map[K, V](m).Keys()
+}
+
+func Values[K comparable, V any](m map[K]V) (values []V) {
+	return Map[K, V](m).Values()
+}
+
+func (m Map[K, V]) Keys() (keys []K) {
+	keys = make([]K, 0, len(m))
+	for k := range m {
+		keys = append(keys, k)
+	}
+	return
+}
+
+func (m Map[K, V]) Values() (values []V) {
+	values = make([]V, 0, len(m))
+	for _, v := range m {
+		values = append(values, v)
+	}
+	return
+}

mapex/mapex_test.go

@@ -0,0 +1,35 @@
+package mapex
+
+import (
+	"sort"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func Test_Keys(t *testing.T) {
+	m := make(map[string]string)
+	m["a"] = "1"
+	m["b"] = "2"
+	m["c"] = "3"
+	get := Keys(m)
+	sort.Strings(get)
+	assert.Equal(t, get, []string{"a", "b", "c"})
+	get = Map[string, string](m).Keys()
+	sort.Strings(get)
+	assert.Equal(t, get, []string{"a", "b", "c"})
+}
+
+func Test_Values(t *testing.T) {
+	m := make(map[string]string)
+	m["a"] = "1"
+	m["b"] = "2"
+	m["c"] = "3"
+	get := Values(m)
+	sort.Strings(get)
+	assert.Equal(t, get, []string{"1", "2", "3"})
+
+	get = Map[string, string](m).Values()
+	sort.Strings(get)
+	assert.Equal(t, get, []string{"1", "2", "3"})
+}