Sarthak | WorkerGroup now supports Close method

SarthakMakhija · Aug 18, 2023 · faa79db · faa79db
1 parent fdfc528
commit faa79db
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/workers/worker_group.go b/workers/worker_group.go
@@ -41,6 +41,12 @@ func (group *WorkerGroup) Run() chan report.LoadGenerationResponse {
 	return loadGenerationResponseChannel
 }
 
+func (group *WorkerGroup) Close() {
+	for count := 1; count <= int(group.options.concurrency); count++ {
+		group.stopChannel <- struct{}{}
+	}
+}
+
 // TODO: close the connection if response reader is nil
 func (group *WorkerGroup) runWorkers(
 	loadGenerationResponseChannel chan report.LoadGenerationResponse,