执行数据库事务接口

最后更新于:2022-04-02 06:48:00

## 执行数据库事务接口 在处理数据库连接服务时,测试相对困难。原因在于Go的鸭子模型导致在运行时模拟很不方便。我建议在使用数据库时使用存储接口,使用这样的方式模拟数据库事务接口同样可行,在后续章节会讨论这一点。本节将着眼于对数据库事务的操作。 我们将改写上一节的create和query文件,最终的输出很类似,但是create和query操作将包含在事务中。 ### 实践 1. 建立 transaction.go: ``` package dbinterface import "database/sql" // DB是sql.DB或sql.Transaction满足的接口 type DB interface { Exec(query string, args ...interface{}) (sql.Result, error) Prepare(query string) (*sql.Stmt, error) Query(query string, args ...interface{}) (*sql.Rows, error) QueryRow(query string, args ...interface{}) *sql.Row } // Transaction可以执行任何 Query, Commit, Rollback, 和 Stmt 操作 type Transaction interface { DB Commit() error Rollback() error Stmt(stmt *sql.Stmt) *sql.Stmt } ``` 2. 建立 create.go: ``` package dbinterface import _ "github.com/go-sql-driver/mysql" // Create建立example表并填充数据 func Create(db DB) error { // create the database if _, err := db.Exec("CREATE TABLE example (name VARCHAR(20), created DATETIME)"); err != nil { return err } if _, err := db.Exec(`INSERT INTO example (name, created) values ("Aaron", NOW())`); err != nil { return err } return nil } ``` 3. 建立 exec.go: ``` package dbinterface func Exec(db DB) error { // 依然不推荐这么写 因为没有处理所返回的可能存在的错误 defer db.Exec("DROP TABLE example") if err := Create(db); err != nil { return err } if err := Query(db); err != nil { return err } return nil } ``` 4. 建立 main.go: ``` package main import ( "github.com/agtorre/go-cookbook/chapter5/database" "github.com/agtorre/go-cookbook/chapter5/dbinterface" _ "github.com/go-sql-driver/mysql" ) func main() { db, err := database.Setup() if err != nil { panic(err) } tx, err := db.Begin() if err != nil { panic(err) } defer tx.Rollback() if err := dbinterface.Exec(db); err != nil { panic(err) } if err := tx.Commit(); err != nil { panic(err) } } ``` 5. 这会输出: ``` Results: Name: Aaron Created: 2017-02-16 20:00:00 +0000 UTC ``` ### 说明 本节与上一节以非常相似的方式工作,演示了使用事务并生成适用于sql.DB连接和sql.Transaction对象的通用数据库函数。我们会在第八章看到对这些模拟接口的测试使用。 * * * * 学识浅薄,错误在所难免。欢迎在群中就本书提出修改意见,以飨后来者,长风拜谢。 Golang中国(211938256) beego实战(258969317) Go实践(386056972)
';