在 SQL Server 2008 R2 的 MERGE 语句中更新插入的记录

Posted 2023-03-29

【中文标题】在 SQL Server 2008 R2 的 MERGE 语句中更新插入的记录

【英文标题】：Updating inserted record within MERGE statement in SQL Server 2008 R2

【发布时间】：2012-12-04 06:37:11

【问题描述】：

我的 SQL Server 2008 R2 存储过程中有以下代码。在该存储过程中，我将一个城市及其家人和人员复制到另一个城市。

这里我在@FamilyIdMap 中维护家庭的源和目标ID。

左栏表示代码行号。

-- Copy Person
1>      DECLARE @PersonIdMap table (TargetId int, SourceId int)
2>      MERGE Person as PersonTargetTable
3>      USING (SELECT PersonID, FamilyID, PersonName, ParentID FROM Person
4>      WHERE FamilyID in (SELECT FamilyID from Family where FamilyName like '%DA%'))
5>      AS PersonSourceTable ON (0=1)
6>      WHEN NOT MATCHED THEN
7>      INSERT(FamilyID, PersonName, ParentID)
8>      VALUES
9>      ((SELECT TOP 1 TargetID from @FamilyIdMap WHERE SourceID=FamilyID),PersonName, 
10>     ParentID) OUTPUT
11>     INSERTED.PersonID, PersonSourceTable.PersonID
12>     INTO @PersonIdMap;

它给出如下输出：

源表

PersonID    FamilyID    PersonName  ParentID
1           1           ABC         Null
2           1           Son of ABC  1
3           1           Son of ABC  1
4           2           XYZ         NULL
5           2           Son of XYZ  4

目标表（使用上面给出的代码从源表复制）

PersonID    FamilyID    PersonName  ParentID
6           1           ABC         Null
7           1           Son of ABC  1 <-- ParentID Remains as it is
8           1           Son of ABC  1 <--
9           2           XYZ         NULL
10          2           Son of XYZ  4 <--

上面输出的问题是它没有更新parentID，我希望输出是这样的：

预期目标表

PersonID    FamilyID    PersonName  ParentID
6           1           ABC         Null
7           1           Son of ABC  6 <-- ParentID should be updated
8           1           Son of ABC  6 <--
9           2           XYZ         NULL
10          2           Son of XYZ  9 <--

我知道问题出在代码的第 10 行

10>     ParentID) OUTPUT

但是我应该用ParentID 替换什么来更新它？提前致谢。

【参考方案1】：

您尝试执行的操作在 SQL Server 2008R2 中无法一步完成。 更新 ParentId 必须是第二步，因为您无法访问插入另一行结果的一行中的 OUTPUT 值。但是，您已经在为第二步收集信息。因此，您只需要添加一个简单的更新即可。

IF OBJECT_ID('dbo.Person') IS NOT NULL DROP TABLE dbo.Person;
IF OBJECT_ID('dbo.Family') IS NOT NULL DROP TABLE dbo.Family;

CREATE TABLE dbo.Family(FamilyID INT IDENTITY(1,1) PRIMARY KEY CLUSTERED, FamilyName NVARCHAR(60));
CREATE TABLE dbo.Person(PersonID INT IDENTITY(1,1) PRIMARY KEY CLUSTERED, FamilyID INT REFERENCES dbo.Family(FamilyID), PersonName NVARCHAR(60), ParentID INT);
INSERT INTO dbo.Family(FamilyName) VALUES
('DA1'),
('DA2');

INSERT INTO dbo.Person(FamilyID, PersonName, ParentID) VALUES
(1, 'ABC', NULL),
(1, 'Son of ABC', 1),
(1, 'Son of ABC', 1),
(2, 'XYZ', NULL),
(2, 'Son of XYZ', 4 );

DECLARE @FamilyIdMap table (TargetId int, SourceId int)
MERGE dbo.Family tf
USING (SELECT * FROM dbo.Family WHERE FamilyName like '%DA%') AS sf
ON 1=0
WHEN NOT MATCHED THEN
INSERT (FamilyName)
VALUES(sf.FamilyName)
OUTPUT INSERTED.FamilyID, sf.FamilyID
INTO @FamilyIdMap;

DECLARE @PersonIdMap table (TargetId int, SourceId int)

MERGE dbo.Person as tp
USING (SELECT p.PersonID, p.FamilyID, p.PersonName, p.ParentID, fm.SourceId,fm.TargetId FROM Person AS p
INNER JOIN @FamilyIdMap AS fm 
ON p.FamilyID = fm.SourceId) AS sp
ON (0=1)
WHEN NOT MATCHED THEN
INSERT(FamilyID, PersonName, ParentID)
VALUES
(sp.TargetId,PersonName, ParentID) OUTPUT
INSERTED.PersonID, sp.PersonID
INTO @PersonIdMap;

UPDATE p SET
  ParentID = pm.TargetId
FROM dbo.Person AS p
JOIN @PersonIdMap pm
ON pm.SourceId = p.ParentID
WHERE EXISTS(SELECT 1 FROM @PersonIdMap pmf WHERE pmf.TargetId = p.PersonID);

SELECT * FROM dbo.Family;
SELECT * FROM @FamilyIdMap;
SELECT * FROM dbo.Person;
SELECT * FROM @PersonIdMap;

我确实添加了代码来创建和填充@FamilyIdMap 表。我还稍微清理了你原来的 MERGE。它现在使用 @FamilyIdMap 表作为选择行的一种方式，而不是再次加入 dbo.Family 表。如果您只在一小部分系列上运行它，这应该会更快。如果您有很多家庭并且您将它们全部复制，那么再次针对 dbo.Family 表可能会更快。

最终的 UPDATE 仅更新 Person 表中的新行（所有新创建的 PersonId 都可以在 @PersonIdMap 表的 TargetId 列中找到），使用 @PersonIdMap 表中的信息将旧的 ParentId 值更改为新的 ParentId 值。

我没有包括事务管理，但至少 MERGE dbo.Person 和以下 UPDATE dbo.Person 应该在同一个事务中执行。

【讨论】：

它按预期工作，非常感谢您的时间和努力:)