feat: hrp boom support set run time

This commit is contained in:
machongwei
2022-08-10 21:59:16 +08:00
parent 61c34bfd16
commit a2352d627f

View File

@@ -253,14 +253,14 @@ func (b *Boomer) SetSpawnRate(spawnRate float64) {
}
// SetRunTime sets run time
func (b *Boomer) SetRunTime(spawnRate int64) {
func (b *Boomer) SetRunTime(runTime int64) {
switch b.mode {
case DistributedMasterMode:
b.masterRunner.setRunTime(spawnRate)
b.masterRunner.setRunTime(runTime)
case DistributedWorkerMode:
b.workerRunner.setRunTime(spawnRate)
b.workerRunner.setRunTime(runTime)
default:
b.localRunner.setRunTime(spawnRate)
b.localRunner.setRunTime(runTime)
}
}