原子独占SQL记录更新

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

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()？或者有更好的方法吗？我需要一个明确的行锁吗？

修改为：
根据Remus的回答,我会重新安排代码：

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

使用输出子句或将结果行的ItemID分配给更新中的变量也是谨慎的.

解决方法

@@ ROWCOUNT很容易出错.例如,在你的情况下.从 MSDN开始：

Statements that make a simple
assignment always set the @@ROWCOUNT
value to 1. No rows are sent to the
client. Examples of these statements
are: SET @local_variable …

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

我可能更喜欢另一种方法,即使用UPDATE的OUTPUT子句：

declare @updated table (ItemId int);

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

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

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

原子独占SQL记录更新

解决方法

相关文章