执行事务

您可以使用 sql.Tx 来执行数据库事务,它代表一个事务。除了表示事务特定语义的 CommitRollback 方法外,sql.Tx 还包含所有用于执行常见数据库操作的方法。要获取 sql.Tx,您可以调用 DB.BeginDB.BeginTx

数据库事务 将多个操作分组作为更大目标的一部分。所有操作必须成功,否则都不能成功,并且在这两种情况下都必须保持数据完整性。通常,事务工作流程包括:

  1. 开始事务。
  2. 执行一组数据库操作。
  3. 如果没有发生错误,则提交事务以进行数据库更改。
  4. 如果发生错误,则回滚事务以使数据库保持不变。

sql 包提供了开始和结束事务的方法,以及执行中间数据库操作的方法。这些方法对应于上述工作流程中的四个步骤。

  • 开始事务。

    DB.BeginDB.BeginTx 开始一个新的数据库事务,并返回一个表示该事务的 sql.Tx

  • 执行数据库操作。

    使用 sql.Tx,您可以通过一系列使用单个连接的操作来查询或更新数据库。为此,Tx 导出了以下方法:

  • 以下列方法中的**一种**结束事务:

    • 使用 Tx.Commit 提交事务。

      如果 Commit 成功(返回 nil 错误),则所有查询结果都将确认为有效,并且所有执行的更新都将作为单个原子更改应用到数据库。如果 Commit 失败,则 Tx 上的 QueryExec 的所有结果都应作为无效结果丢弃。

    • 使用 Tx.Rollback 回滚事务。

      即使 Tx.Rollback 失败,事务也将不再有效,也不会提交到数据库。

最佳实践

遵循以下最佳实践,以更好地驾驭事务有时需要的复杂语义和连接管理。

  • 使用本节中描述的 API 来管理事务。**不要**直接使用与事务相关的 SQL 语句,例如 BEGINCOMMIT——这样做可能会使您的数据库处于不可预测的状态,尤其是在并发程序中。
  • 使用事务时,请注意不要直接调用非事务性的 sql.DB 方法,因为这些方法将在事务外部执行,从而导致您的代码对数据库状态的视图不一致,甚至可能导致死锁。

示例

以下示例中的代码使用事务为专辑创建新的客户订单。在此过程中,代码将:

  1. 开始事务。
  2. 延迟事务的回滚。如果事务成功,它将在函数退出之前提交,使延迟的回滚调用成为空操作。如果事务失败,它将不会提交,这意味着在函数退出时将调用回滚。
  3. 确认客户订购的专辑库存充足。
  4. 如果库存充足,则更新库存数量,将其减少订购的专辑数量。
  5. 创建新订单并为客户端检索新订单的生成 ID。
  6. 提交事务并返回 ID。

此示例使用接受 context.Context 参数的 Tx 方法。这使得函数的执行(包括数据库操作)可以在运行时间过长或客户端连接关闭时被取消。更多信息,请参阅取消进行中的操作

// CreateOrder creates an order for an album and returns the new order ID.
func CreateOrder(ctx context.Context, albumID, quantity, custID int) (orderID int64, err error) {

    // Create a helper function for preparing failure results.
    fail := func(err error) (int64, error) {
        return 0, fmt.Errorf("CreateOrder: %v", err)
    }

    // Get a Tx for making transaction requests.
    tx, err := db.BeginTx(ctx, nil)
    if err != nil {
        return fail(err)
    }
    // Defer a rollback in case anything fails.
    defer tx.Rollback()

    // Confirm that album inventory is enough for the order.
    var enough bool
    if err = tx.QueryRowContext(ctx, "SELECT (quantity >= ?) from album where id = ?",
        quantity, albumID).Scan(&enough); err != nil {
        if err == sql.ErrNoRows {
            return fail(fmt.Errorf("no such album"))
        }
        return fail(err)
    }
    if !enough {
        return fail(fmt.Errorf("not enough inventory"))
    }

    // Update the album inventory to remove the quantity in the order.
    _, err = tx.ExecContext(ctx, "UPDATE album SET quantity = quantity - ? WHERE id = ?",
        quantity, albumID)
    if err != nil {
        return fail(err)
    }

    // Create a new row in the album_order table.
    result, err := tx.ExecContext(ctx, "INSERT INTO album_order (album_id, cust_id, quantity, date) VALUES (?, ?, ?, ?)",
        albumID, custID, quantity, time.Now())
    if err != nil {
        return fail(err)
    }
    // Get the ID of the order item just created.
    orderID, err = result.LastInsertId()
    if err != nil {
        return fail(err)
    }

    // Commit the transaction.
    if err = tx.Commit(); err != nil {
        return fail(err)
    }

    // Return the order ID.
    return orderID, nil
}