如何在 Golang 中使用事务？-Golang-PHP中文网

如何在 Golang 中使用事务？

王林

发布： 2024-05-14 14:12:01

原创

1171人浏览过

在 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)
}

登录后复制

在事务块中，您可以执行任何数据库操作。例如，以下代码从表中选择一条记录并将其更新为新值：

立即学习“go语言免费学习笔记（深入）”；

易森网络企业版

如果您是新用户，请直接将本程序的所有文件上传在任一文件夹下，Rewrite 目录下放置了伪静态规则和筛选器，可将规则添加进IIS，即可正常使用，不用进行任何设置；(可修改图片等)默认的管理员用户名、密码和验证码都是：yeesen系统默认关闭，请上传后登陆后台点击“核心管理”里操作如下:进入“配置管理”中的&ld

查看详情

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 表。我们希望创建一个事务来执行以下操作：

检查产品是否有足够的库存。
如果有库存，则扣除产品的库存并创建订单。

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中文网其它相关文章！