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

optimization callbacks/create.go Create func if...else logic

Browse Source
This commit is contained in:
daheige 2021-05-27 22:50:55 +08:00
parent c27b80ab28
commit 5dbcc21152
1 changed files with 61 additions and 56 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

117
callbacks/create.go
View File

@ -33,75 +33,80 @@ func BeforeCreate(db *gorm.DB) {
func Create(config *Config) func(db *gorm.DB) { func Create(config *Config) func(db *gorm.DB) {
if config.WithReturning { if config.WithReturning {
return CreateWithReturning return CreateWithReturning
} else { }
return func(db *gorm.DB) {
if db.Error == nil {
if db.Statement.Schema != nil && !db.Statement.Unscoped {
for _, c := range db.Statement.Schema.CreateClauses {
db.Statement.AddClause(c)
}
}
if db.Statement.SQL.String() == "" { return func(db *gorm.DB) {
db.Statement.SQL.Grow(180) if db.Error != nil {
db.Statement.AddClauseIfNotExists(clause.Insert{}) // maybe record logger TODO
db.Statement.AddClause(ConvertToCreateValues(db.Statement)) return
}
db.Statement.Build(db.Statement.BuildClauses...) if db.Statement.Schema != nil && !db.Statement.Unscoped {
} for _, c := range db.Statement.Schema.CreateClauses {
db.Statement.AddClause(c)
}
}
if !db.DryRun && db.Error == nil { if db.Statement.SQL.String() == "" {
result, err := db.Statement.ConnPool.ExecContext(db.Statement.Context, db.Statement.SQL.String(), db.Statement.Vars...) db.Statement.SQL.Grow(180)
db.Statement.AddClauseIfNotExists(clause.Insert{})
db.Statement.AddClause(ConvertToCreateValues(db.Statement))
if err == nil { db.Statement.Build(db.Statement.BuildClauses...)
db.RowsAffected, _ = result.RowsAffected() }
if db.RowsAffected > 0 { if !db.DryRun && db.Error == nil {
if db.Statement.Schema != nil && db.Statement.Schema.PrioritizedPrimaryField != nil && db.Statement.Schema.PrioritizedPrimaryField.HasDefaultValue { result, err := db.Statement.ConnPool.ExecContext(db.Statement.Context, db.Statement.SQL.String(), db.Statement.Vars...)
if insertID, err := result.LastInsertId(); err == nil && insertID > 0 {
switch db.Statement.ReflectValue.Kind() {
case reflect.Slice, reflect.Array:
if config.LastInsertIDReversed {
for i := db.Statement.ReflectValue.Len() - 1; i >= 0; i-- {
rv := db.Statement.ReflectValue.Index(i)
if reflect.Indirect(rv).Kind() != reflect.Struct {
break
}
_, isZero := db.Statement.Schema.PrioritizedPrimaryField.ValueOf(rv) if err != nil {
if isZero { db.AddError(err)
db.Statement.Schema.PrioritizedPrimaryField.Set(rv, insertID) }
insertID -= db.Statement.Schema.PrioritizedPrimaryField.AutoIncrementIncrement
}
}
} else {
for i := 0; i < db.Statement.ReflectValue.Len(); i++ {
rv := db.Statement.ReflectValue.Index(i)
if reflect.Indirect(rv).Kind() != reflect.Struct {
break
}
if _, isZero := db.Statement.Schema.PrioritizedPrimaryField.ValueOf(rv); isZero { db.RowsAffected, _ = result.RowsAffected()
db.Statement.Schema.PrioritizedPrimaryField.Set(rv, insertID) if !(db.RowsAffected > 0) {
insertID += db.Statement.Schema.PrioritizedPrimaryField.AutoIncrementIncrement return
} }
}
} if db.Statement.Schema != nil && db.Statement.Schema.PrioritizedPrimaryField != nil && db.Statement.Schema.PrioritizedPrimaryField.HasDefaultValue {
case reflect.Struct: if insertID, err := result.LastInsertId(); err == nil && insertID > 0 {
if _, isZero := db.Statement.Schema.PrioritizedPrimaryField.ValueOf(db.Statement.ReflectValue); isZero { switch db.Statement.ReflectValue.Kind() {
db.Statement.Schema.PrioritizedPrimaryField.Set(db.Statement.ReflectValue, insertID) case reflect.Slice, reflect.Array:
} if config.LastInsertIDReversed {
} for i := db.Statement.ReflectValue.Len() - 1; i >= 0; i-- {
} else { rv := db.Statement.ReflectValue.Index(i)
db.AddError(err) if reflect.Indirect(rv).Kind() != reflect.Struct {
break
}
_, isZero := db.Statement.Schema.PrioritizedPrimaryField.ValueOf(rv)
if isZero {
db.Statement.Schema.PrioritizedPrimaryField.Set(rv, insertID)
insertID -= db.Statement.Schema.PrioritizedPrimaryField.AutoIncrementIncrement
}
}
} else {
for i := 0; i < db.Statement.ReflectValue.Len(); i++ {
rv := db.Statement.ReflectValue.Index(i)
if reflect.Indirect(rv).Kind() != reflect.Struct {
break
}
if _, isZero := db.Statement.Schema.PrioritizedPrimaryField.ValueOf(rv); isZero {
db.Statement.Schema.PrioritizedPrimaryField.Set(rv, insertID)
insertID += db.Statement.Schema.PrioritizedPrimaryField.AutoIncrementIncrement
} }
} }
} }
} else { case reflect.Struct:
db.AddError(err) if _, isZero := db.Statement.Schema.PrioritizedPrimaryField.ValueOf(db.Statement.ReflectValue); isZero {
db.Statement.Schema.PrioritizedPrimaryField.Set(db.Statement.ReflectValue, insertID)
}
} }
} else {
db.AddError(err)
} }
} }
} }
} }
} }