使用锁来控制数据库并发

想象一下您正在开发一个电子商务系统，成千上万的人试图同时购买最后剩下的产品。然而，他们中的许多人可以继续结账并完成订单。当您检查库存时，您的产品数量为负数。这是怎么可能的，你该如何解决这个问题？

让我们编码吧！您可能想到的第一件事是在结帐前检查库存。也许是这样的：

public void validateAndDecreaseSolution(long ProductId, int 数量 {
可选 stockByProductId =
stockRepository.findStockByProductId(productId);

int stock = stockByProductId.orElseThrow().getStock();
int possibleStock = 库存 - 数量；

if (股票 

您可以使用此验证，但是当我们谈论每秒数百、数千、数百万甚至数十个请求时，此验证是不够的。当 10 个请求同时到达这段代码并且数据库为 stockByProductId 返回相同的值时，您的代码将崩溃。在我们进行验证时，您需要一种方法来阻止其他请求。

第一个解决方案 - 用于更新

在 SELECT 上添加锁定语句。在此示例中，我使用 Spring Data 的 FOR UPDATE 来完成此操作。正如 PostgreSQL 文档所说

FOR UPDATE 会导致 SELECT 语句检索到的行被锁定，就像要进行更新一样。这可以防止它们被其他交易修改或删除，直到当前交易结束。

@Query(value = "从股票 s 中选择 * WHERE s.product_id = ?1 FOR UPDATE", nativeQuery = true)
可选 findStockByProductIdWithLock(Long ProductId);
登录后复制

public void validateAndDecreaseSolution1(long ProductId, int amount) {
    可选 stockByProductId = stockRepository.findStockByProductIdWithLock(productId);

    // ... 证实

    stockRepository.decreaseStock(productId, 数量);
}
登录后复制

所有使用产品ID对stocks表的请求都将等待，直到实际交易完成。这里的目标是确保您获得股票的最新更新价值。
立即学习“Java免费学习笔记（深入）”；
点击下载“C盘瘦身工具,一键清理C盘”；

  第二种解决方案 - pg_advisory_xact_lock

此解决方案与上一个类似，但您可以选择锁定键是什么。我们将锁定整个交易，直到完成所有验证和库存减量的处理。

public void acquireLockAndDecreaseSolution2(long ProductId, int amount) {
    查询nativeQuery =entityManager.createNativeQuery("选择pg_advisory_xact_lock(:lockId)");
    nativeQuery.setParameter("lockId",productId);
    nativeQuery.getSingleResult();

    可选 stockByProductId = stockRepository.findStockByProductId(productId);

    // 检查库存并在必要时抛出异常

    stockRepository.decreaseStock(productId, 数量);
}
登录后复制

本次交易结束后，下一次请求只会与同ID的产品进行交互。

  第三种解决方案 - WHERE 子句

在这种情况下，我们不会锁定行或事务。让我们允许此事务继续进行，直到更新语句为止。请注意最后一个条件：库存 > 0。这将不允许我们的库存小于零。因此，如果两个人尝试同时购买，其中一个人会收到错误，因为我们的数据库不允许库存 

@交易
@修改
@Query(nativeQuery = true, value = "更新库存 SET stock = stock - :quantity WHERE Product_id = :productId AND stock > 0")
int reduceStockWhereQuantityGreaterThanZero(@Param("productId") Long ProductId, @Param("quantity") int amount);
登录后复制

  结论

第一个和第二个解决方案使用悲观锁作为策略。第三是乐观锁。当您在执行涉及某个资源的任何任务时希望限制对该资源的访问时，可以使用悲观锁定策略。在您完成进程之前，目标资源将被锁定以进行任何其他访问。小心死锁！

使用乐观锁，您可以在没有任何阻塞的情况下对同一资源执行各种查询。当冲突不太可能发生时使用它。通常，您会有一个与您的行相关的版本，当您更新该行时，数据库会将您的行版本与数据库中的行版本进行比较。如果两者相等，则更改将成功。如果没有，您必须重试。正如您所看到的，我在本文中没有使用任何版本行，但我的第三个解决方案不会阻止任何请求并使用 stock > 0 条件控制并发。

如果想看完整代码，可以查看我的GitHub。

还有很多其他策略来实现悲观锁定和乐观锁定，例如您可以搜索更多有关 FOR UPDATE WITH SKIP LOCKED 的信息。

        以上就是如何使用 Java 和 PostgreSQL 处理竞争条件的详细内容，更多请关注php中文网其它相关文章！