forked from mirror/gorm
Refact untils.go
This commit is contained in:
parent
67b9e2a9ad
commit
2d3bcf5297
8
field.go
8
field.go
|
@ -23,13 +23,13 @@ type Field struct {
|
||||||
}
|
}
|
||||||
|
|
||||||
func (f *Field) IsScanner() bool {
|
func (f *Field) IsScanner() bool {
|
||||||
_, is_scanner := reflect.New(reflect.ValueOf(f.Value).Type()).Interface().(sql.Scanner)
|
_, isScanner := reflect.New(reflect.ValueOf(f.Value).Type()).Interface().(sql.Scanner)
|
||||||
return is_scanner
|
return isScanner
|
||||||
}
|
}
|
||||||
|
|
||||||
func (f *Field) IsTime() bool {
|
func (f *Field) IsTime() bool {
|
||||||
_, is_time := f.Value.(time.Time)
|
_, isTime := f.Value.(time.Time)
|
||||||
return is_time
|
return isTime
|
||||||
}
|
}
|
||||||
|
|
||||||
func parseSqlTag(str string) (typ string, additionalType string, size int) {
|
func parseSqlTag(str string) (typ string, additionalType string, size int) {
|
||||||
|
|
2
scope.go
2
scope.go
|
@ -82,7 +82,7 @@ func (scope *Scope) PrimaryKey() string {
|
||||||
return scope.primaryKey
|
return scope.primaryKey
|
||||||
}
|
}
|
||||||
|
|
||||||
scope.primaryKey = scope.getPrimaryKey()
|
scope.primaryKey = ToSnake(GetPrimaryKey(scope.Value))
|
||||||
return scope.primaryKey
|
return scope.primaryKey
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -5,7 +5,6 @@ import (
|
||||||
"database/sql/driver"
|
"database/sql/driver"
|
||||||
"errors"
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
"go/ast"
|
|
||||||
"reflect"
|
"reflect"
|
||||||
"regexp"
|
"regexp"
|
||||||
"strconv"
|
"strconv"
|
||||||
|
@ -466,33 +465,3 @@ func (scope *Scope) autoMigrate() *Scope {
|
||||||
}
|
}
|
||||||
return scope
|
return scope
|
||||||
}
|
}
|
||||||
|
|
||||||
func (scope *Scope) getPrimaryKey() string {
|
|
||||||
var indirectValue reflect.Value
|
|
||||||
|
|
||||||
indirectValue = reflect.Indirect(reflect.ValueOf(scope.Value))
|
|
||||||
|
|
||||||
if indirectValue.Kind() == reflect.Slice {
|
|
||||||
indirectValue = reflect.New(indirectValue.Type().Elem()).Elem()
|
|
||||||
}
|
|
||||||
|
|
||||||
if !indirectValue.IsValid() {
|
|
||||||
return "id"
|
|
||||||
}
|
|
||||||
|
|
||||||
scopeTyp := indirectValue.Type()
|
|
||||||
for i := 0; i < scopeTyp.NumField(); i++ {
|
|
||||||
fieldStruct := scopeTyp.Field(i)
|
|
||||||
if !ast.IsExported(fieldStruct.Name) {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
|
|
||||||
// if primaryKey tag found, return column name
|
|
||||||
if fieldStruct.Tag.Get("primaryKey") != "" {
|
|
||||||
return ToSnake(fieldStruct.Name)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
//If primaryKey tag not found, fallback to id
|
|
||||||
return "id"
|
|
||||||
}
|
|
||||||
|
|
90
utils.go
90
utils.go
|
@ -2,12 +2,8 @@ package gorm
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"bytes"
|
"bytes"
|
||||||
"database/sql"
|
"go/ast"
|
||||||
"fmt"
|
|
||||||
"os"
|
|
||||||
"reflect"
|
"reflect"
|
||||||
"regexp"
|
|
||||||
"runtime"
|
|
||||||
"strings"
|
"strings"
|
||||||
"sync"
|
"sync"
|
||||||
)
|
)
|
||||||
|
@ -95,82 +91,26 @@ func SnakeToUpperCamel(s string) string {
|
||||||
return u
|
return u
|
||||||
}
|
}
|
||||||
|
|
||||||
func toSearchableMap(attrs ...interface{}) (result interface{}) {
|
func GetPrimaryKey(value interface{}) string {
|
||||||
if len(attrs) > 1 {
|
var indirectValue = reflect.Indirect(reflect.ValueOf(value))
|
||||||
if str, ok := attrs[0].(string); ok {
|
|
||||||
result = map[string]interface{}{str: attrs[1]}
|
if indirectValue.Kind() == reflect.Slice {
|
||||||
}
|
indirectValue = reflect.New(indirectValue.Type().Elem()).Elem()
|
||||||
} else if len(attrs) == 1 {
|
|
||||||
if attr, ok := attrs[0].(map[string]interface{}); ok {
|
|
||||||
result = attr
|
|
||||||
}
|
}
|
||||||
|
|
||||||
if attr, ok := attrs[0].(interface{}); ok {
|
if indirectValue.IsValid() {
|
||||||
result = attr
|
scopeTyp := indirectValue.Type()
|
||||||
}
|
for i := 0; i < scopeTyp.NumField(); i++ {
|
||||||
}
|
fieldStruct := scopeTyp.Field(i)
|
||||||
return
|
if !ast.IsExported(fieldStruct.Name) {
|
||||||
|
continue
|
||||||
}
|
}
|
||||||
|
|
||||||
func fileWithLineNum() string {
|
if fieldStruct.Tag.Get("primaryKey") != "" {
|
||||||
for i := 1; i < 15; i++ {
|
return fieldStruct.Name
|
||||||
_, file, line, ok := runtime.Caller(i)
|
|
||||||
if ok && (!regexp.MustCompile(`jinzhu/gorm/.*.go`).MatchString(file) || regexp.MustCompile(`jinzhu/gorm/.*test.go`).MatchString(file)) {
|
|
||||||
return fmt.Sprintf("%v:%v", strings.TrimPrefix(file, os.Getenv("GOPATH")+"src/"), line)
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
return ""
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func setFieldValue(field reflect.Value, value interface{}) (result bool) {
|
return "Id"
|
||||||
result = false
|
|
||||||
if field.IsValid() && field.CanAddr() {
|
|
||||||
result = true
|
|
||||||
if scanner, ok := field.Addr().Interface().(sql.Scanner); ok {
|
|
||||||
scanner.Scan(value)
|
|
||||||
} else if reflect.TypeOf(value).ConvertibleTo(field.Type()) {
|
|
||||||
field.Set(reflect.ValueOf(value).Convert(field.Type()))
|
|
||||||
} else {
|
|
||||||
result = false
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
func isBlank(value reflect.Value) bool {
|
|
||||||
return reflect.DeepEqual(value.Interface(), reflect.Zero(value.Type()).Interface())
|
|
||||||
}
|
|
||||||
|
|
||||||
func convertInterfaceToMap(values interface{}) map[string]interface{} {
|
|
||||||
attrs := map[string]interface{}{}
|
|
||||||
|
|
||||||
switch value := values.(type) {
|
|
||||||
case map[string]interface{}:
|
|
||||||
for k, v := range value {
|
|
||||||
attrs[ToSnake(k)] = v
|
|
||||||
}
|
|
||||||
case []interface{}:
|
|
||||||
for _, v := range value {
|
|
||||||
for key, value := range convertInterfaceToMap(v) {
|
|
||||||
attrs[key] = value
|
|
||||||
}
|
|
||||||
}
|
|
||||||
case interface{}:
|
|
||||||
reflectValue := reflect.ValueOf(values)
|
|
||||||
|
|
||||||
switch reflectValue.Kind() {
|
|
||||||
case reflect.Map:
|
|
||||||
for _, key := range reflectValue.MapKeys() {
|
|
||||||
attrs[ToSnake(key.Interface().(string))] = reflectValue.MapIndex(key).Interface()
|
|
||||||
}
|
|
||||||
default:
|
|
||||||
scope := Scope{Value: values}
|
|
||||||
for _, field := range scope.Fields() {
|
|
||||||
if !field.IsBlank {
|
|
||||||
attrs[field.DBName] = field.Value
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return attrs
|
|
||||||
}
|
}
|
||||||
|
|
|
@ -0,0 +1,91 @@
|
||||||
|
package gorm
|
||||||
|
|
||||||
|
import (
|
||||||
|
"database/sql"
|
||||||
|
"fmt"
|
||||||
|
"os"
|
||||||
|
"reflect"
|
||||||
|
"regexp"
|
||||||
|
"runtime"
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
|
func fileWithLineNum() string {
|
||||||
|
for i := 1; i < 15; i++ {
|
||||||
|
_, file, line, ok := runtime.Caller(i)
|
||||||
|
if ok && (!regexp.MustCompile(`jinzhu/gorm/.*.go`).MatchString(file) || regexp.MustCompile(`jinzhu/gorm/.*test.go`).MatchString(file)) {
|
||||||
|
return fmt.Sprintf("%v:%v", strings.TrimPrefix(file, os.Getenv("GOPATH")+"src/"), line)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
|
||||||
|
func setFieldValue(field reflect.Value, value interface{}) (result bool) {
|
||||||
|
result = false
|
||||||
|
if field.IsValid() && field.CanAddr() {
|
||||||
|
result = true
|
||||||
|
if scanner, ok := field.Addr().Interface().(sql.Scanner); ok {
|
||||||
|
scanner.Scan(value)
|
||||||
|
} else if reflect.TypeOf(value).ConvertibleTo(field.Type()) {
|
||||||
|
field.Set(reflect.ValueOf(value).Convert(field.Type()))
|
||||||
|
} else {
|
||||||
|
result = false
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func isBlank(value reflect.Value) bool {
|
||||||
|
return reflect.DeepEqual(value.Interface(), reflect.Zero(value.Type()).Interface())
|
||||||
|
}
|
||||||
|
|
||||||
|
func toSearchableMap(attrs ...interface{}) (result interface{}) {
|
||||||
|
if len(attrs) > 1 {
|
||||||
|
if str, ok := attrs[0].(string); ok {
|
||||||
|
result = map[string]interface{}{str: attrs[1]}
|
||||||
|
}
|
||||||
|
} else if len(attrs) == 1 {
|
||||||
|
if attr, ok := attrs[0].(map[string]interface{}); ok {
|
||||||
|
result = attr
|
||||||
|
}
|
||||||
|
|
||||||
|
if attr, ok := attrs[0].(interface{}); ok {
|
||||||
|
result = attr
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func convertInterfaceToMap(values interface{}) map[string]interface{} {
|
||||||
|
attrs := map[string]interface{}{}
|
||||||
|
|
||||||
|
switch value := values.(type) {
|
||||||
|
case map[string]interface{}:
|
||||||
|
for k, v := range value {
|
||||||
|
attrs[ToSnake(k)] = v
|
||||||
|
}
|
||||||
|
case []interface{}:
|
||||||
|
for _, v := range value {
|
||||||
|
for key, value := range convertInterfaceToMap(v) {
|
||||||
|
attrs[key] = value
|
||||||
|
}
|
||||||
|
}
|
||||||
|
case interface{}:
|
||||||
|
reflectValue := reflect.ValueOf(values)
|
||||||
|
|
||||||
|
switch reflectValue.Kind() {
|
||||||
|
case reflect.Map:
|
||||||
|
for _, key := range reflectValue.MapKeys() {
|
||||||
|
attrs[ToSnake(key.Interface().(string))] = reflectValue.MapIndex(key).Interface()
|
||||||
|
}
|
||||||
|
default:
|
||||||
|
scope := Scope{Value: values}
|
||||||
|
for _, field := range scope.Fields() {
|
||||||
|
if !field.IsBlank {
|
||||||
|
attrs[field.DBName] = field.Value
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return attrs
|
||||||
|
}
|
Loading…
Reference in New Issue