首頁 > 後端開發 > Golang > 如何在 Golang 中使用事務?

如何在 Golang 中使用事務?

WBOY
發布: 2024-06-04 12:13:56
原創
876 人瀏覽過

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

如何在 Golang 中使用事务?

如何在 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中文網其他相關文章!

相關標籤:
來源:php.cn
本網站聲明
本文內容由網友自願投稿,版權歸原作者所有。本站不承擔相應的法律責任。如發現涉嫌抄襲或侵權的內容,請聯絡admin@php.cn
熱門教學
更多>
最新下載
更多>
網站特效
網站源碼
網站素材
前端模板