-
Notifications
You must be signed in to change notification settings - Fork 0
/
utils.go
70 lines (60 loc) · 1.25 KB
/
utils.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
package main
import (
"encoding/json"
"log"
"math/rand"
"strings"
"time"
)
type stringType int
const (
numericType stringType = iota
alphanumericType
alphabeticalType
)
func toJSON(params map[string][]string) string {
flattenedParams := make(map[string]string)
for k, p := range params {
flattenedParams[k] = p[0]
}
js, err := json.Marshal(flattenedParams)
if err != nil {
log.Fatal(err)
}
return strings.Replace(string(js), ",", ", ", -1)
}
func getRandomASCIIIndex(t stringType) int {
rand.Seed(time.Now().UnixNano())
switch t {
case numericType:
return randInt(48, 57)
case alphanumericType:
coinFlip := rand.Intn(2)
if coinFlip == 0 {
// Random character is a number
return randInt(48, 57)
}
// Random character is a letter
return randInt(65, 90)
case alphabeticalType:
coinFlip := rand.Intn(2)
if coinFlip == 0 {
// Random uppercase character
return randInt(65, 90)
}
// Random lowercase character
return randInt(97, 122)
default:
return randInt(48, 57)
}
}
func randomString(l int, t stringType) string {
bytes := make([]byte, l)
for i := 0; i < l; i++ {
bytes[i] = byte(getRandomASCIIIndex(t))
}
return string(bytes)
}
func randInt(min int, max int) int {
return min + rand.Intn(max-min)
}