事務(wù)處理的流程示例
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
|
database := db.DBtx, 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} |
以上是我們使用事務(wù)時(shí)的一般操作，如果每做一次事務(wù)的操作均要進(jìn)行重新寫一遍代碼豈不是很麻煩，尤其是出錯(cuò)時(shí)，Rollback需要多次在不同錯(cuò)誤的地方的進(jìn)行調(diào)用處理。
簡單封裝
偷懶第一步
采用defer處理Rollback
|
1
|
defer tx.Rollback() |
無論成功與否，均進(jìn)行Rollback操作，只是有點(diǎn)影響，如果成功還調(diào)用Rollback的話，將會(huì)報(bào)錯(cuò)。雖然可以忽略，但作為程序員，有必要進(jìn)一步調(diào)整。
偷懶第二步
根據(jù)執(zhí)行結(jié)果來選擇執(zhí)行Rollback，避免無效使用。
|
1
2
3
4
5
6
|
defer func() { //根據(jù)執(zhí)行結(jié)果選擇執(zhí)行Rollback if err != nil && tx != nil { log.Println("ExecSqlWithTransaction defer err :", err) tx.Rollback() } }() |
如此，我們就可以根據(jù)事務(wù)的執(zhí)行結(jié)果決定是否Rollback了。
偷懶第三步
封裝，以上代碼本身就具有極大的普適性，因此，我們抽出通用的參數(shù)，將此過程封裝成一個(gè)func，以后就可以直接調(diào)用了。
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
|
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()} |
封裝后我們可以如下使用：
|
1
2
3
|
if err := ExecSqlWithTransaction(database,sqlQuery,paras...);err != nil{ //錯(cuò)誤處理} |
封裝后是不是很簡潔啊？
進(jìn)一步封裝
在一個(gè)事務(wù)中可能會(huì)出現(xiàn)多個(gè)SELECT、UPDATE等操作，以上封裝僅處理了一種操作，還不能滿足我們的實(shí)際需求，因此需要更進(jìn)一步封裝。
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
|
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)可以直接使用事務(wù)tx進(jìn)行增刪改查。
到此這篇關(guān)于golang db事務(wù)的統(tǒng)一封裝的實(shí)現(xiàn)的文章就介紹到這了,更多相關(guān)golang db事務(wù)封裝內(nèi)容請搜索服務(wù)器之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持服務(wù)器之家！
原文鏈接：https://blog.csdn.net/xz_studying/article/details/90610993
