123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051 |
- package callbacks
- import (
- "gorm.io/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_before_associations", DeleteBeforeAssociations)
- 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:row", RowQuery)
- db.Callback().Raw().Register("gorm:raw", RawExec)
- }
|