-
Notifications
You must be signed in to change notification settings - Fork 194
/
clienttoken.go
46 lines (38 loc) · 1.06 KB
/
clienttoken.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
package common
import (
"fmt"
"crypto/rand"
"reflect"
)
const (
fieldClientToken = "ClientToken"
)
func safeInjectClientToken(obj interface{}) {
// obj Must be struct ptr
getType := reflect.TypeOf(obj)
if getType.Kind() != reflect.Ptr || getType.Elem().Kind() != reflect.Struct {
return
}
// obj Must exist named field
_, ok := getType.Elem().FieldByName(fieldClientToken)
if !ok {
return
}
field := reflect.ValueOf(obj).Elem().FieldByName(fieldClientToken)
// field Must be string ptr
if field.Kind() != reflect.Ptr {
return
}
// Set if ClientToken is nil or empty
if field.IsNil() || (field.Elem().Kind() == reflect.String && field.Elem().Len() == 0) {
uuidVal := randomClientToken()
field.Set(reflect.ValueOf(&uuidVal))
}
}
// randomClientToken generate random string as ClientToken
// ref: https://stackoverflow.com/questions/15130321/is-there-a-method-to-generate-a-uuid-with-go-language
func randomClientToken() string {
b := make([]byte, 16)
_, _ = rand.Read(b)
return fmt.Sprintf("%X-%X-%X-%X-%X", b[0:4], b[4:6], b[6:8], b[8:10], b[10:])
}