callbacks.go 2.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051
  1. package callbacks
  2. import (
  3. "gorm.io/gorm"
  4. )
  5. type Config struct {
  6. LastInsertIDReversed bool
  7. WithReturning bool
  8. }
  9. func RegisterDefaultCallbacks(db *gorm.DB, config *Config) {
  10. enableTransaction := func(db *gorm.DB) bool {
  11. return !db.SkipDefaultTransaction
  12. }
  13. createCallback := db.Callback().Create()
  14. createCallback.Match(enableTransaction).Register("gorm:begin_transaction", BeginTransaction)
  15. createCallback.Register("gorm:before_create", BeforeCreate)
  16. createCallback.Register("gorm:save_before_associations", SaveBeforeAssociations)
  17. createCallback.Register("gorm:create", Create(config))
  18. createCallback.Register("gorm:save_after_associations", SaveAfterAssociations)
  19. createCallback.Register("gorm:after_create", AfterCreate)
  20. createCallback.Match(enableTransaction).Register("gorm:commit_or_rollback_transaction", CommitOrRollbackTransaction)
  21. queryCallback := db.Callback().Query()
  22. queryCallback.Register("gorm:query", Query)
  23. queryCallback.Register("gorm:preload", Preload)
  24. queryCallback.Register("gorm:after_query", AfterQuery)
  25. deleteCallback := db.Callback().Delete()
  26. deleteCallback.Match(enableTransaction).Register("gorm:begin_transaction", BeginTransaction)
  27. deleteCallback.Register("gorm:before_delete", BeforeDelete)
  28. deleteCallback.Register("gorm:delete_before_associations", DeleteBeforeAssociations)
  29. deleteCallback.Register("gorm:delete", Delete)
  30. deleteCallback.Register("gorm:after_delete", AfterDelete)
  31. deleteCallback.Match(enableTransaction).Register("gorm:commit_or_rollback_transaction", CommitOrRollbackTransaction)
  32. updateCallback := db.Callback().Update()
  33. updateCallback.Match(enableTransaction).Register("gorm:begin_transaction", BeginTransaction)
  34. updateCallback.Register("gorm:setup_reflect_value", SetupUpdateReflectValue)
  35. updateCallback.Register("gorm:before_update", BeforeUpdate)
  36. updateCallback.Register("gorm:save_before_associations", SaveBeforeAssociations)
  37. updateCallback.Register("gorm:update", Update)
  38. updateCallback.Register("gorm:save_after_associations", SaveAfterAssociations)
  39. updateCallback.Register("gorm:after_update", AfterUpdate)
  40. updateCallback.Match(enableTransaction).Register("gorm:commit_or_rollback_transaction", CommitOrRollbackTransaction)
  41. db.Callback().Row().Register("gorm:row", RowQuery)
  42. db.Callback().Raw().Register("gorm:raw", RawExec)
  43. }