mirror/go-patterns
1
0
Fork 0
mirror of https://github.com/tmrts/go-patterns.git synced 2024-11-21 20:46:08 +03:00
Code Issues 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{}
}
func (p *Pipeline) Start() {
func (p *Pipeline) Start(ctx context.Context) {
go func(pipe *Pipeline) {
for {
expectationWorkers := len(pipe.chain)
if expectationWorkers > MaxWorkers {
expectationWorkers = expectationWorkers % MaxWorkers
expectationWorkers := len(pipe.chain) % MaxWorkers
if expectationWorkers >= MaxWorkers {
expectationWorkers = 0
}
for _, c := range pipe.workers {
if expectationWorkers < int(c.index) {
break
}
select {
case val := <-pipe.chain:
go c.stream(val)
select {
case <-ctx.Done():
return
case val, ok := <-pipe.chain:
if !ok {
return
}
go pipe.workers[expectationWorkers].stream(val)
}
}
}(p)