原子独占SQL记录更新

Question

我想更新表中的单个记录,以反映给定的客户端会话已在多会话环境中获取记录(现在拥有该记录以进行进一步更新).到目前为止我有这个:

create procedure AcquireRow(
  @itemNo   int,          -- Item ID to acquire
  @sessNo   int,          -- Session ID
  @res      char(1) out)  -- Result
as
begin
  -- Attempt to acquire the row
  update Items
    set
      State = 'A',      -- 'A'=Acquired
      SessionID = @sessNo
    where ItemID = @itemNo
      and State = 'N';  -- 'N'=Not acquired  

  -- Verify that the session actually acquired the row
  set @res = 'T';       -- 'T'=Success
  if @@rowcount = 0
    set @res = 'F';     -- 'F'=Failure
end;

如果过程成功获取了行,则将out变量@state设置为'T',否则将其设置'F'为指示失败.

我的问题:这是否保证以原子方式工作,以便只有一个会话成功获取(更新)行,如果多个会话同时调用AcquireRow()？或者有更好的方法吗？我需要明确rowlock吗？

修正:
根据Remus的回答,我会重新安排代码:

set @res = 'F';
update ...;
if @@rowcount > 0
    set @res = 'T';

使用一个output子句或将结果行分配ItemID给一个变量update也是谨慎的.

Answer 1

@@ROWCOUNT众所周知,容易出错.例如,在你的情况下.来自MSDN:

进行简单赋值的语句始终将@@ROWCOUNT 值设置为1.不向客户端发送任何行.这些陈述的例子是:SET @local_variable......

为了正确,你应该在之后立即检查@@ ROWCOUNT UPDATE.使用ROWLOCK更安全,但不是必需的.即使优化器决定使用页锁,'acquire'语义也是正确的.

我可能更喜欢另一种方法,即使用以下OUTPUT条款UPDATE:

declare @updated table (ItemId int);

update Items
set ...
output inserted.ItemId
into @updated (ItemId)
where ...

这种方案更加防错,也更灵活,因为它允许获取未知的ItemId:获取的id放在@updated表变量中,并可以返回给调用者.

作为一般说明,使用真实的,已提交的"获取"更新时会遇到问题,因为您无法知道哪些行真正被获取以及哪些行被放弃(客户端断开连接或崩溃而不释放"获取").