fix: resolve the problem of reporting last stats failure.

cli/hrp/cmd/boom.go

@@ -65,6 +65,7 @@ func init() {
 	boomCmd.Flags().StringVar(&requestIncreaseRate, "request-increase-rate", "-1", "Request increase rate, disabled by default.")
 	boomCmd.Flags().IntVar(&spawnCount, "spawn-count", 1, "The number of users to spawn for load testing")
 	boomCmd.Flags().Float64Var(&spawnRate, "spawn-rate", 1, "The rate for spawning users")
+	boomCmd.Flags().Int64Var(&loopCount, "loop-count", -1, "The specify running cycles for load testing")
 	boomCmd.Flags().StringVar(&memoryProfile, "mem-profile", "", "Enable memory profiling.")
 	boomCmd.Flags().DurationVar(&memoryProfileDuration, "mem-profile-duration", 30*time.Second, "Memory profile duration.")
 	boomCmd.Flags().StringVar(&cpuProfile, "cpu-profile", "", "Enable CPU profiling.")

internal/boomer/runner.go

@@ -273,12 +273,15 @@ func (r *localRunner) start() {
 	// all running workers(goroutines) will select on this channel.
 	// close this channel will stop all running workers.
 	quitChan := make(chan bool)
+	// when this channel is closed, all statistics are reported successfully
+	reportedChan := make(chan bool)
 	go r.spawnWorkers(r.spawnCount, r.spawnRate, quitChan, nil)
 
 	// output setup
 	r.outputOnStart()
 
 	// start running
+	go func() {
 		var ticker = time.NewTicker(reportStatsInterval)
 		for {
 			select {
@@ -293,30 +296,36 @@ func (r *localRunner) start() {
 			// report stats
 			case <-ticker.C:
 				r.reportStats()
+				// close reportedChan and return if the last stats is reported successfully
+				if atomic.LoadInt32(&r.state) == stateQuitting {
+					close(reportedChan)
+					return
+				}
+			}
+		}
+	}()
+
 	// stop
-		case <-r.stopChan:
+	<-r.stopChan
 	atomic.StoreInt32(&r.state, stateQuitting)
 
 	// stop previous goroutines without blocking
 	// those goroutines will exit when r.safeRun returns
 	close(quitChan)
 
+	// wait until all stats are reported successfully
+	<-reportedChan
+
 	// stop rate limiter
 	if r.rateLimitEnabled {
 		r.rateLimiter.Stop()
 	}
 
-			// report last stats
-			<-ticker.C
-			r.reportStats()
-
 	// output teardown
 	r.outputOnStop()
 
 	atomic.StoreInt32(&r.state, stateStopped)
 	return
-		}
-	}
 }
 
 func (r *localRunner) stop() {