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 // SetRunTime sets run time
func (b *Boomer) SetRunTime(spawnRate int64) { func (b *Boomer) SetRunTime(runTime int64) {
switch b.mode { switch b.mode {
case DistributedMasterMode: case DistributedMasterMode:
b.masterRunner.setRunTime(spawnRate) b.masterRunner.setRunTime(runTime)
case DistributedWorkerMode: case DistributedWorkerMode:
b.workerRunner.setRunTime(spawnRate) b.workerRunner.setRunTime(runTime)
default: default:
b.localRunner.setRunTime(spawnRate) b.localRunner.setRunTime(runTime)
} }
} }