对于数据库执行多个更新操作时，事务会将多个操作当成单个单元处理，要成功都成功，否则都失败。本文通过实例说明Go Sql事务处理。
1. 环境准备
搭建PostgreSql数据库环境，我们打算在postgre数据库中新建表meta_data(id, source)，然后插入记录测试事务。
go操作数据库的包为"database/sql"，这里数据库使用postgreSql，需要导入github.com/lib/pq；下面是准备数据库连接代码。
import (
"database/sql"
"fmt"
_ "github.com/lib/pq"
uuid "github.com/satori/go.uuid"
"log"
"strings"
)
const (
host = "192.168.0.111"
port = 5432
user = "yourDbUser"
password = "yourDbPassword"
dbname = "postgres"
)
var db *sql.DB
func init() {
var err error
connStr := fmt.Sprintf("host=%s port=%d user=%s password=%s dbname=%s sslmode=disable", host, port, user, password, dbname)
db, err = sql.Open("postgres", connStr)
if err != nil {
log.Fatal(err)
}
db.SetConnMaxLifetime(500)
db.SetMaxIdleConns(50)
db.SetMaxOpenConns(10)
db.Stats()
}
func CheckError(err error) {
if err != nil {
panic(err)
}
}
go默认实现了数据库连接池，无需引入第三方连接池实现。
2. 事务实现
下面是事务示例实现，我们使用postgreSql，因此变量占位符使用 $ , MySql使用 ？。
func InsertData() {
data := [] string{
"DGraph","Redis"}
sql := "INSERT INTO meta_data(id, source) VALUES($1, $2)"
tx,err := db.Begin()
CheckError(err)
defer tx.Commit()
for _, item := range data {
sid := strings.Replace(uuid.NewV4().String(),"-","",-1)
_, err = db.Exec(sql, sid, item)
if err != nil {
_ = tx.Rollback()
break
}
}
}
函数开始之前首先启动事务，使用defer语句确保最后提交事务。接着执行多个插入语句，如果有错误回滚事务。
这个实现没有问题，只是每次需要写和事务相关的代码，显得多余，最好能封装事务相关代码，用户只关心业务。
3. 事务操作封装
上节手动实现了事务，本节对事务操作进行封装，让代码更有通用性。
这里定义UpdateWithTx()函数，其中封装了事务相关代码，具体执行和数据库相关操作通过其函数参数传入。
func UpdateWithTx(fn func(tx *sql.Tx) error) error {
var tx,err = db.Begin()
if err != nil{
return err
}
defer func() {
switch err {
case nil:
err = tx.Commit()
default:
tx.Rollback()
}
}()
err = fn(tx)
return err
}
首先获得事务指针，接着写匿名函数，使用defer关键词确保函数执行完成之前自动管控事务。最后执行实际操作的函数并返回状态。下面我们测试该函数。
func TestUpdateWithTx(t *testing.T) {
UpdateWithTx(func(tx *sql.Tx) error {
var err error
data := [] string{
"DGraph1","Redis1","MongoDb","BoltDB"}
sql := "INSERT INTO meta_data(id, source) VALUES($1, $2)"
for _, item := range data {
sid := strings.Replace(uuid.NewV4().String(),"-","",-1)
if _, err = tx.Exec(sql, sid, item); err != nil{
break
}
// err = errors.New("人为错误")
}
return err
})
}
具体业务和上节代码一样，但我们不在关心事务相关代码，用户可测试有错误时是否回滚事务。
4. 总结
本文通过示例介绍了Go Sql事务实现，并对事务相关操作进行封装，使用户只需关心业务操作。
本文参考链接：https://blog.csdn.net/neweastsun/article/details/107817240