从 INSTEAD OF TRIGGER 获取 ROWCOUNT

Question

旧版应用使用代替触发器对表执行 INSERT，然后使用行数进行进一步处理。

我们现在需要使用INSTEAD OF INSERT 触发器来选择退出某些 INSERT。

问题是@@ROWCOUNT 仍然返回尝试插入的数量。

例如，一个永远不会完成插入的虚构触发器可能是

ALTER TRIGGER [dbo].[trig_ACCOUNT_CREDITS_RunningTotalINSERT]
   ON  [dbo].[ACCOUNT_CREDITS] 
   INSTEAD OF INSERT
AS 
BEGIN
 --tried with NOCOUNT ON and OFF
 SET NOCOUNT OFF;

 --This is an example of the branching logic that might determine 
 --whether or not to do the INSERT
IF 1=2 --no insert will ever occur (example only)
BEGIN
    INSERT INTO dbo.ACCOUNT_CREDITS (COL1, COL2)
    SELECT COL1, COL2 from INSERTED
END
END

有些 INSERT 语句可能是

--No rows will be inserted because value of COL1 < 5
INSERT INTO dbo.ACCOUNT_CREDITS (COL1, COL2) VALUES ( 3, 3)

--We would assume row count to be 0, but returns 1
select @@ROWCOUNT

--No rows will be inserted because value of COL1 < 5       
INSERT INTO dbo.ACCOUNT_CREDITS (COL1, COL2) 
SELECT 1, 1
union all
SELECT 2, 2

--We would assume row count to be 0, but returns 2
select @@ROWCOUNT

我可以解决这个问题，但我无法信任 @@ROWCOUNT 让我很困扰。我在 SO 或那些 other 知识库中找不到有关此问题的参考。这仅仅是 TRIGGERS ARE EVIL 的一个例子吗？

我可以影响@@ROWCOUNT 吗？

Answer 1

某些语句可能会更改触发器内的@@ROWCOUNT。

声明

SELECT * FROM INSERTED WHERE COL1 < 5

执行并将@@ROWCOUNT 设置为 1

Put 声明

SET NOCOUNT ON;

然后

IF NOT EXISTS (SELECT * FROM INSERTED WHERE COL1 < 5)
BEGIN
    SET NOCOUNT OFF;

    INSERT INTO dbo.ACCOUNT_CREDITS (COL1, COL2)
    SELECT COL1, COL2 from INSERTED
END

Answer 2

问题

我需要仅在触发器上下文中可用的主进程上下文中的信息。

解决方案

无论是从触发器获取@@ROWCOUNT 或任何其他，甚至将信息传递到触发器，有两种方法可以与触发器共享信息：

• SET CONTEXT_INFO/CONTEXT_INFO()

• 本地临时表（即名称以单个 #:#tmp 开头的表）

我在 DBA.StackExchange:Passing info on who deleted record onto a Delete trigger 的一个相关问题的答案中发布了一个使用 CONTEXT_INFO 的示例。 cmets 中讨论了与 CONTEXT_INFO 周围可能出现的并发症有关的答案，因此我使用临时表在该问题上发布了 another answer。

由于该示例处理发送信息到触发器，下面是一个获取信息的示例从触发器，因为这就是这个问题的意义：

首先：创建一个简单的表格

CREATE TABLE dbo.InsteadOfTriggerTest (Col1 INT);

第二：创建触发器

CREATE TRIGGER dbo.tr_InsteadOfTriggerTest
   ON dbo.InsteadOfTriggerTest
   INSTEAD OF INSERT
AS
BEGIN
   PRINT 'Trigger (starting): ' + CONVERT(NVARCHAR(50), @@ROWCOUNT);

   SET NOCOUNT ON; -- do AFTER the PRINT else @@ROWCOUNT will be 0

   DECLARE @Rows INT;
   INSERT INTO dbo.InsteadOfTriggerTest (Col1)
      SELECT TOP (5) ins.Col1
      FROM inserted ins;

   SET @Rows = @@ROWCOUNT;
   PRINT 'Trigger (after INSERT): ' + CONVERT(NVARCHAR(50), @Rows);

   -- make sure temp table exists; no error if table is missing
   IF (OBJECT_ID('tempdb..#TriggerRows') IS NOT NULL)
   BEGIN
      INSERT INTO #TriggerRows (RowsAffected)
      VALUES (@Rows);
   END;
END;

第三次：进行测试

SET NOCOUNT ON;

IF (OBJECT_ID('tempdb..#TriggerRows') IS NOT NULL)
BEGIN
    DROP TABLE #TriggerRows;
END;
CREATE TABLE #TriggerRows (RowsAffected INT);

INSERT INTO dbo.InsteadOfTriggerTest (Col1)
   SELECT so.[object_id]
   FROM   [master].[sys].[objects] so;

PRINT 'Final @@ROWCOUNT (what we do NOT want): ' + CONVERT(NVARCHAR(50), @@ROWCOUNT);

SELECT * FROM #TriggerRows;

输出（在消息选项卡中）：

触发器（开始）：91
触发器（插入后）：5
最终的@@ROWCOUNT（我们不想要的）：91

结果：

受影响的行
5