-
Notifications
You must be signed in to change notification settings - Fork 0
/
util.go
63 lines (58 loc) · 1.5 KB
/
util.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
package goeloquent
import (
"reflect"
"regexp"
"strings"
)
var matchFirstCap = regexp.MustCompile("(.)([A-Z][a-z]+)")
var matchAllCap = regexp.MustCompile("([a-z0-9])([A-Z])")
func ToSnakeCase(str string) string {
snake := matchFirstCap.ReplaceAllString(str, "${1}_${2}")
snake = matchAllCap.ReplaceAllString(snake, "${1}_${2}")
return strings.ToLower(snake)
}
func ExtractStruct(target interface{}) map[string]interface{} {
tv := reflect.Indirect(reflect.ValueOf(target))
tt := tv.Type()
result := make(map[string]interface{}, tv.NumField())
if tt.Kind() == reflect.Struct {
m := GetParsedModel(tt)
for _, key := range m.DbFields {
keyIndex := m.FieldsByDbName[key].Index
if !tv.Field(keyIndex).IsZero() {
result[key] = tv.Field(keyIndex).Interface()
}
}
} else {
for i := 0; i < tv.NumField(); i++ {
key := ToSnakeCase(tt.Field(i).Name)
result[key] = tv.Field(i).Interface()
}
}
return result
}
func InterfaceToSlice(param interface{}) []interface{} {
if p, ok := param.([]interface{}); ok {
return p
}
tv := reflect.Indirect(reflect.ValueOf(param))
var res []interface{}
if tv.Type().Kind() == reflect.Slice {
for i := 0; i < tv.Len(); i++ {
res = append(res, tv.Index(i).Interface())
}
} else {
panic("not slice")
}
return res
}
//StrPtr
//return a pointer of string helper function for create mapping
func StrPtr() *string {
return new(string)
}
//UintPtr
//return a pointer of uint64 helper function for create mapping
func UintPtr() *uint64 {
return new(uint64)
}