Merge pull request #30 from hatena/PickByKeys

PickByKeys を追加

Author: ne-sachirou

godash.go

@@ -110,3 +110,8 @@ func Associate[T any, K comparable, V any](collection []T, transform func(item T
 func FlatMap[T any, R any](collection []T, iteratee func(item T, index int) []R) []R {
 	return lo.FlatMap(collection, iteratee)
 }
+
+// PickByKeys returns same map type filtered by given keys.
+func PickByKeys[K comparable, V any, Map ~map[K]V](in Map, keys []K) Map {
+	return lo.PickByKeys(in, keys)
+}

godash_test.go

@@ -258,3 +258,10 @@ func ExampleFlatMap() {
 	}))
 	// Output: [0 1 2 3 4 5 6 7]
 }
+
+func ExamplePickByKeys() {
+	kv := map[string]int{"foo": 1, "bar": 2, "baz": 3}
+	result := godash.PickByKeys(kv, []string{"foo", "baz"})
+	fmt.Printf("%v", result)
+	// Output: map[baz:3 foo:1]
+}