Go基础之--操作Mysql

Posted 2020-10-26 coder

事务是数据库的一个非常重要的特性，尤其对于银行，支付系统，等等。
database/sql提供了事务处理的功能。通过Tx对象实现。db.Begin会创建tx对象，后者的Exec和Query执行事务的数据库操作，最后在tx的Commit和Rollback中完成数据库事务的提交和回滚，同时释放连接。

tx对象

我们在之前查询以及操作数据库都是用的db对象，而事务则是使用另外一个对象.
使用db.Begin 方法可以创建tx对象，tx对象也可以对数据库交互的Query,Exec方法
用法和我们之前操作基本一样，但是需要在查询或者操作完毕之后执行tx对象的Commit提交或者Rollback方法回滚。

一旦创建了tx对象，事务处理都依赖于tx对象，这个对象会从连接池中取出一个空闲的连接，接下来的sql执行都基于这个连接，知道commit或者Roolback调用之后，才会把这个连接释放到连接池。

在事务处理的时候，不能使用db的查询方法，当然你如果使用也能执行语句成功，但是这和你事务里执行的操作将不是一个事务，将不会接受commit和rollback的改变，如下面操作时：

tx,err := Db.Begin()
Db.Exec()
tx.Exec()
tx.Commit()

上面这个伪代码中，调用Db.Exec方法的时候，和tx执行Exec方法时候是不同的，只有tx的会绑定到事务中，db则是额外的一个连接，两者不是同一个事务。

事务与连接

创建Tx对象的时候，会从连接池中取出连接，然后调用相关的Exec方法的时候，连接仍然会绑定在该事务处理中。
事务的连接生命周期从Beigin函数调用起，直到Commit和Rollback函数的调用结束。

事务并发

对于sql.Tx对象，因为事务过程只有一个连接，事务内的操作都是顺序执行的，在开始下一个数据库交互之前，必须先完成上一个数据库交互。

rows, _ := db.Query("SELECT id FROM user") 
for rows.Next() {
    var mid, did int
    rows.Scan(&mid)
    db.QueryRow("SELECT id FROM detail_user WHERE master = ?", mid).Scan(&did)

}

调用了Query方法之后，在Next方法中取结果的时候，rows是维护了一个连接，再次调用QueryRow的时候，db会再从连接池取出一个新的连接。rows和db的连接两者可以并存，并且相互不影响。

但是如果逻辑在事务处理中会失效，如下代码：

rows, _ := tx.Query("SELECT id FROM user")
for rows.Next() {
   var mid, did int
   rows.Scan(&mid)
   tx.QueryRow("SELECT id FROM detail_user WHERE master = ?", mid).Scan(&did)
}

tx执行了Query方法后，连接转移到rows上，在Next方法中，tx.QueryRow将尝试获取该连接进行数据库操作。因为还没有调用rows.Close，因此底层的连接属于busy状态，tx是无法再进行查询的。

完整的小结

通过下面一个完整的例子就行更好的理解：

func doSomething(){
    panic("A Panic Running Error")
}

func clearTransaction(tx *sql.Tx){
    err := tx.Rollback()
    if err != sql.ErrTxDone && err != nil{
        log.Fatalln(err)
    }
}


func main() {
    db, err := sql.Open("mysql", "root:@tcp(127.0.0.1:3306)/test?parseTime=true")
    if err != nil {
        log.Fatalln(err)
    }

    defer db.Close()

    tx, err := db.Begin()
    if err != nil {
        log.Fatalln(err)
    }
    defer clearTransaction(tx)

    rs, err := tx.Exec("UPDATE user SET gold=50 WHERE real_name=\'vanyarpy\'")
    if err != nil {
        log.Fatalln(err)
    }
    rowAffected, err := rs.RowsAffected()
    if err != nil {
        log.Fatalln(err)
    }
    fmt.Println(rowAffected)

    rs, err = tx.Exec("UPDATE user SET gold=150 WHERE real_name=\'noldorpy\'")
    if err != nil {
        log.Fatalln(err)
    }
    rowAffected, err = rs.RowsAffected()
    if err != nil {
        log.Fatalln(err)
    }
    fmt.Println(rowAffected)

    doSomething()

    if err := tx.Commit(); err != nil {
        // tx.Rollback() 此时处理错误，会忽略doSomthing的异常
        log.Fatalln(err)
    }

}

这里定义了一个clearTransaction(tx)函数，该函数会执行rollback操作。因为我们事务处理过程中，任何一个错误都会导致main函数退出，因此在main函数退出执行defer的rollback操作，回滚事务和释放连接。

如果不添加defer，只在最后Commit后check错误err后再rollback，那么当doSomething发生异常的时候，函数就退出了，此时还没有执行到tx.Commit。这样就导致事务的连接没有关闭，事务也没有回滚。

tx事务环境中，只有一个数据库连接，事务内的Eexc都是依次执行的，事务中也可以使用db进行查询，但是db查询的过程会新建连接，这个连接的操作不属于该事务。