From 8a25b1359c75efe57170b8a41af80d03f3235385 Mon Sep 17 00:00:00 2001 From: machongwei Date: Wed, 10 Aug 2022 21:59:16 +0800 Subject: [PATCH] feat: hrp boom support set run time --- hrp/internal/boomer/boomer.go | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/hrp/internal/boomer/boomer.go b/hrp/internal/boomer/boomer.go index 6c700229..58b8f56c 100644 --- a/hrp/internal/boomer/boomer.go +++ b/hrp/internal/boomer/boomer.go @@ -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) } }