詳解如何利用GORM實現MySQL事務

2022-09-08 18:04:23

前言

為了確保資料一致性，在專案中會經常用到事務處理，回滾操作還是比較常見的需求；事務處理可以用來維護資料庫的完整性，保證成批的sql語句要麼全部都執行，要麼全不執行，對於MySQL事務相信大家應該都不陌生，這篇文章主要總結一下在Go語言中Gorm是如何實現事務的；感興趣的小夥伴們可以參考借鑑，希望對大家能有所幫助。

禁用預設事務

gorm事務預設是開啟的。為了確保資料一致性，Gorm會在事務裡執行寫入操作（增刪改）。

如果對資料一致性要求不高的話，可以在初始化時禁用它，效能將提升大約30%。

一般不推薦禁用。

// 全域性禁用
db, err := gorm.Open(sqlite.Open("gorm.db"), &gorm.Config{
  SkipDefaultTransaction: true,
})

自動事務

通過db.Transaction函數實現事務，如果閉包函數返回錯誤，則回滾事務。

db.Transaction(func(tx *gorm.DB) error {
  // 在事務中執行一些 db 操作（從這裡開始，您應該使用 'tx' 而不是 'db'）
  if err := tx.Create(&User{Name: "Lili"}).Error; err != nil {
    // 返回任何錯誤都會回滾事務
    return err
  }

  if err := tx.Create(&User{Name: "xiaoming"}).Error; err != nil {
    return err
  }

  // 返回 nil 提交事務
  return nil
})

手動事務

在開發中經常需要資料庫事務來保證多個資料庫寫操作的原子性。例如電商系統中的扣減庫存和儲存訂單。

gorm事務用法：

// 開啟事務
tx := db.Begin()

//在事務中執行資料庫操作，使用的是tx變數，不是db。

//庫存減一
//等價於: UPDATE `goods` SET `stock` = stock - 1  WHERE `goods`.`id` = '2' and stock > 0
//RowsAffected用於返回sql執行後影響的行數
rowsAffected := tx.Model(&goods).Where("stock > 0").Update("stock", gorm.Expr("stock - 1")).RowsAffected
if rowsAffected == 0 {
    //如果更新庫存操作，返回影響行數為0，說明沒有庫存了，結束下單流程
    //這裡回滾作用不大，因為前面沒成功執行什麼資料庫更新操作，也沒什麼資料需要回滾。
    //這裡就是舉個例子，事務中可以執行多個sql語句，錯誤了可以回滾事務
    tx.Rollback()
    return
}
err := tx.Create(儲存訂單).Error

//儲存訂單失敗，則回滾事務
if err != nil {
    tx.Rollback()
} else {
    tx.Commit()
}

巢狀事務

GORM 支援巢狀事務，可以回滾較大事務內執行的一部分操作，例如：

db.Transaction(func(tx *gorm.DB) error {
  tx.Create(&user1)

  tx.Transaction(func(tx2 *gorm.DB) error {
    tx2.Create(&user2)
    return errors.New("rollback user2") // Rollback user2
  })

  tx.Transaction(func(tx2 *gorm.DB) error {
    tx2.Create(&user3)
    return nil
  })

  return nil
})

// Commit user1, user3

SavePoint、RollbackTo

GORM 提供了 SavePoint、Rollbackto 來提供儲存點以及回滾至儲存點，例如：

tx := db.Begin()
tx.Create(&user1)
 
tx.SavePoint("sp1") 
tx.Create(&user2)
tx.RollbackTo("sp1") // Rollback user2
 
tx.Commit() // Commit user1

這裡rollback到了sp1的位置，也就是說，資料庫中只存了user1這條資料。

小結

關於Go GORM 事務詳細介紹的文章就介紹到這了，總的來說Gorm事務的實現非常簡單，go語言操作mysql事務主要用到了三個函數：

Db.Begin() 開始事務
Db.Commit() 提交事務
Db.Rollback() 回滾事務

到此這篇關於詳解如何利用GORM實現MySQL事務的文章就介紹到這了,更多相關Golang GORM實現MySQL事務內容請搜尋it145.com以前的文章或繼續瀏覽下面的相關文章希望大家以後多多支援it145.com！