feat: hrp boom support set run time

This commit is contained in:
machongwei
2022-08-10 22:03:24 +08:00
parent a2352d627f
commit 8ae5ba81bc
+7 -8
View File
@@ -380,10 +380,15 @@ func (r *runner) reset() {
} }
func (r *runner) runTimeCheck(runTime int64) { func (r *runner) runTimeCheck(runTime int64) {
if runTime <= 0 {
return
}
for range time.Tick(time.Second * 3) { for range time.Tick(time.Second * 3) {
nowTime := time.Now().Unix() nowTime := time.Now().Unix()
if nowTime > runTime { if nowTime > runTime {
r.stop() r.stop()
return
} }
} }
} }
@@ -646,10 +651,7 @@ func (r *localRunner) start() {
// output setup // output setup
r.outputOnStart() r.outputOnStart()
runTime := r.getRunTime() go r.runTimeCheck(r.getRunTime())
if runTime != 0 {
go r.runTimeCheck(runTime)
}
go r.spawnWorkers(r.getSpawnCount(), r.getSpawnRate(), r.stoppingChan, nil) go r.spawnWorkers(r.getSpawnCount(), r.getSpawnRate(), r.stoppingChan, nil)
@@ -941,10 +943,7 @@ func (r *workerRunner) start() {
r.outputOnStart() r.outputOnStart()
runTime := r.getRunTime() go r.runTimeCheck(r.getRunTime())
if runTime != 0 {
go r.runTimeCheck(runTime)
}
go r.spawnWorkers(r.getSpawnCount(), r.getSpawnRate(), r.stoppingChan, r.spawnComplete) go r.spawnWorkers(r.getSpawnCount(), r.getSpawnRate(), r.stoppingChan, r.spawnComplete)