re
/
gorm
forked from mirror/gorm
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

More compatible prioritized primary field #3156

This commit is contained in:
Jinzhu 2020-08-30 20:57:58 +08:00
parent b4166d9515
commit 53f8c9fc1c
1 changed files with 11 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
schema/schema.go
View File

@ -161,13 +161,18 @@ func Parse(dest interface{}, cacheStore *sync.Map, namer Namer) (*Schema, error)
field.setupValuerAndSetter() field.setupValuerAndSetter()
} }
if f := schema.LookUpField("id"); f != nil { prioritizedPrimaryField := schema.LookUpField("id")
if f.PrimaryKey { if prioritizedPrimaryField == nil {
schema.PrioritizedPrimaryField = f prioritizedPrimaryField = schema.LookUpField("ID")
}
if prioritizedPrimaryField != nil {
if prioritizedPrimaryField.PrimaryKey {
schema.PrioritizedPrimaryField = prioritizedPrimaryField
} else if len(schema.PrimaryFields) == 0 { } else if len(schema.PrimaryFields) == 0 {
f.PrimaryKey = true prioritizedPrimaryField.PrimaryKey = true
schema.PrioritizedPrimaryField = f schema.PrioritizedPrimaryField = prioritizedPrimaryField
schema.PrimaryFields = append(schema.PrimaryFields, f) schema.PrimaryFields = append(schema.PrimaryFields, prioritizedPrimaryField)
} }
} }