mirror of https://github.com/go-gorm/gorm.git
79 lines
1.7 KiB
Go
79 lines
1.7 KiB
Go
package gorm
|
|
|
|
import (
|
|
"database/sql"
|
|
"errors"
|
|
"reflect"
|
|
"time"
|
|
)
|
|
|
|
type relationship struct {
|
|
JoinTable string
|
|
ForeignKey string
|
|
ForeignType string
|
|
AssociationForeignKey string
|
|
Kind string
|
|
}
|
|
|
|
// FIXME
|
|
func (r relationship) ForeignDBName() string {
|
|
return ToSnake(r.ForeignKey)
|
|
}
|
|
|
|
func (r relationship) AssociationForeignDBName(name string) string {
|
|
return ToSnake(r.AssociationForeignKey)
|
|
}
|
|
|
|
type Field struct {
|
|
Name string
|
|
DBName string
|
|
Field reflect.Value
|
|
Tag reflect.StructTag
|
|
Relationship *relationship
|
|
IsNormal bool
|
|
IsBlank bool
|
|
IsIgnored bool
|
|
IsPrimaryKey bool
|
|
DefaultValue interface{}
|
|
}
|
|
|
|
func (field *Field) IsScanner() bool {
|
|
_, isScanner := reflect.New(field.Field.Type()).Interface().(sql.Scanner)
|
|
return isScanner
|
|
}
|
|
|
|
func (field *Field) IsTime() bool {
|
|
reflectValue := field.Field
|
|
if reflectValue.Kind() == reflect.Ptr {
|
|
reflectValue = reflect.New(reflectValue.Type().Elem()).Elem()
|
|
}
|
|
_, isTime := reflectValue.Interface().(time.Time)
|
|
return isTime
|
|
}
|
|
|
|
func (field *Field) Set(value interface{}) (err error) {
|
|
if !field.Field.IsValid() {
|
|
return errors.New("field value not valid")
|
|
}
|
|
|
|
if !field.Field.CanAddr() {
|
|
return errors.New("field value not addressable")
|
|
}
|
|
|
|
if rvalue, ok := value.(reflect.Value); ok {
|
|
value = rvalue.Interface()
|
|
}
|
|
|
|
if scanner, ok := field.Field.Addr().Interface().(sql.Scanner); ok {
|
|
scanner.Scan(value)
|
|
} else if reflect.TypeOf(value).ConvertibleTo(field.Field.Type()) {
|
|
field.Field.Set(reflect.ValueOf(value).Convert(field.Field.Type()))
|
|
} else {
|
|
return errors.New("could not convert argument")
|
|
}
|
|
|
|
field.IsBlank = isBlank(field.Field)
|
|
|
|
return
|
|
}
|