mirror of
https://github.com/tmrts/go-patterns.git
synced 2026-02-04 22:56:18 +00:00
Add fan-in/out messaging patterns
This commit is contained in:
40
fan/fan_out.go
Normal file
40
fan/fan_out.go
Normal file
@@ -0,0 +1,40 @@
|
||||
package fan
|
||||
|
||||
// Out implements fan.Out messaging pattern
|
||||
// Split a channel into n channels that receive messages
|
||||
// in a round-robin fashion.
|
||||
func Out(ch <-chan int, n int) []<-chan int {
|
||||
cs := make([]chan int)
|
||||
for i := 0; i < n; i++ {
|
||||
cs = append(cs, make(chan int))
|
||||
}
|
||||
|
||||
// Distributes the work in a round robin fashion among the stated number
|
||||
// of channels until the main channel has been closed. In that case, close
|
||||
// all channels and return.
|
||||
distributeToChannels := func(ch <-chan int, cs []chan<- int) {
|
||||
// Close every channel when the execution ends.
|
||||
defer func(cs []chan<- int) {
|
||||
for _, c := range cs {
|
||||
close(c)
|
||||
}
|
||||
}(cs)
|
||||
|
||||
for {
|
||||
for _, c := range cs {
|
||||
select {
|
||||
case val, ok := <-ch:
|
||||
if !ok {
|
||||
return
|
||||
}
|
||||
|
||||
c <- val
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
go distributeToChannels(ch, cs)
|
||||
|
||||
return cs
|
||||
}
|
||||
Reference in New Issue
Block a user