🔨 Refactor client and request generation to use random value cycling

requests/client.go

@@ -280,23 +280,7 @@ func getDialFunc(proxy *config.Proxy, timeout time.Duration) (fasthttp.DialFunc,
 // getSharedClientFuncMultiple returns a ClientGeneratorFunc that cycles through a list of fasthttp.HostClient instances.
 // The function uses a local random number generator to determine the starting index and stop index for cycling through the clients.
 func getSharedClientFuncMultiple(clients []*fasthttp.HostClient, localRand *rand.Rand) ClientGeneratorFunc {
-	var (
-		clientsCount int = len(clients)
-		currentIndex int = localRand.Intn(clientsCount)
-		stopIndex    int = clientsCount
-	)
-
-	return func() *fasthttp.HostClient {
-		client := clients[currentIndex%clientsCount]
-		if currentIndex == stopIndex {
-			currentIndex = localRand.Intn(clientsCount)
-			stopIndex = currentIndex - 1
-		} else {
-			currentIndex = (currentIndex + 1) % clientsCount
-		}
-
-		return client
-	}
+	return utils.RandomValueCycle(clients, localRand)
 }
 
 // getSharedClientFuncSingle returns a ClientGeneratorFunc that always returns the provided fasthttp.HostClient instance.

requests/request.go

@@ -8,6 +8,7 @@ import (
 
 	"github.com/aykhans/dodo/config"
 	customerrors "github.com/aykhans/dodo/custom_errors"
+	"github.com/aykhans/dodo/utils"
 	"github.com/valyala/fasthttp"
 )
 
@@ -108,19 +109,7 @@ func getRequestGeneratorFunc(
 	if bodiesLen == 1 {
 		getBody = func() string { return Bodies[0] }
 	} else if bodiesLen > 1 {
-		currentIndex := localRand.Intn(bodiesLen)
-		stopIndex := bodiesLen - 1
-
-		getBody = func() string {
-			body := Bodies[currentIndex%bodiesLen]
-			if currentIndex == stopIndex {
-				currentIndex = localRand.Intn(bodiesLen)
-				stopIndex = currentIndex - 1
-			} else {
-				currentIndex = (currentIndex + 1) % bodiesLen
-			}
-			return body
-		}
+		getBody = utils.RandomValueCycle(Bodies, localRand)
 	}
 	getHeaders := getKeyValueSetFunc(Headers, localRand)
 	getCookies := getKeyValueSetFunc(Cookies, localRand)
@@ -226,20 +215,7 @@ func getKeyValueSetFunc[
 		if valuesLen == 1 {
 			getKeyValue = func() string { return values[0] }
 		} else if valuesLen > 1 {
-			currentIndex := localRand.Intn(valuesLen)
-			stopIndex := valuesLen - 1
-
-			getKeyValue = func() string {
-				value := values[currentIndex%valuesLen]
-				if currentIndex == stopIndex {
-					currentIndex = localRand.Intn(valuesLen)
-					stopIndex = currentIndex - 1
-				} else {
-					currentIndex = (currentIndex + 1) % valuesLen
-				}
-				return value
-			}
-
+			getKeyValue = utils.RandomValueCycle(values, localRand)
 			isRandom = true
 		}
 

utils/slice.go

@@ -1,5 +1,7 @@
 package utils
 
+import "math/rand"
+
 func Flatten[T any](nested [][]*T) []*T {
 	flattened := make([]*T, 0)
 	for _, n := range nested {
@@ -16,3 +18,25 @@ func Contains[T comparable](slice []T, item T) bool {
 	}
 	return false
 }
+
+func RandomValueCycle[Value any](values []Value, localRand *rand.Rand) func() Value {
+	var (
+		clientsCount int = len(values)
+		currentIndex int = localRand.Intn(clientsCount)
+		stopIndex    int = currentIndex
+	)
+
+	return func() Value {
+		client := values[currentIndex]
+		currentIndex++
+		if currentIndex == clientsCount {
+			currentIndex = 0
+		}
+		if currentIndex == stopIndex {
+			currentIndex = localRand.Intn(clientsCount)
+			stopIndex = currentIndex
+		}
+
+		return client
+	}
+}