如何在 Golang 中使用事務？

在 Go 中，可以使用 Tx 類型進行交易操作。若要開啟事務，請使用 db.Begin()。在事務區塊中，執行資料庫操作，如查詢、更新。執行成功後，使用 tx.Commit() 提交交易。在實戰中，事務可確保並發操作的一致性，如同時更新庫存和建立訂單。

如何在 Go 中使用交易？

在 Go 中，事務是一組原子操作的集合，要麼全部成功，要麼全部失敗。它們用於確保資料庫操作的完整性，尤其是在多個並發操作可能導致資料損壞的情況下。

Go 中的交易由 database/sql 中的 Tx 類型表示。若要開始一個事務，請執行下列操作：

tx, err := db.Begin()
if err != nil {
    return fmt.Errorf("failed to begin transaction: %w", err)
}

在事務區塊中，您可以執行任何資料庫操作。例如，以下程式碼從表中選擇一筆記錄並將其更新為新值：

var id, version int
if err := tx.QueryRow("SELECT id, version FROM example WHERE name = ?", "foo").Scan(&id, &version); err != nil {
    return fmt.Errorf("failed to select: %w", err)
}

result, err := tx.Exec("UPDATE example SET version = ? WHERE id = ? AND version = ?", version+1, id, version)
if err != nil {
    return fmt.Errorf("failed to update: %w", err)
}

在所有操作成功執行後，您可以使用tx.Commit() 提交交易：

if err := tx.Commit(); err != nil {
    // 放弃事务
    if err := tx.Rollback(); err != nil {
        return fmt.Errorf("failed to rollback transaction: %w", err)
    }
    return fmt.Errorf("failed to commit transaction: %w", err)
}

實戰案例

考慮一個網路商店的資料庫，該資料庫包含一個products 表和一個orders 表。我們希望建立一個事務來執行以下操作：

1. 檢查產品是否有足夠的庫存。
2. 如果有庫存，則扣除產品的庫存並建立訂單。

func CreateOrder(db *sql.DB, productID int, quantity int) error {
    tx, err := db.Begin()
    if err != nil {
        return fmt.Errorf("failed to begin transaction: %w", err)
    }

    var stock int
    if err := tx.QueryRow("SELECT stock FROM products WHERE id = ?", productID).Scan(&stock); err != nil {
        return fmt.Errorf("failed to select: %w", err)
    }

    if stock < quantity {
        return tx.Rollback() // 放弃事务，因为没有足够的库存
    }

    if _, err := tx.Exec("UPDATE products SET stock = ? WHERE id = ?", stock-quantity, productID); err != nil {
        return fmt.Errorf("failed to update: %w", err)
    }

    if _, err := tx.Exec("INSERT INTO orders (product_id, quantity) VALUES (?, ?)", productID, quantity); err != nil {
        return fmt.Errorf("failed to insert: %w", err)
    }

    if err := tx.Commit(); err != nil {
        return fmt.Errorf("failed to commit transaction: %w", err)
    }

    return nil
}

透過使用事務，我們確保了商店的庫存始終準確，即使有多個並發訂單嘗試從同一個產品減去庫存。

以上是如何在 Golang 中使用事務？的詳細內容。更多資訊請關注PHP中文網其他相關文章！