mirror of https://github.com/go-gorm/gorm.git
130 lines
2.6 KiB
Go
130 lines
2.6 KiB
Go
package gorm
|
|
|
|
import (
|
|
"database/sql"
|
|
"reflect"
|
|
"strconv"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
type Field struct {
|
|
Name string
|
|
Value interface{}
|
|
model *Model
|
|
dbName string
|
|
isPrimaryKey bool
|
|
autoCreateTime bool
|
|
autoUpdateTime bool
|
|
foreignKey string
|
|
beforeAssociation bool
|
|
afterAssociation bool
|
|
reflectValue reflect.Value
|
|
structField reflect.StructField
|
|
}
|
|
|
|
func (f *Field) isBlank() bool {
|
|
return isBlank(f.reflectValue)
|
|
}
|
|
|
|
func (f *Field) isScanner() bool {
|
|
_, is_scanner := reflect.New(f.reflectValue.Type()).Interface().(sql.Scanner)
|
|
return is_scanner
|
|
}
|
|
|
|
func (f *Field) isTime() bool {
|
|
_, is_time := f.Value.(time.Time)
|
|
return is_time
|
|
}
|
|
|
|
func (f *Field) sqlTag() (str string) {
|
|
value := f.Value
|
|
if f.isScanner() {
|
|
value = f.reflectValue.Field(0).Interface()
|
|
}
|
|
reflect_value := f.reflectValue
|
|
|
|
switch reflect_value.Kind() {
|
|
case reflect.Slice:
|
|
return
|
|
case reflect.Struct:
|
|
if !f.isTime() && !f.isScanner() {
|
|
return
|
|
}
|
|
}
|
|
|
|
typ, addational_typ, size := parseSqlTag(f.structField.Tag.Get(f.model.do.db.tagIdentifier))
|
|
|
|
if typ == "-" {
|
|
return
|
|
}
|
|
|
|
if len(typ) == 0 {
|
|
if f.isPrimaryKey {
|
|
typ = f.model.do.db.dialect.PrimaryKeyTag(value, size)
|
|
} else {
|
|
typ = f.model.do.db.dialect.SqlTag(value, size)
|
|
}
|
|
}
|
|
|
|
if len(addational_typ) > 0 {
|
|
typ = typ + " " + addational_typ
|
|
}
|
|
return typ
|
|
}
|
|
|
|
func (f *Field) parseAssociation() {
|
|
reflect_value := f.reflectValue
|
|
|
|
switch reflect_value.Kind() {
|
|
case reflect.Slice:
|
|
foreign_key := f.model.typeName() + "Id"
|
|
if reflect.New(reflect_value.Type().Elem()).Elem().FieldByName(foreign_key).IsValid() {
|
|
f.foreignKey = foreign_key
|
|
}
|
|
f.afterAssociation = true
|
|
case reflect.Struct:
|
|
if !f.isTime() && !f.isScanner() {
|
|
if f.model.reflectData().FieldByName(f.Name + "Id").IsValid() {
|
|
f.foreignKey = f.Name + "Id"
|
|
f.beforeAssociation = true
|
|
} else {
|
|
foreign_key := f.model.typeName() + "Id"
|
|
if reflect.New(reflect_value.Type()).Elem().FieldByName(foreign_key).IsValid() {
|
|
f.foreignKey = foreign_key
|
|
}
|
|
f.afterAssociation = true
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
func parseSqlTag(str string) (typ string, addational_typ string, size int) {
|
|
if str == "-" {
|
|
typ = str
|
|
} else if str != "" {
|
|
tags := strings.Split(str, ";")
|
|
m := make(map[string]string)
|
|
for _, value := range tags {
|
|
v := strings.Split(value, ":")
|
|
k := strings.Trim(strings.ToUpper(v[0]), " ")
|
|
if len(v) == 2 {
|
|
m[k] = v[1]
|
|
} else {
|
|
m[k] = k
|
|
}
|
|
}
|
|
|
|
if len(m["SIZE"]) > 0 {
|
|
size, _ = strconv.Atoi(m["SIZE"])
|
|
}
|
|
|
|
if len(m["TYPE"]) > 0 {
|
|
typ = m["TYPE"]
|
|
}
|
|
|
|
addational_typ = m["NOT NULL"] + " " + m["UNIQUE"]
|
|
}
|
|
return
|
|
}
|