forked from mirror/gorm
112 lines
3.1 KiB
Go
112 lines
3.1 KiB
Go
package callbacks
|
|
|
|
import (
|
|
"reflect"
|
|
|
|
"gorm.io/gorm"
|
|
"gorm.io/gorm/clause"
|
|
"gorm.io/gorm/schema"
|
|
)
|
|
|
|
func BeforeDelete(db *gorm.DB) {
|
|
if db.Error == nil && db.Statement.Schema != nil && db.Statement.Schema.BeforeDelete {
|
|
tx := db.Session(&gorm.Session{})
|
|
callMethod := func(value interface{}) bool {
|
|
if db.Statement.Schema.BeforeDelete {
|
|
if i, ok := value.(gorm.BeforeDeleteInterface); ok {
|
|
db.AddError(i.BeforeDelete(tx))
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
if ok := callMethod(db.Statement.Dest); !ok {
|
|
switch db.Statement.ReflectValue.Kind() {
|
|
case reflect.Slice, reflect.Array:
|
|
for i := 0; i < db.Statement.ReflectValue.Len(); i++ {
|
|
callMethod(db.Statement.ReflectValue.Index(i).Interface())
|
|
}
|
|
case reflect.Struct:
|
|
callMethod(db.Statement.ReflectValue.Interface())
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
func Delete(db *gorm.DB) {
|
|
if db.Error == nil {
|
|
if db.Statement.Schema != nil && !db.Statement.Unscoped {
|
|
for _, c := range db.Statement.Schema.DeleteClauses {
|
|
db.Statement.AddClause(c)
|
|
}
|
|
}
|
|
|
|
if db.Statement.SQL.String() == "" {
|
|
db.Statement.AddClauseIfNotExists(clause.Delete{})
|
|
|
|
if db.Statement.Schema != nil {
|
|
_, queryValues := schema.GetIdentityFieldValuesMap(db.Statement.ReflectValue, db.Statement.Schema.PrimaryFields)
|
|
column, values := schema.ToQueryValues(db.Statement.Schema.PrimaryFieldDBNames, queryValues)
|
|
|
|
if len(values) > 0 {
|
|
db.Statement.AddClause(clause.Where{Exprs: []clause.Expression{clause.IN{Column: column, Values: values}}})
|
|
}
|
|
|
|
if db.Statement.Dest != db.Statement.Model && db.Statement.Model != nil {
|
|
_, queryValues = schema.GetIdentityFieldValuesMap(reflect.ValueOf(db.Statement.Model), db.Statement.Schema.PrimaryFields)
|
|
column, values = schema.ToQueryValues(db.Statement.Schema.PrimaryFieldDBNames, queryValues)
|
|
|
|
if len(values) > 0 {
|
|
db.Statement.AddClause(clause.Where{Exprs: []clause.Expression{clause.IN{Column: column, Values: values}}})
|
|
}
|
|
}
|
|
}
|
|
|
|
if _, ok := db.Statement.Clauses["WHERE"]; !ok {
|
|
db.AddError(gorm.ErrMissingWhereClause)
|
|
return
|
|
}
|
|
|
|
db.Statement.AddClauseIfNotExists(clause.From{})
|
|
db.Statement.Build("DELETE", "FROM", "WHERE")
|
|
}
|
|
|
|
if !db.DryRun {
|
|
result, err := db.Statement.ConnPool.ExecContext(db.Statement.Context, db.Statement.SQL.String(), db.Statement.Vars...)
|
|
|
|
if err == nil {
|
|
db.RowsAffected, _ = result.RowsAffected()
|
|
} else {
|
|
db.AddError(err)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
func AfterDelete(db *gorm.DB) {
|
|
if db.Error == nil && db.Statement.Schema != nil && db.Statement.Schema.AfterDelete {
|
|
tx := db.Session(&gorm.Session{})
|
|
callMethod := func(value interface{}) bool {
|
|
if db.Statement.Schema.AfterDelete {
|
|
if i, ok := value.(gorm.AfterDeleteInterface); ok {
|
|
db.AddError(i.AfterDelete(tx))
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
if ok := callMethod(db.Statement.Dest); !ok {
|
|
switch db.Statement.ReflectValue.Kind() {
|
|
case reflect.Slice, reflect.Array:
|
|
for i := 0; i < db.Statement.ReflectValue.Len(); i++ {
|
|
callMethod(db.Statement.ReflectValue.Index(i).Interface())
|
|
}
|
|
case reflect.Struct:
|
|
callMethod(db.Statement.ReflectValue.Interface())
|
|
}
|
|
}
|
|
}
|
|
}
|