mirror of
https://github.com/aykhans/dodo.git
synced 2025-04-16 01:43:13 +00:00
Aykhan Shahsuvarov
00f0bcb2de
- Moved readers to the config package - Added an option to read remote config files - Moved the validation package to the config package and removed the validator dependency - Moved the customerrors package to the config package - Replaced fatih/color with jedib0t/go-pretty/v6/text - Removed proxy check functionality - Added param, header, cookie, body, and proxy flags to the CLI - Allowed multiple values for the same key in params, headers, and cookies
42 lines
1.3 KiB
Go
42 lines
1.3 KiB
Go
package utils
|
|
|
|
import "math/rand"
|
|
|
|
func Flatten[T any](nested [][]*T) []*T {
|
|
flattened := make([]*T, 0)
|
|
for _, n := range nested {
|
|
flattened = append(flattened, n...)
|
|
}
|
|
return flattened
|
|
}
|
|
|
|
// RandomValueCycle returns a function that cycles through the provided slice of values
|
|
// in a random order. Each call to the returned function will yield a value from the slice.
|
|
// The order of values is determined by the provided random number generator.
|
|
//
|
|
// The returned function will cycle through the values in a random order until all values
|
|
// have been returned at least once. After all values have been returned, the function will
|
|
// reset and start cycling through the values in a random order again.
|
|
// The returned function isn't thread-safe and should be used in a single-threaded context.
|
|
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
|
|
}
|
|
}
|