From 8ae5ba81bca7f583de78e517049210761657b151 Mon Sep 17 00:00:00 2001 From: machongwei Date: Wed, 10 Aug 2022 22:03:24 +0800 Subject: [PATCH] feat: hrp boom support set run time --- hrp/internal/boomer/runner.go | 15 +++++++-------- 1 file changed, 7 insertions(+), 8 deletions(-) diff --git a/hrp/internal/boomer/runner.go b/hrp/internal/boomer/runner.go index 43c5fcfb..fce9fa83 100644 --- a/hrp/internal/boomer/runner.go +++ b/hrp/internal/boomer/runner.go @@ -380,10 +380,15 @@ func (r *runner) reset() { } func (r *runner) runTimeCheck(runTime int64) { + if runTime <= 0 { + return + } + for range time.Tick(time.Second * 3) { nowTime := time.Now().Unix() if nowTime > runTime { r.stop() + return } } } @@ -646,10 +651,7 @@ func (r *localRunner) start() { // output setup r.outputOnStart() - runTime := r.getRunTime() - if runTime != 0 { - go r.runTimeCheck(runTime) - } + go r.runTimeCheck(r.getRunTime()) go r.spawnWorkers(r.getSpawnCount(), r.getSpawnRate(), r.stoppingChan, nil) @@ -941,10 +943,7 @@ func (r *workerRunner) start() { r.outputOnStart() - runTime := r.getRunTime() - if runTime != 0 { - go r.runTimeCheck(runTime) - } + go r.runTimeCheck(r.getRunTime()) go r.spawnWorkers(r.getSpawnCount(), r.getSpawnRate(), r.stoppingChan, r.spawnComplete)