无游标循环和更新

Question

我有一个存储项目余额的表。

CREATE TABLE itembalance (
   ItemID VARCHAR(15),
   RemainingQty INT,
   Cost Money,
   Id INT
)

我需要确保每次发出物品时，都会从 itembalance 表中扣除适当的余额。我是这样做的：

DECLARE crsr CURSOR LOCAL FAST_FORWARD FOR 
        SELECT 
           itembalance.Cost, 
           itembalance.RemainingQty
           itembalance.Id
        FROM dbo.itembalance
        WHERE itembalance.ItemID = @v_item_to_be_updated AND RemainingQty > 0

OPEN crsr
FETCH crsr
INTO 
  @cost, 
  @qty, 
  @id

WHILE @@FETCH_STATUS = 0

BEGIN
IF @qty >= @qty_to_be_deducted
BEGIN
    UPDATE itembalance SET RemainingQty = RemainingQty - @qty_to_be_deducted WHERE Id = @id

    /*do something with cost*/ BREAK
END
ELSE
BEGIN
    UPDATE itembalance SET RemainingQty = 0 WHERE Id = @id
    /*do something with cost*/ SET @qty_to_be_deducted = @qty_to_be_deducted - @qty
END
FETCH crsr
INTO 
  @cost, 
  @qty, 
  @id


END
CLOSE crsr
DEALLOCATE crsr

该表可能包含相同的项目代码，但成本不同。此代码适用于一次更新的少量项目，但每当发送大量项目/数量时，该过程变得非常缓慢。有没有办法优化这段代码？我猜光标让它变慢了，所以我想为这个过程探索不同的代码。

Answer 1

这看起来你只需要一个简单的CASE 表达式：

UPDATE dbo.itembalance
SET Qty = CASE WHEN Qty >= @qty_to_be_deducted THEN Qty - @qty_to_be_deducted ELSE 0 END
WHERE ItemID = @v_item_to_be_updated
  --What is the difference between Qty and RemainingQty?
  --Why are you checking one and updating the other?
  AND RemainingQty > 0; 

Answer 2

您的代码不太清楚该机制是如何以及为什么需要和工作的。

但是，假设您必须有多个未结余额的记录，并且您必须按顺序考虑多个记录作为此机制的一部分，那么您有两个选项可以在 SQL 中解决这个问题（在客户端代码中处理是另一个选项）：

1) 像以前一样使用光标

2) 使用临时表或表变量并对其进行迭代 - 非常类似于游标，但可能更快 - 您必须尝试查看例如

declare @TableVariable table (Cost money, RemainingQty int, Id int, OrderBy int, Done bit default(0))
declare @Id int, @Cost money, @RemainingQty int

insert into @TableVariable (Cost, RemainingQty, Id, OrderBy)
  SELECT 
    itembalance.Cost 
    , itembalance.RemainingQty
    , itembalance.Id
    , 1 /* Some order by condition */
  FROM dbo.itembalance
  WHERE itembalance.ItemID = @v_item_to_be_updated AND RemainingQty > 0

while exists (select 1 from @TableVariable where Done = 0) begin
  select top 1 @Id = id, @Cost = Cost, @RemainingQty
  from @TableVariable
  where Done = 0
  order by OrderBy

  -- Do stuff here

  update @TableVariable set Done = 1 where id = @Id
end

但是，您显示的代码似乎并不应该很慢 - 因此可能是您缺少适当的索引，并且单个 ItemId 更新锁定了 ItemBalance 表中的太多行，这会影响其他 ItemId 更新。