SQL Merge输出匹配和不匹配的结果

Posted 2021-04-05

我有两个数据键，pk应该是在插入行时从数据库生成的，而fk是由另一个系统给出的复合键。我想为每个fk密钥生成一个pk密钥。

CREATE TABLE test_target (
[pk] [INT] IDENTITY(1,1),
[fk] [varchar](20) NOT NULL)

并且我可以使用merge来确保在表中没有相应的fk时生成新的pk，并且我知道我可以输出新创建的id。

CREATE TABLE test_source (
[fk] [varchar](20) NOT NULL)

INSERT INTO test_source VALUES('abc123'),('def456'),('ghi789')

MERGE test_target WITH (SERIALIZABLE) AS T
USING test_source AS U
ON U.fk = T.fk
WHEN NOT MATCHED THEN
INSERT (fk) VALUES(U.fk)
OUTPUT inserted.pk, inserted.fk;

但是，我真正想要的是与test_source表中的fk相关的所有pk。所以我可以通过加入两个表来获得所有。

SELECT test_target.* FROM test_target
INNER JOIN test_source ON test_target.fk = test_source.fk

但我觉得在合并声明中的MATCHED情况下已经找到了关联pk，因此重复工作是在目标表上进行另一次搜索。我的问题是有没有办法在同一个合并语句中输出MATCHED pk？

答案

是的 - 起初我以为我必须触摸行并以某种形式更新它，但我意识到我们可以欺骗它。输出子句将输出语句触及的任何行，而不仅仅是您不匹配的行，因此您可以包含when匹配的子句 - 问题是使其为空操作。

create table foo
(
    id int
    ,bar varchar(30)
)

insert into foo (id, bar) values (1,'test1');
insert into foo (id, bar) values (2,'test2');
insert into foo (id, bar) values (3,'test3');

declare @temp int;
merge foo as dest
using
(
    values (2, 'an updated value')
        , (4, 'a new value')
) as src(id, bar) on (dest.id = src.id)
when matched then
update set @temp=1
when not matched then
insert (id,bar)
values (src.id, src.bar)
output $action, src.id;

你可以在when匹配子句中看到，我将声明的变量设置为1.奇怪的是，输出子句可以考虑将其输出为输出。如果需要在输出中使用$ action，则可以区分发生了哪个操作（插入与更新）。

这给出了以下结果：

$action     id
UPDATE  2
INSERT  4

性能方面我想测试如何大规模操作，或者变量赋值是否会导致节流效果