TSQL 实现双重检查锁定

Question

我有一个任意的存储过程usp_DoubleCheckLockInsert，它为多个客户端执行INSERT，当它位于临界区Begin lock 和End lock.

CREATE PROCEDURE usp_DoubleCheckLockInsert
     @Id INT
    ,@SomeValue INT
AS
BEGIN
    IF (EXISTS(SELECT 1 FROM SomeTable WHERE Id = @Id AND SomeValue = @SomeValue)) RETURN
    BEGIN TRAN
        --Begin lock
        IF (EXISTS(SELECT 1 FROM SomeTable WHERE Id = @Id AND SomeValue = @SomeValue)) ROLLBACK

        INSERT INTO SomeTable(Id, SomeValue)
        VALUES(@Id,@SomeValue);
        --End lock
    COMMIT
END

我已经看到隔离级别与更新的关系，但是有没有办法在关键部分实现锁定，给事务写锁，或者 TSQL 不是这样工作的？

Obtain Update Table Lock at start of Stored Procedure in SQL Server

Answer 1

对我有用的第二种方法是将INSERT 和SELECT 组合成一个操作。

这个索引只需要高效查询SomeTable。请注意，没有唯一性约束。但是，如果我采用这种方法，我实际上会使索引唯一。

CREATE INDEX [IX_SomeTable_Id_SomeValue_IsDelete] ON [dbo].[SomeTable]
(
    [Id] ASC,
    [SomeValue] ASC,
    [IsDelete] ASC
)

存储过程，结合了 INSERT/SELECT 操作：

CREATE PROCEDURE [dbo].[usp_DoubleCheckLockInsert]
     @Id INT
    ,@SomeValue INT
    ,@IsDelete bit
AS
BEGIN

        -- Don't allow dirty reads

        SET TRANSACTION ISOLATION LEVEL READ COMMITTED

    BEGIN TRAN
        -- insert only if data not existing
        INSERT INTO dbo.SomeTable(Id, SomeValue, IsDelete)
        SELECT @Id, @SomeValue, @IsDelete
        where not exists (
            select * from dbo.SomeTable WITH (HOLDLOCK, UPDLOCK)
            where Id = @Id
            and SomeValue = @SomeValue
            and IsDelete = @IsDelete)

    COMMIT
END

我确实尝试过使用多个进程插入数据的这种方法。 （尽管我承认我并没有对 SQL Server 施加太大压力）。从来没有任何重复或失败的插入。

Answer 2

看来您要做的只是防止插入重复的行。你可以通过添加一个唯一索引来做到这一点，选项IGNORE_DUP_KEY = ON：

CREATE UNIQUE INDEX [IX_SomeTable_Id_SomeValue_IsDelete]
ON [dbo].[SomeTable]
(
    [Id] ASC,
    [SomeValue] ASC,
    [IsDelete] ASC
) WITH (IGNORE_DUP_KEY = ON)

SQL Server 将忽略任何具有重复键的插入。运行以下命令：

INSERT INTO [dbo].[SomeTable] ([Id],[SomeValue],[IsDelete])
VALUES(0,0,0)

INSERT INTO [dbo].[SomeTable] ([Id],[SomeValue],[IsDelete])
VALUES(1,1,0)

INSERT INTO [dbo].[SomeTable] ([Id],[SomeValue],[IsDelete])
VALUES(2,2,0)

INSERT INTO [dbo].[SomeTable] ([Id],[SomeValue],[IsDelete])
VALUES(0,0,0)

结果：

(1 row(s) affected)

(1 row(s) affected)

(1 row(s) affected)

Duplicate key was ignored.
(0 row(s) affected)

我没有使用多个进程（线程）测试上述内容，但这种情况下的结果应该是相同的 - SQL Server 仍然应该忽略任何重复项，无论哪个线程正在尝试插入。

另请参阅 MSDN 上的 Index Options。

Answer 3

我想我可能不明白这个问题，但你为什么不能这样做：

begin tran
if ( not exists ( select 1 from SomeTable where Id = @ID and SomeValue = @SomeValue ) )
    insert into SomeTable ( Id, SomeValue ) values ( @ID, @SomeValue )
commit

是的，您每次执行此操作时都会进行交易，但只要您的速度很快，这应该不是问题。

我感觉我没有理解这个问题。

杰夫。

Answer 4

一旦你开始搞乱 sql 首选锁定管理，你就会承担责任，但如果你确定这是你需要的，请更新你的 sp 以选择一个测试变量并将你的“EXISTS”检查替换为那个变量。当您查询变量时，请使用排他表锁，在您完成之前，该表是您的。

CREATE PROCEDURE usp_DoubleCheckLockInsert
     @Id INT
    ,@SomeValue INT
AS
BEGIN
     IF (EXISTS(SELECT 1 FROM SomeTable WHERE Id = @Id AND SomeValue = @SomeValue)) RETURN
     BEGIN TRAN
        --Begin lock
         DECLARE @tId as INT
         -- You already checked and the record doesn't exist, so lock the table
         SELECT @tId 
         FROM SomeTable WITH (TABLOCKX)
         WHERE Id = @Id AND SomeValue = @SomeValue
         IF @tID IS NULL
           BEGIN
             -- no one snuck in between first and second checks, so commit
           INSERT INTO SomeTable(Id, SomeValue)
           VALUES(@Id,@SomeValue);
        --End lock
     COMMIT
END

如果您将其作为查询执行，但未命中提交，然后尝试从不同上下文的表中进行选择，您将坐下来等待提交生效。

Answer 5

Romoku，你得到的答案基本上是正确的，除了

• 你甚至不需要BEGIN TRAN。
• 您无需担心隔离级别。

您只需要一个简单的insert ... select ... where not exists (select ...)，正如 Jeff B 和 Chue X 所建议的那样。

您对并发性的担忧（“我说的是并发性，而您的回答不起作用。”）揭示了对 SQL 工作原理的深刻误解。

SQL INSERT 是原子的。您不必锁定桌子；这就是 DBMS 为您所做的。

我建议不要为基于错误的先入为主的观念而提出的错误问题提供奖励——然后将正确的答案视为错误的答案——我建议坐下来阅读一本好书。在 SQL 上。如果你喜欢，我可以推荐一些标题。