执行数据库事务接口

在处理数据库连接服务时，测试相对困难。原因在于Go的鸭子模型导致在运行时模拟很不方便。我建议在使用数据库时使用存储接口，使用这样的方式模拟数据库事务接口同样可行，在后续章节会讨论这一点。本节将着眼于对数据库事务的操作。

我们将改写上一节的create和query文件，最终的输出很类似，但是create和query操作将包含在事务中。

实践

建立 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
}

建立 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
}

建立 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
}

建立 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)
    }
}

这会输出：

Results:
    Name: Aaron
    Created: 2017-02-16 20:00:00 +0000 UTC

说明

本节与上一节以非常相似的方式工作，演示了使用事务并生成适用于sql.DB连接和sql.Transaction对象的通用数据库函数。我们会在第八章看到对这些模拟接口的测试使用。