71
vendor/github.com/gocraft/dbr/util.go
generated
vendored
Normal file
71
vendor/github.com/gocraft/dbr/util.go
generated
vendored
Normal file
@@ -0,0 +1,71 @@
|
||||
package dbr
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"database/sql/driver"
|
||||
"reflect"
|
||||
"unicode"
|
||||
)
|
||||
|
||||
func camelCaseToSnakeCase(name string) string {
|
||||
buf := new(bytes.Buffer)
|
||||
|
||||
runes := []rune(name)
|
||||
|
||||
for i := 0; i < len(runes); i++ {
|
||||
buf.WriteRune(unicode.ToLower(runes[i]))
|
||||
if i != len(runes)-1 && unicode.IsUpper(runes[i+1]) &&
|
||||
(unicode.IsLower(runes[i]) || unicode.IsDigit(runes[i]) ||
|
||||
(i != len(runes)-2 && unicode.IsLower(runes[i+2]))) {
|
||||
buf.WriteRune('_')
|
||||
}
|
||||
}
|
||||
|
||||
return buf.String()
|
||||
}
|
||||
|
||||
func structMap(value reflect.Value) map[string]reflect.Value {
|
||||
m := make(map[string]reflect.Value)
|
||||
structValue(m, value)
|
||||
return m
|
||||
}
|
||||
|
||||
var (
|
||||
typeValuer = reflect.TypeOf((*driver.Valuer)(nil)).Elem()
|
||||
)
|
||||
|
||||
func structValue(m map[string]reflect.Value, value reflect.Value) {
|
||||
if value.Type().Implements(typeValuer) {
|
||||
return
|
||||
}
|
||||
switch value.Kind() {
|
||||
case reflect.Ptr:
|
||||
if value.IsNil() {
|
||||
return
|
||||
}
|
||||
structValue(m, value.Elem())
|
||||
case reflect.Struct:
|
||||
t := value.Type()
|
||||
for i := 0; i < t.NumField(); i++ {
|
||||
field := t.Field(i)
|
||||
if field.PkgPath != "" && !field.Anonymous {
|
||||
// unexported
|
||||
continue
|
||||
}
|
||||
tag := field.Tag.Get("db")
|
||||
if tag == "-" {
|
||||
// ignore
|
||||
continue
|
||||
}
|
||||
if tag == "" {
|
||||
// no tag, but we can record the field name
|
||||
tag = camelCaseToSnakeCase(field.Name)
|
||||
}
|
||||
fieldValue := value.Field(i)
|
||||
if _, ok := m[tag]; !ok {
|
||||
m[tag] = fieldValue
|
||||
}
|
||||
structValue(m, fieldValue)
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user