在前文介绍访问数据库时介绍了
github.com/jmoiron/sqlx
包,本文基于这个包使用数据库事务。
defer
在使用数据库事务之前,首先需要了解go语言的defer关键字。defer是go语言的延迟执行语句,defer后面的语句会被go进行延迟处理,在函数即将结束的时候,defer后面的语句将逆序执行。也就是说,先defer的语句最后执行。defer很像java或者C#中的finally语句。下面通过一个例子看一下defer。
package main
import "fmt"
func main() {
fmt.Println("defer 开始")
defer fmt.Println(1)
defer fmt.Println(2)
defer fmt.Println(3)
panic("测试 panic")
fmt.Println("defer 结束")
}
代码运行结果:
defer 开始
3
2
1
panic: 测试 panic
goroutine 1 [running]:
main.main()
/tests/gookokok/main.go:10 +0x1b8
Process finished with exit code 2
通过运行结果可以看到defer的逆序输出。在之后又手动触发了一个panic,影响了最后一行的输出
defer 结束
,go在触发panic时优先执行了defer,足以证明defer是非常安全的,所以defer也常常被用来互斥解锁、关闭文件或数据库事务的处理。
事务
sqlx使用事务和
database/sql
相比扩展出了MustBegin()、MustExec()等方法,这样就不需要在代码中手动处理很多错误。MustBegin会在出现错误的时候触发panic而不是返回错误,这样就可以在代码的更上一层统一处理错误。
tx, err := db.Begin()
err = tx.Exec(...)
err = tx.Commit()
tx := db.MustBegin()
tx.MustExec(...)
err = tx.Commit()
sqlx支持以上两种开启事务的方法。MustBegin返回sqlx.Tx,sqlx.Tx也提供了Select,Get之类的API,执行数据库操作和使用sqlx.DB是一样的。
以上代码执行
tx.MustExec(...)
如果报错的话,代码将没有机会运行到
tx.Commit()
,这样数据库连接会等到go进行垃圾回收的时候才能关闭,而且很高并发的话,可能会占满数据库连接数,造成站点无法访问的情况。
tx := db.MustBegin()
defer tx.Rollback()
tx.MustExec(...)
err = tx.Commit()
代码中加入
defer tx.Rollback()
就可以解决问题。通过前面的介绍已知defer是在方法即将结束时执行,哪怕是代码出现异常也不会影响数据库连接。
文章出处:基于gin的golang web开发:使用数据库事务