Skip to content

Commit

Permalink
add ForEachCondition implement (#485)
Browse files Browse the repository at this point in the history
* add ForEachCondition implement

* add example and test code

* rename ForEachCondition() to ForEachWhile()

* update test and example code

* rename ExampleForForEachWhile() to ExampleForEachWhile()
  • Loading branch information
Sianao authored Jul 15, 2024
1 parent 0f4679b commit cbfd1c6
Show file tree
Hide file tree
Showing 3 changed files with 47 additions and 0 deletions.
10 changes: 10 additions & 0 deletions slice.go
Original file line number Diff line number Diff line change
Expand Up @@ -94,6 +94,16 @@ func ForEach[T any](collection []T, iteratee func(item T, index int)) {
}
}

// ForEachWhile iterates over elements of collection and invokes iteratee for each element
// collection return value decide to continue or break ,just like do while()
func ForEachWhile[T any](collection []T, iteratee func(item T, index int) (goon bool)) {
for i := range collection {
if !iteratee(collection[i], i) {
break
}
}
}

// Times invokes the iteratee n times, returning an array of the results of each invocation.
// The iteratee is invoked with index as argument.
// Play: https://go.dev/play/p/vgQj3Glr6lT
Expand Down
14 changes: 14 additions & 0 deletions slice_example_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -88,7 +88,21 @@ func ExampleForEach() {
// 3
// 4
}
func ExampleForEachWhile() {
list := []int64{1, 2, -math.MaxInt, 4}

ForEachWhile(list, func(x int64, _ int) bool {
if x < 0 {
return false
}
fmt.Println(x)
return true
})

// Output:
// 1
// 2
}
func ExampleTimes() {
result := Times(3, func(i int) string {
return strconv.FormatInt(int64(i), 10)
Expand Down
23 changes: 23 additions & 0 deletions slice_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -157,6 +157,29 @@ func TestForEach(t *testing.T) {
is.IsIncreasing(callParams2)
}

func TestForEachWhile(t *testing.T) {
t.Parallel()
is := assert.New(t)

// check of callback is called for every element and in proper order

var callParams1 []string
var callParams2 []int

ForEachWhile([]string{"a", "b", "c"}, func(item string, i int) bool {
if item == "c" {
return false
}
callParams1 = append(callParams1, item)
callParams2 = append(callParams2, i)
return true
})

is.ElementsMatch([]string{"a", "b"}, callParams1)
is.ElementsMatch([]int{0, 1}, callParams2)
is.IsIncreasing(callParams2)
}

func TestUniq(t *testing.T) {
t.Parallel()
is := assert.New(t)
Expand Down

0 comments on commit cbfd1c6

Please sign in to comment.