从 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 或那些其他知识库中找不到任何关于此问题的参考。这只是触发器是邪恶的吗？

我可以影响@@ROWCOUNT吗？

Answer 1

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

执行语句

SELECT * FROM INSERTED WHERE COL1 < 5

并将 @@ROWCOUNT 设置为 1

语句

SET NOCOUNT ON;

Put

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)

我发布了一个使用 CONTEXT_INFO 的示例> 在 DBA.StackExchange 上的相关问题的回答中： 将有关删除记录的人的信息传递到删除触发器。评论中对该答案进行了讨论，该答案与 CONTEXT_INFO 可能出现的并发症有关，因此我发布了 关于该问题的另一个答案，而是使用临时表。

由于该示例涉及向触发器发送信息，因此下面是从触发器获取信息的示例，这就是该问题的内容：

第一：< /strong> 创建一个简单的表

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