Merge pull request #91 from bbx-winner/main

fix: broadcast to all rendezvous at once when spawn done
This commit is contained in:
debugtalk
2022-02-17 20:00:50 +08:00
committed by GitHub
4 changed files with 16 additions and 10 deletions

View File

@@ -128,8 +128,8 @@ func (b *Boomer) Quit() {
b.localRunner.stop()
}
func (b *Boomer) IsSpawnDone() bool {
return b.localRunner.isSpawnDone
func (b *Boomer) GetSpawnDoneChan() chan struct{} {
return b.localRunner.spawnDone
}
func (b *Boomer) GetSpawnCount() int {

View File

@@ -63,7 +63,7 @@ type runner struct {
spawnCount int // target clients to spawn
spawnRate float64
loop *Loop // specify running cycles
isSpawnDone bool
spawnDone chan struct{}
outputs []Output
}
@@ -194,7 +194,7 @@ func (r *localRunner) spawnWorkers(spawnCount int, spawnRate float64, quit chan
}
}
r.isSpawnDone = true
close(r.spawnDone)
if spawnCompleteFunc != nil {
spawnCompleteFunc()
}
@@ -256,6 +256,7 @@ func newLocalRunner(spawnCount int, spawnRate float64) *localRunner {
spawnCount: spawnCount,
stats: newRequestStats(),
outputs: make([]Output, 0),
spawnDone: make(chan struct{}),
},
stopChan: make(chan bool),
}