如何使用 DELETE 强制 MERGE 语句使用索引查找？

Question

我已经为我的 Facebook 应用程序创建了 MS SQL 2014 数据库，可以与朋友一起使用。我在数据库中为我的所有用户保留朋友，并在应用程序启动时从 Facebook 更新他们。为此，我使用了 MERGE 语句（表变量 @FriendUserIds 包含朋友 ID 的列表；表 UserFriends 具有聚集的主键（UserId，FriendUserId））：

MERGE UserFriends
    USING (
        SELECT
                UserId
            FROM @FriendUserIds
    ) AS source (FriendUserId)
        ON UserFriends.UserId = @UserId
            AND UserFriends.FriendUserId = source.FriendUserId
    WHEN NOT MATCHED BY TARGET
        THEN INSERT (UserId, FriendUserId)
            VALUES (@UserId, source.FriendUserId)
    WHEN NOT MATCHED BY SOURCE
        AND UserFriends.UserId = @UserId
        THEN DELETE;

问题是查询优化器无法识别它可以在 UserFriends 上使用 INDEX SEEK。它使用 SCAN 代替，我不知道强制 SEEK 的方法。 现在，我通过将操作拆分为两个查询（用于添加新朋友的 MERGE 和用于删除不再朋友的 DELETE）来规避该问题，这仍然比单个 MERGE 语句（没有 DELETE 语句的 MERGE 使用 SEEK）快得多：

DELETE
    FROM UserFriends
    WHERE UserFriends.UserId = @UserId
        AND UserFriends.FriendUserId NOT IN (
            SELECT
                    UF.UserId
                FROM @FriendUserIds UF
        )

MERGE UserFriends
    USING (
        SELECT
                UserId
            FROM @FriendUserIds
    ) AS source (FriendUserId)
        ON UserFriends.UserId = @UserId
            AND UserFriends.FriendUserId = source.FriendUserId
    WHEN NOT MATCHED BY TARGET
        THEN INSERT (UserId, FriendUserId)
            VALUES (@UserId, source.FriendUserId);

Answer 1

尝试使用公用表表达式 (CTE) 作为您的“目标”：

;WITH UserFriends_CTE
     AS (SELECT [UserID],
                [FriendUserID]
         FROM   [UserFriends]
         WHERE  [UserID] = @UserId)
MERGE UserFriends_CTE
USING (SELECT [UserId]
       FROM   @FriendUserIds) AS source ([FriendUserId])
ON UserFriends_CTE.[UserId] = @UserId
   AND UserFriends_CTE.[FriendUserId] = source.[FriendUserId]
WHEN NOT MATCHED BY TARGET THEN
  INSERT ([UserId],
          [FriendUserId])
  VALUES (@UserId,
          source.[FriendUserId])
WHEN NOT MATCHED BY SOURCE THEN
  DELETE; 

MERGE 语句通常比拆分为多个语句和there are a few known problems with MERGE 执行得更差。使用 CTE 可能会导致问题according to Paul White in this answer，所以测试一下。

如果您确实使用了拆分版本，我将按照以下方式实现它：

DELETE uf
FROM   [UserFriends] uf
WHERE  uf.[UserId] = @UserId
       AND NOT EXISTS
               (SELECT 1
                FROM   @FriendUserIds fu
                WHERE  uf.[FriendUserId] = fu.[FriendUserId]);

INSERT INTO [UserFriends]
            ([UserId],
             [FriendUserId])
SELECT @UserId,
       fu.[FriendUserId]
FROM   @FriendUserIds fu
WHERE  NOT EXISTS
           (SELECT 1
            FROM   [UserFriends] uf
            WHERE  fu.[FriendUserId] = uf.[FriendUserId]
                   AND uf.[UserId] = @UserId);

Answer 2

第一个明显的变体是使用两个显式语句：DELETE 和 INSERT。您永远不会更新现有行，因此您可以使用传统的 INSERT 而不是 MERGE。

DELETE FROM UserFriends
WHERE 
    UserFriends.UserId = @UserId
    AND UserFriends.FriendUserId NOT IN 
    (
        SELECT UF.UserId
        FROM @FriendUserIds AS UF
    )
;

INSERT INTO UserFriends(UserId, FriendUserId)
SELECT @UserId, UF.UserId
FROM @FriendUserIds AS UF
WHERE
    UF.UserId NOT IN
    (
        SELECT UserFriends.FriendUserId
        FROM UserFriends
        WHERE UserFriends.UserId = @UserId
    )
;

使用适当的错误处理将其包装在事务和TRY ... CATCH 中。

---

第二种变体是尽量保持单个MERGE，但要确保表变量具有主键/聚集唯一索引。它可能有助于优化器。

表类型的定义如下所示：

CREATE TYPE [dbo].[UserIdsTableType] AS TABLE
(
    [UserId] [int] NOT NULL,
    PRIMARY KEY CLUSTERED 
(
    [UserId] ASC
))

---

第三种变体是再次使用带有主键/聚集唯一索引的#temp 表而不是表变量。它可能会进一步帮助优化器，因为表变量的基数估计与普通表或临时表不同。它通常为 1，即优化器不知道表变量中有多少行，并假定它始终为 1 行。对于临时表，它应该知道行数。

---

事实上，即使您使用两个显式的DELETE 和INSERT 语句而不是单个MERGE，第三种变体也是有意义的。

看看使用临时表的MERGE 的实际执行计划与使用临时表的两个单独语句的实际计划会很有趣。理论上单个MERGE 可能更快，因为它可能只需要连接两个表一次。