tablet/gorm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

insert_id only support auto_increment field

Browse Source
This commit is contained in:
liaoqiang 2015-06-04 16:39:07 +08:00
parent 0b8c9f29a9
commit 75ec40de67
2 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
model_struct.go
View File

@ -40,6 +40,7 @@ type StructField struct {
Tag reflect.StructTag Tag reflect.StructTag
Struct reflect.StructField Struct reflect.StructField
IsForeignKey bool IsForeignKey bool
IsAutoIncrement bool
Relationship *Relationship Relationship *Relationship
} }
@ -148,6 +149,10 @@ func (scope *Scope) GetModelStruct() *ModelStruct {
field.HasDefaultValue = true field.HasDefaultValue = true
} }
if _, ok := sqlSettings["AUTO_INCREMENT"]; ok {
field.IsAutoIncrement = true
}
if value, ok := gormSettings["COLUMN"]; ok { if value, ok := gormSettings["COLUMN"]; ok {
field.DBName = value field.DBName = value
} else { } else {

7
scope.go
View File

@ -112,12 +112,11 @@ func (scope *Scope) HasError() bool {
func (scope *Scope) PrimaryField() *Field { func (scope *Scope) PrimaryField() *Field {
if primaryFields := scope.GetModelStruct().PrimaryFields; len(primaryFields) > 0 { if primaryFields := scope.GetModelStruct().PrimaryFields; len(primaryFields) > 0 {
if len(primaryFields) > 1 { for i := 0; i < len(primaryFields); i++ {
if field, ok := scope.Fields()["id"]; ok { if primaryFields[i].IsAutoIncrement {
return field return scope.Fields()[primaryFields[i].DBName]
} }
} }
return scope.Fields()[primaryFields[0].DBName]
} }
return nil return nil
} }