如何優雅的使用 GORM 來實現 MySQL 事務
theme: awesome-green
我報名參加金石計劃1期挑戰——瓜分10萬獎池，這是我的第2篇文章，點選檢視活動詳情
前言
為了確保資料一致性，在專案中會經常用到事務處理，回滾操作還是比較常見的需求；事務處理可以用來維護資料庫的完整性，保證成批的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() 回滾事務