mirror/awesome-patterns
1
0
Fork 0
mirror of https://github.com/crazybber/awesome-patterns.git synced 2024-11-22 20:56:02 +03:00
Code Issues Packages Projects Releases Wiki Activity

[fanout] Optimize fanout

Browse Source
This commit is contained in:
paulaan 2019-05-06 21:43:42 +07:00
parent 9c0c9709ff
commit 44115024e2
1 changed files with 11 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
messaging/fanout.go
View File

@ -22,21 +22,21 @@ type Pipeline struct {
chain chan interface{} chain chan interface{}
} }
func (p *Pipeline) Start() { func (p *Pipeline) Start(ctx context.Context) {
go func(pipe *Pipeline) { go func(pipe *Pipeline) {
for { for {
expectationWorkers := len(pipe.chain) expectationWorkers := len(pipe.chain) % MaxWorkers
if expectationWorkers > MaxWorkers { if expectationWorkers >= MaxWorkers {
expectationWorkers = expectationWorkers % MaxWorkers expectationWorkers = 0
} }
for _, c := range pipe.workers { select {
if expectationWorkers < int(c.index) { case <-ctx.Done():
break return
} case val, ok := <-pipe.chain:
select { if !ok {
case val := <-pipe.chain: return
go c.stream(val)
} }
go pipe.workers[expectationWorkers].stream(val)
} }
} }
}(p) }(p)