forked from mirror/gorm
51 lines
2.2 KiB
Go
51 lines
2.2 KiB
Go
package callbacks
|
|
|
|
import (
|
|
"github.com/jinzhu/gorm"
|
|
)
|
|
|
|
type Config struct {
|
|
LastInsertIDReversed bool
|
|
WithReturning bool
|
|
}
|
|
|
|
func RegisterDefaultCallbacks(db *gorm.DB, config *Config) {
|
|
enableTransaction := func(db *gorm.DB) bool {
|
|
return !db.SkipDefaultTransaction
|
|
}
|
|
|
|
createCallback := db.Callback().Create()
|
|
createCallback.Match(enableTransaction).Register("gorm:begin_transaction", BeginTransaction)
|
|
createCallback.Register("gorm:before_create", BeforeCreate)
|
|
createCallback.Register("gorm:save_before_associations", SaveBeforeAssociations)
|
|
createCallback.Register("gorm:create", Create(config))
|
|
createCallback.Register("gorm:save_after_associations", SaveAfterAssociations)
|
|
createCallback.Register("gorm:after_create", AfterCreate)
|
|
createCallback.Match(enableTransaction).Register("gorm:commit_or_rollback_transaction", CommitOrRollbackTransaction)
|
|
|
|
queryCallback := db.Callback().Query()
|
|
queryCallback.Register("gorm:query", Query)
|
|
queryCallback.Register("gorm:preload", Preload)
|
|
queryCallback.Register("gorm:after_query", AfterQuery)
|
|
|
|
deleteCallback := db.Callback().Delete()
|
|
deleteCallback.Match(enableTransaction).Register("gorm:begin_transaction", BeginTransaction)
|
|
deleteCallback.Register("gorm:before_delete", BeforeDelete)
|
|
deleteCallback.Register("gorm:delete", Delete)
|
|
deleteCallback.Register("gorm:after_delete", AfterDelete)
|
|
deleteCallback.Match(enableTransaction).Register("gorm:commit_or_rollback_transaction", CommitOrRollbackTransaction)
|
|
|
|
updateCallback := db.Callback().Update()
|
|
updateCallback.Match(enableTransaction).Register("gorm:begin_transaction", BeginTransaction)
|
|
updateCallback.Register("gorm:setup_reflect_value", SetupUpdateReflectValue)
|
|
updateCallback.Register("gorm:before_update", BeforeUpdate)
|
|
updateCallback.Register("gorm:save_before_associations", SaveBeforeAssociations)
|
|
updateCallback.Register("gorm:update", Update)
|
|
updateCallback.Register("gorm:save_after_associations", SaveAfterAssociations)
|
|
updateCallback.Register("gorm:after_update", AfterUpdate)
|
|
updateCallback.Match(enableTransaction).Register("gorm:commit_or_rollback_transaction", CommitOrRollbackTransaction)
|
|
|
|
db.Callback().Row().Register("gorm:raw", RowQuery)
|
|
db.Callback().Raw().Register("gorm:raw", RawExec)
|
|
}
|