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(tagIdentifier)) if typ == "-" { return } if len(typ) == 0 { if f.isPrimaryKey { typ = f.model.do.chain.d.dialect.PrimaryKeyTag(value, size) } else { typ = f.model.do.chain.d.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 }