From 8d8a658508b19c2a1699fade8117f39eb48e78c5 Mon Sep 17 00:00:00 2001
From: buyuxiang <347586493@qq.com>
Date: Mon, 9 May 2022 18:15:28 +0800
Subject: [PATCH] fix: disable keep alive

---
 hrp/boomer.go   | 8 +++++---
 hrp/cmd/boom.go | 1 +
 2 files changed, 6 insertions(+), 3 deletions(-)

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()