Adding workers flag to configure higher worker counts for super high …

…load testing
loveholidays · Jan 4, 2023 · 07796d7 · 07796d7
1 parent 24a7748
commit 07796d7
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/main.go b/main.go
@@ -34,10 +34,11 @@ func main() {
 	timeout := flag.Int("timeout", 10, "HTTP client timeout in seconds")
 	strict := flag.Bool("strict", false, "Panic on bad input")
 	memprofile := flag.String("memprofile", "", "Write memory profile to `file` before exit")
+	numWorkers := flag.Int("workers", 1000, "Number of client workers to use")
 
 	flag.Parse()
 
-	exitCode := ripley.Replay(*paceStr, *silent, *dryRun, *timeout, *strict)
+	exitCode := ripley.Replay(*paceStr, *silent, *dryRun, *timeout, *strict, *numWorkers)
 	defer os.Exit(exitCode)
 
 	if *memprofile != "" {

diff --git a/pkg/replay.go b/pkg/replay.go
@@ -27,7 +27,7 @@ import (
 	"time"
 )
 
-func Replay(phasesStr string, silent, dryRun bool, timeout int, strict bool) int {
+func Replay(phasesStr string, silent, dryRun bool, timeout int, strict bool, numWorkers int) int {
 	// Default exit code
 	var exitCode int = 0
 	// Ensures we have handled all HTTP request results before exiting
@@ -52,7 +52,7 @@ func Replay(phasesStr string, silent, dryRun bool, timeout int, strict bool) int
 	scanner := bufio.NewScanner(os.Stdin)
 
 	// Start HTTP client goroutine pool
-	startClientWorkers(1000, requests, results, dryRun, timeout)
+	startClientWorkers(numWorkers, requests, results, dryRun, timeout)
 	pacer.start()
 
 	for scanner.Scan() {