FlatMap function introduced
This commit is contained in:
parent
4612cb74b2
commit
78436e629c
15
flat.go
15
flat.go
@ -15,6 +15,21 @@ func FlatSlice[T any](source <-chan []T) <-chan T {
|
|||||||
return out
|
return out
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func FlatMap[K comparable, V, T any](source <-chan map[K]V, unmapper func(key K, value V) T) <-chan T {
|
||||||
|
out := make(chan T, cap(source))
|
||||||
|
|
||||||
|
go func() {
|
||||||
|
defer close(out)
|
||||||
|
for slice := range source {
|
||||||
|
for k, v := range slice {
|
||||||
|
out <- unmapper(k, v)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
|
||||||
|
return out
|
||||||
|
}
|
||||||
|
|
||||||
func FlatChan[T any](source <-chan <-chan T) <-chan T {
|
func FlatChan[T any](source <-chan <-chan T) <-chan T {
|
||||||
out := make(chan T, cap(source))
|
out := make(chan T, cap(source))
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user