本文中作者展示了golang事務(wù)的三種寫(xiě)法。

第一種寫(xiě)法

這種寫(xiě)法非常樸實(shí)，程序流程也非常明確，但是事務(wù)處理與程序流程嵌入太深，容易遺漏，造成嚴(yán)重的問(wèn)題

funcDoSomething()(errerror){
tx,err:=db.Begin()
iferr!=nil{
return
}


deferfunc(){
ifp:=recover();p!=nil{
tx.Rollback()
panic(p)//re-throwpanicafterRollback
}
}()


if_,err=tx.Exec(...);err!=nil{
tx.Rollback()
return
}
if_,err=tx.Exec(...);err!=nil{
tx.Rollback()
return
}
//...


err=tx.Commit()
return
}

第二種寫(xiě)法

下面這種寫(xiě)法把事務(wù)處理從程序流程抽離了出來(lái)，不容易遺漏，但是作用域是整個(gè)函數(shù)，程序流程不是很清晰

funcDoSomething()(errerror){
tx,err:=db.Begin()
iferr!=nil{
return
}


deferfunc(){
ifp:=recover();p!=nil{
tx.Rollback()
panic(p)//re-throwpanicafterRollback
}elseiferr!=nil{
tx.Rollback()
}else{
err=tx.Commit()
}
}()


if_,err=tx.Exec(...);err!=nil{
return
}
if_,err=tx.Exec(...);err!=nil{
return
}
//...
return
}

第三種寫(xiě)法

寫(xiě)法三是對(duì)寫(xiě)法二的進(jìn)一步封裝，寫(xiě)法高級(jí)一點(diǎn)，缺點(diǎn)同上

funcTransact(db*sql.DB,txFuncfunc(*sql.Tx)error)(errerror){
tx,err:=db.Begin()
iferr!=nil{
return
}


deferfunc(){
ifp:=recover();p!=nil{
tx.Rollback()
panic(p)//re-throwpanicafterRollback
}elseiferr!=nil{
tx.Rollback()
}else{
err=tx.Commit()
}
}()


err=txFunc(tx)
returnerr
}


funcDoSomething()error{
returnTransact(db,func(tx*sql.Tx)error{
if_,err:=tx.Exec(...);err!=nil{
returnerr
}
if_,err:=tx.Exec(...);err!=nil{
returnerr
}
})
}

我的寫(xiě)法

經(jīng)過(guò)總結(jié)和實(shí)驗(yàn)，我采用了下面這種寫(xiě)法，defer tx.Rollback() 使得事務(wù)回滾始終得到執(zhí)行。當(dāng) tx.Commit() 執(zhí)行后，tx.Rollback() 起到關(guān)閉事務(wù)的作用， 當(dāng)程序因?yàn)槟硞€(gè)錯(cuò)誤中止，tx.Rollback() 起到回滾事務(wù)，同事關(guān)閉事務(wù)的作用。

普通場(chǎng)景

funcDoSomething()(errerror){
tx,_:=db.Begin()
defertx.Rollback()

if_,err=tx.Exec(...);err!=nil{
return
}
if_,err=tx.Exec(...);err!=nil{
return
}
//...


err=tx.Commit()
return
}

循環(huán)場(chǎng)景

(1) 小事務(wù) 每次循環(huán)提交一次 在循環(huán)內(nèi)部使用這種寫(xiě)法的時(shí)候，defer 不能使用，所以要把事務(wù)部分抽離到獨(dú)立的函數(shù)當(dāng)中

funcDoSomething()(errerror){
tx,_:=db.Begin()
defertx.Rollback()

if_,err=tx.Exec(...);err!=nil{
return
}
if_,err=tx.Exec(...);err!=nil{
return
}
//...


err=tx.Commit()
return
}


for{
iferr:=DoSomething();err!=nil{
//...
}
}

(2) 大事務(wù) 批量提交 大事務(wù)的場(chǎng)景和普通場(chǎng)景是一樣的，沒(méi)有任何區(qū)別

funcDoSomething()(errerror){
tx,_:=db.Begin()
defertx.Rollback()

for{
if_,err=tx.Exec(...);err!=nil{
return
}
if_,err=tx.Exec(...);err!=nil{
return
}
//...
}

err=tx.Commit()
return
}