67 lines
1.2 KiB
Go
67 lines
1.2 KiB
Go
package index
|
|
|
|
type IterateFunction[K, V any] func(key K, value V) (stop bool)
|
|
|
|
type IteratorFunction[K, V any] func(f IterateFunction[K, V]) (complete bool)
|
|
|
|
type Iterator[K, V any] interface {
|
|
All(f IterateFunction[K, V]) (complete bool)
|
|
}
|
|
|
|
func IteratorToIterator[K, V any](i Iterator[K, V], _ ...error) Iterator[K, V] {
|
|
return i
|
|
}
|
|
|
|
func IterateToSlice[T any](i Iterator[int, T], _ ...error) (s []T) {
|
|
i.All(func(key int, value T) (stop bool) {
|
|
s = append(s, value)
|
|
return false
|
|
})
|
|
return s
|
|
}
|
|
|
|
func IterateToSliceWithoutPointer[T any](i Iterator[int, *T], _ ...error) (s []T) {
|
|
i.All(func(key int, value *T) (stop bool) {
|
|
s = append(s, *value)
|
|
return false
|
|
})
|
|
return s
|
|
}
|
|
|
|
func SliceIterate[S ~[]T, T any](s S, f IterateFunction[int, T]) (complete bool) {
|
|
for i, v := range s {
|
|
if f(i, v) {
|
|
return (len(s) - 1) == i
|
|
}
|
|
}
|
|
return true
|
|
}
|
|
|
|
func ChanIterate[T any](c <-chan T, f IterateFunction[int, T]) (complete bool) {
|
|
var i int
|
|
for {
|
|
v, more := <-c
|
|
if !more {
|
|
return true
|
|
}
|
|
if f(i, v) {
|
|
return false
|
|
}
|
|
i++
|
|
}
|
|
}
|
|
|
|
func ChanConsume[T any](c <-chan T) {
|
|
for range c {
|
|
|
|
}
|
|
}
|
|
|
|
func ChanToSlice[S ~[]T, T any](c <-chan T) (s S) {
|
|
s = make(S, 0, len(c))
|
|
for v := range c {
|
|
s = append(s, v)
|
|
}
|
|
return s
|
|
}
|