feat: adding FilterMapToSlice

Author: samber

README.md

@@ -1455,6 +1455,21 @@ s := lo.MapToSlice(m, func(k int, v int64) string {
 
 [[play](https://go.dev/play/p/ZuiCZpDt6LD)]
 
+### FilterMapToSlice
+
+Transforms a map into a slice based on specific iteratee. The iteratee returns a value and a boolean. If the boolean is true, the value is added to the result slice.
+
+If the boolean is false, the value is not added to the result slice. The order of the keys in the input map is not specified and the order of the keys in the output slice is not guaranteed.
+
+```go
+kv := map[int]int64{1: 1, 2: 2, 3: 3, 4: 4}
+
+result := lo.FilterMapToSlice(kv, func(k int, v int64) (string, bool) {
+    return fmt.Sprintf("%d_%d", k, v), k%2 == 0
+})
+// []{"2_2", "4_4"}
+```
+
 ### Range / RangeFrom / RangeWithSteps
 
 Creates an array of numbers (positive and/or negative) progressing from start up to, but not including end.

map.go

@@ -325,3 +325,19 @@ func MapToSlice[K comparable, V any, R any](in map[K]V, iteratee func(key K, val
 
 	return result
 }
+
+// FilterMapToSlice transforms a map into a slice based on specific iteratee.
+// The iteratee returns a value and a boolean. If the boolean is true, the value is added to the result slice.
+// If the boolean is false, the value is not added to the result slice.
+// The order of the keys in the input map is not specified and the order of the keys in the output slice is not guaranteed.
+func FilterMapToSlice[K comparable, V any, R any](in map[K]V, iteratee func(key K, value V) (R, bool)) []R {
+	result := make([]R, 0, len(in))
+
+	for k := range in {
+		if v, ok := iteratee(k, in[k]); ok {
+			result = append(result, v)
+		}
+	}
+
+	return result
+}

map_example_test.go

@@ -232,3 +232,15 @@ func ExampleMapToSlice() {
 	fmt.Printf("%v", result)
 	// Output: [1_1 2_2 3_3 4_4]
 }
+
+func ExampleFilterMapToSlice() {
+	kv := map[int]int64{1: 1, 2: 2, 3: 3, 4: 4}
+
+	result := FilterMapToSlice(kv, func(k int, v int64) (string, bool) {
+		return fmt.Sprintf("%d_%d", k, v), k%2 == 0
+	})
+
+	sort.StringSlice(result).Sort()
+	fmt.Printf("%v", result)
+	// Output: [2_2 4_4]
+}

map_test.go

@@ -513,6 +513,23 @@ func TestMapToSlice(t *testing.T) {
 	is.ElementsMatch(result2, []string{"1", "2", "3", "4"})
 }
 
+func TestFilterMapToSlice(t *testing.T) {
+	t.Parallel()
+	is := assert.New(t)
+
+	result1 := FilterMapToSlice(map[int]int{1: 5, 2: 6, 3: 7, 4: 8}, func(k int, v int) (string, bool) {
+		return fmt.Sprintf("%d_%d", k, v), k%2 == 0
+	})
+	result2 := FilterMapToSlice(map[int]int{1: 5, 2: 6, 3: 7, 4: 8}, func(k int, _ int) (string, bool) {
+		return strconv.FormatInt(int64(k), 10), k%2 == 0
+	})
+
+	is.Equal(len(result1), 2)
+	is.Equal(len(result2), 2)
+	is.ElementsMatch(result1, []string{"2_6", "4_8"})
+	is.ElementsMatch(result2, []string{"2", "4"})
+}
+
 func BenchmarkAssign(b *testing.B) {
 	counts := []int{32768, 1024, 128, 32, 2}