golang?db事務的統(tǒng)一封裝的實現(xiàn)
事務處理的流程示例
database := db.DB
tx, err := database.Begin()
if err != nil {
return err
}
stmt, err := tx.Prepare(sqlQuery)
if err != nil {
tx.Rollback()
return err
}
_, err = stmt.Exec(paras...)
if err != nil {
tx.Rollback()
return err
}
err = tx.Commit()
if err != nil {
tx.Rollback()
return err
}
以上是我們使用事務時的一般操作，如果每做一次事務的操作均要進行重新寫一遍代碼豈不是很麻煩，尤其是出錯時，Rollback需要多次在不同錯誤的地方的進行調(diào)用處理。
簡單封裝
偷懶第一步
采用defer處理Rollback
defer tx.Rollback()
無論成功與否，均進行Rollback操作，只是有點影響，如果成功還調(diào)用Rollback的話，將會報錯。雖然可以忽略，但作為程序員，有必要進一步調(diào)整。
偷懶第二步
根據(jù)執(zhí)行結(jié)果來選擇執(zhí)行Rollback，避免無效使用。
defer func() { //根據(jù)執(zhí)行結(jié)果選擇執(zhí)行Rollback
if err != nil && tx != nil {
log.Println("ExecSqlWithTransaction defer err :", err)
tx.Rollback()
}
}()
如此，我們就可以根據(jù)事務的執(zhí)行結(jié)果決定是否Rollback了。
偷懶第三步
封裝，以上代碼本身就具有極大的普適性，因此，我們抽出通用的參數(shù)，將此過程封裝成一個func，以后就可以直接調(diào)用了。
func ExecSqlWithTransaction(database *sql.DB, query string, args ...interface{}) (err error) {
tx, err := database.Begin()
if err != nil {
return err
}
defer func() {
if err != nil && tx != nil {
tx.Rollback()
}
}()
stmt, err := tx.Prepare(query)
if err != nil {
return err
}
defer stmt.Close()
_, err = stmt.Exec(args...)
if err != nil {
return err
}
return tx.Commit()
}
封裝后我們可以如下使用：
if err := ExecSqlWithTransaction(database,sqlQuery,paras...);err != nil{
//錯誤處理
}
封裝后是不是很簡潔啊？
進一步封裝
在一個事務中可能會出現(xiàn)多個SELECT、UPDATE等操作，以上封裝僅處理了一種操作，還不能滿足我們的實際需求，因此需要更進一步封裝。
func ExecSqlWithTransaction(db *sql.DB, handle func(tx *sql.Tx) error) (err error) {
tx, err := db.Begin()
if err != nil {
return err
}
defer func() {
if err != nil {
tx.Rollback()
}
}()
if err = handle(tx); err != nil {
return err
}
return tx.Commit()
}
在handle func內(nèi)可以直接使用事務tx進行增刪改查。
到此這篇關于golang db事務的統(tǒng)一封裝的實現(xiàn)的文章就介紹到這了,更多相關golang db事務封裝內(nèi)容請搜索本站以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持本站！
版權聲明：本站文章來源標注為YINGSOO的內(nèi)容版權均為本站所有，歡迎引用、轉(zhuǎn)載，請保持原文完整并注明來源及原文鏈接。禁止復制或仿造本網(wǎng)站，禁止在非maisonbaluchon.cn所屬的服務器上建立鏡像，否則將依法追究法律責任。本站部分內(nèi)容來源于網(wǎng)友推薦、互聯(lián)網(wǎng)收集整理而來，僅供學習參考，不代表本站立場，如有內(nèi)容涉嫌侵權，請聯(lián)系alex-e#qq.com處理。
關注官方微信