diff --git a/hrp/boomer.go b/hrp/boomer.go index 377e61f3..c47db199 100644 --- a/hrp/boomer.go +++ b/hrp/boomer.go @@ -19,9 +19,6 @@ func NewBoomer(spawnCount int, spawnRate float64) *HRPBoomer { } b.hrpRunner = NewRunner(nil) - // set client transport for high concurrency load testing - b.hrpRunner.SetClientTransport(b.GetSpawnCount(), b.GetDisableKeepAlive(), b.GetDisableCompression()) - return b } @@ -32,6 +29,11 @@ type HRPBoomer struct { pluginsMutex *sync.RWMutex // avoid data race } +func (b *HRPBoomer) SetClientTransport() { + // set client transport for high concurrency load testing + b.hrpRunner.SetClientTransport(b.GetSpawnCount(), b.GetDisableKeepAlive(), b.GetDisableCompression()) +} + // Run starts to run load test for one or multiple testcases. func (b *HRPBoomer) Run(testcases ...ITestCase) { event := sdk.EventTracking{ diff --git a/hrp/cmd/boom.go b/hrp/cmd/boom.go index 2751ab52..1350b981 100644 --- a/hrp/cmd/boom.go +++ b/hrp/cmd/boom.go @@ -41,6 +41,7 @@ var boomCmd = &cobra.Command{ } hrpBoomer.SetDisableKeepAlive(disableKeepalive) hrpBoomer.SetDisableCompression(disableCompression) + hrpBoomer.SetClientTransport() hrpBoomer.EnableCPUProfile(cpuProfile, cpuProfileDuration) hrpBoomer.EnableMemoryProfile(memoryProfile, memoryProfileDuration) hrpBoomer.EnableGracefulQuit()