Model对象也可以通过TX事务接口创建,通过事务对象创建的Model对象与通过DB数据库对象创建的Model对象功能是一样的,只不过前者的所有操作都是基于事务,而当事务提交或者回滚后,对应的Model对象不能被继续使用,否则会返回错误。因为该TX接口不能被继续使用,一个事务对象仅对应于一个事务流程,Commit/Rollback后即结束。

本章节仅对链式操作涉及到的事务处理方法做简单介绍,更详细的介绍请参考 ORM事务处理 章节。

示例1,通过Transaction

为方便事务操作,gdb提供了事务的闭包操作,通过Transaction方法实现,该方法定义如下:

  1. func (db DB) Transaction(ctx context.Context, f func(ctx context.Context, tx TX) error) (err error)

当给定的闭包方法返回的errornil时,那么闭包执行结束后当前事务自动执行Commit提交操作;否则自动执行Rollback回滚操作。

如果闭包内部操作产生panic中断,该事务也将进行回滚。

  1. func Register() error {
  2. return g.DB().Transaction(ctx, func(ctx context.Context, tx gdb.TX) error {
  3. var (
  4. result sql.Result
  5. err error
  6. )
  7. // 写入用户基础数据
  8. result, err = tx.Table("user").Insert(g.Map{
  9. "name": "john",
  10. "score": 100,
  11. //...
  12. })
  13. if err != nil {
  14. return err
  15. }
  16. // 写入用户详情数据,需要用到上一次写入得到的用户uid
  17. result, err = tx.Table("user_detail").Insert(g.Map{
  18. "uid": result.LastInsertId(),
  19. "phone": "18010576258",
  20. //...
  21. })
  22. return err
  23. })
  24. }

示例2,通过TX链式操作

我们也可以在链式操作中通过TX方法切换绑定的事务对象。多次链式操作可以绑定同一个事务对象,在该事务对象中执行对应的链式操作。

  1. func Register() error {
  2. var (
  3. uid int64
  4. err error
  5. )
  6. tx, err := g.DB().Begin()
  7. if err != nil {
  8. return err
  9. }
  10. // 方法退出时检验返回值,
  11. // 如果结果成功则执行tx.Commit()提交,
  12. // 否则执行tx.Rollback()回滚操作。
  13. defer func() {
  14. if err != nil {
  15. tx.Rollback()
  16. } else {
  17. tx.Commit()
  18. }
  19. }()
  20. // 写入用户基础数据
  21. uid, err = AddUserInfo(tx, g.Map{
  22. "name": "john",
  23. "score": 100,
  24. //...
  25. })
  26. if err != nil {
  27. return err
  28. }
  29. // 写入用户详情数据,需要用到上一次写入得到的用户uid
  30. err = AddUserDetail(tx, g.Map{
  31. "uid": uid,
  32. "phone": "18010576259",
  33. //...
  34. })
  35. return err
  36. }
  37. func AddUserInfo(tx gdb.TX, data g.Map) (int64, error) {
  38. result, err := g.Model("user").TX(tx).Data(data).Insert()
  39. if err != nil {
  40. return 0, err
  41. }
  42. uid, err := result.LastInsertId()
  43. if err != nil {
  44. return 0, err
  45. }
  46. return uid, nil
  47. }
  48. func AddUserDetail(tx gdb.TX, data g.Map) error {
  49. _, err := g.Model("user_detail").TX(tx).Data(data).Insert()
  50. return err
  51. }