169 lines
4.4 KiB
Go
169 lines
4.4 KiB
Go
|
package gofakeit
|
||
|
|
||
|
import (
|
||
|
"encoding/hex"
|
||
|
"math/rand"
|
||
|
"reflect"
|
||
|
|
||
|
"github.com/brianvoe/gofakeit/v6/data"
|
||
|
)
|
||
|
|
||
|
// Bool will generate a random boolean value
|
||
|
func Bool() bool { return boolFunc(globalFaker.Rand) }
|
||
|
|
||
|
// Bool will generate a random boolean value
|
||
|
func (f *Faker) Bool() bool { return boolFunc(f.Rand) }
|
||
|
|
||
|
func boolFunc(r *rand.Rand) bool { return randIntRange(r, 0, 1) == 1 }
|
||
|
|
||
|
// UUID (version 4) will generate a random unique identifier based upon random numbers
|
||
|
// Format: xxxxxxxx-xxxx-xxxx-xxxx-xxxxxxxxxxxx
|
||
|
func UUID() string { return uuid(globalFaker.Rand) }
|
||
|
|
||
|
// UUID (version 4) will generate a random unique identifier based upon random numbers
|
||
|
// Format: xxxxxxxx-xxxx-xxxx-xxxx-xxxxxxxxxxxx 8-4-4-4-12
|
||
|
func (f *Faker) UUID() string { return uuid(f.Rand) }
|
||
|
|
||
|
func uuid(r *rand.Rand) string {
|
||
|
version := byte(4)
|
||
|
uuid := make([]byte, 16)
|
||
|
|
||
|
// Commented out due to io.ReadFull not being race condition safe
|
||
|
// io.ReadFull(r, uuid[:])
|
||
|
|
||
|
// Read 16 random bytes
|
||
|
for i := 0; i < 16; i++ {
|
||
|
uuid[i] = byte(r.Intn(256))
|
||
|
}
|
||
|
|
||
|
// Set version
|
||
|
uuid[6] = (uuid[6] & 0x0f) | (version << 4)
|
||
|
|
||
|
// Set variant
|
||
|
uuid[8] = (uuid[8] & 0xbf) | 0x80
|
||
|
|
||
|
buf := make([]byte, 36)
|
||
|
hex.Encode(buf[0:8], uuid[0:4])
|
||
|
buf[8] = dash
|
||
|
hex.Encode(buf[9:13], uuid[4:6])
|
||
|
buf[13] = dash
|
||
|
hex.Encode(buf[14:18], uuid[6:8])
|
||
|
buf[18] = dash
|
||
|
hex.Encode(buf[19:23], uuid[8:10])
|
||
|
buf[23] = dash
|
||
|
hex.Encode(buf[24:], uuid[10:])
|
||
|
|
||
|
return string(buf)
|
||
|
}
|
||
|
|
||
|
// ShuffleAnySlice takes in a slice and outputs it in a random order
|
||
|
func ShuffleAnySlice(v any) { shuffleAnySlice(globalFaker.Rand, v) }
|
||
|
|
||
|
// ShuffleAnySlice takes in a slice and outputs it in a random order
|
||
|
func (f *Faker) ShuffleAnySlice(v any) { shuffleAnySlice(f.Rand, v) }
|
||
|
|
||
|
func shuffleAnySlice(r *rand.Rand, v any) {
|
||
|
if v == nil {
|
||
|
return
|
||
|
}
|
||
|
|
||
|
// Check type of passed in value, if not a slice return with no action taken
|
||
|
typ := reflect.TypeOf(v)
|
||
|
if typ.Kind() != reflect.Slice {
|
||
|
return
|
||
|
}
|
||
|
|
||
|
s := reflect.ValueOf(v)
|
||
|
n := s.Len()
|
||
|
|
||
|
if n <= 1 {
|
||
|
return
|
||
|
}
|
||
|
|
||
|
swap := func(i, j int) {
|
||
|
tmp := reflect.ValueOf(s.Index(i).Interface())
|
||
|
s.Index(i).Set(s.Index(j))
|
||
|
s.Index(j).Set(tmp)
|
||
|
}
|
||
|
|
||
|
//if size is > int32 probably it will never finish, or ran out of entropy
|
||
|
i := n - 1
|
||
|
for ; i > 0; i-- {
|
||
|
j := int(r.Int31n(int32(i + 1)))
|
||
|
swap(i, j)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// FlipACoin will return a random value of Heads or Tails
|
||
|
func FlipACoin() string { return flipACoin(globalFaker.Rand) }
|
||
|
|
||
|
// FlipACoin will return a random value of Heads or Tails
|
||
|
func (f *Faker) FlipACoin() string { return flipACoin(f.Rand) }
|
||
|
|
||
|
func flipACoin(r *rand.Rand) string {
|
||
|
if boolFunc(r) {
|
||
|
return "Heads"
|
||
|
}
|
||
|
|
||
|
return "Tails"
|
||
|
}
|
||
|
|
||
|
// RandomMapKey will return a random key from a map
|
||
|
func RandomMapKey(mapI any) any { return randomMapKey(globalFaker.Rand, mapI) }
|
||
|
|
||
|
// RandomMapKey will return a random key from a map
|
||
|
func (f *Faker) RandomMapKey(mapI any) any { return randomMapKey(f.Rand, mapI) }
|
||
|
|
||
|
func randomMapKey(r *rand.Rand, mapI any) any {
|
||
|
keys := reflect.ValueOf(mapI).MapKeys()
|
||
|
return keys[r.Intn(len(keys))].Interface()
|
||
|
}
|
||
|
|
||
|
// Categories will return a map string array of available data categories and sub categories
|
||
|
func Categories() map[string][]string {
|
||
|
types := make(map[string][]string)
|
||
|
for category, subCategoriesMap := range data.Data {
|
||
|
subCategories := make([]string, 0)
|
||
|
for subType := range subCategoriesMap {
|
||
|
subCategories = append(subCategories, subType)
|
||
|
}
|
||
|
types[category] = subCategories
|
||
|
}
|
||
|
return types
|
||
|
}
|
||
|
|
||
|
func addMiscLookup() {
|
||
|
AddFuncLookup("uuid", Info{
|
||
|
Display: "UUID",
|
||
|
Category: "misc",
|
||
|
Description: "128-bit identifier used to uniquely identify objects or entities in computer systems",
|
||
|
Example: "590c1440-9888-45b0-bd51-a817ee07c3f2",
|
||
|
Output: "string",
|
||
|
Generate: func(r *rand.Rand, m *MapParams, info *Info) (any, error) {
|
||
|
return uuid(r), nil
|
||
|
},
|
||
|
})
|
||
|
|
||
|
AddFuncLookup("bool", Info{
|
||
|
Display: "Boolean",
|
||
|
Category: "misc",
|
||
|
Description: "Data type that represents one of two possible values, typically true or false",
|
||
|
Example: "true",
|
||
|
Output: "bool",
|
||
|
Generate: func(r *rand.Rand, m *MapParams, info *Info) (any, error) {
|
||
|
return boolFunc(r), nil
|
||
|
},
|
||
|
})
|
||
|
|
||
|
AddFuncLookup("flipacoin", Info{
|
||
|
Display: "Flip A Coin",
|
||
|
Category: "misc",
|
||
|
Description: "Decision-making method involving the tossing of a coin to determine outcomes",
|
||
|
Example: "Tails",
|
||
|
Output: "string",
|
||
|
Generate: func(r *rand.Rand, m *MapParams, info *Info) (any, error) {
|
||
|
return flipACoin(r), nil
|
||
|
},
|
||
|
})
|
||
|
}
|