golang db事務的統一封裝

事務處理的流程示例

    database := db.DB
    tx, err := database.Begin()
    if err != nil {
        return err
    }
    stmt, err := tx.Prepare(sqlQuery)
    if err != nil {
        tx.Rollback()
        return err
    }
    _, err = stmt.Exec(paras...)
    if err != nil {
        tx.Rollback()
        return err
    }
    err = tx.Commit()
    if err != nil {
        tx.Rollback()
        return err
    }

以上是我們使用事務時的一般操作，如果每做一次事務的操作均要進行重新寫一遍代碼豈不是很麻煩，尤其是出錯時，Rollback需要多次在不同錯誤的地方的進行調用處理。

偷懶第一步:

採用defer處理Rollback

defer tx.Rollback()

無論成功與否，均進行Rollback操作，只是有點影響，如果成功還調用Rollback的話，將會報錯。雖然可以忽略，但作爲程序員，有必要進一步調整。

偷懶第二步：

根據執行結果來選擇執行Rollback，避免無效使用。

defer func() { //根據執行結果選擇執行Rollback
        log.Println("ExecSqlWithTransaction defer err :", err)
        if err != nil && tx != nil {
            tx.Rollback()
        }
    }()

如此，我們就可以根據事務的執行結果決定是否Rollback了。

偷懶第三步：

封裝，以上代碼本身就具有極大的普適性，因此，我們抽出通用的參數，將此過程封裝成一個func，以後就可以直接調用了。

func ExecSqlWithTransaction(database *sql.DB, query string, args ...interface{}) (err error) {
    tx, err := database.Begin()
    if err != nil {
        return err
    }
    defer func() {
        if err != nil && tx != nil {
            tx.Rollback()
        }
    }()
    stmt, err := tx.Prepare(query)
    if err != nil {
        return err
    }
    defer stmt.Close()
    _, err = stmt.Exec(args...)
    if err != nil {
        return err
    }
    if err = tx.Commit(); err != nil {
        return err
    }
    return nil
}

封裝後我們可以如下使用：

if err := ExecSqlWithTransaction(database,sqlQuery,paras...);err != nil{
    //錯誤處理
}

封裝後是不是很簡潔啊？