在MS SQL Server中执行DML前确定受影响表与行的方案

一、提取目标表 #

方法1：用SMO解析SQL语法（可靠，支持复杂场景） #

SQL Server的SMO（SQL Server Management Objects）是官方提供的工具，能精准解析SQL语句的结构，不管是单表DML还是带JOIN、别名的复杂语句都能处理。以下是C#示例代码：

using Microsoft.SqlServer.Management.Smo;
using Microsoft.SqlServer.Management.Common;

var targetSql = "update CUSTOMERS set NAME = 'XYZ' where age > 60;";
var serverConn = new ServerConnection("你的数据库实例名");
var sqlParser = new Parser(targetSql);
var parsedTree = sqlParser.Parse();

// 遍历解析树提取表名
foreach (var token in parsedTree.Tokens)
{
    if (token.TokenType == TokenType.Table)
    {
        Console.WriteLine($"受影响表：{token.Text}");
    }
}

方法2：TSQL字符串截取（适合简单单表场景） #

如果只处理简单的单表UPDATE/DELETE/INSERT，可以直接通过字符串处理提取表名，缺点是复杂SQL会失效：

DECLARE @sql NVARCHAR(MAX) = 'update CUSTOMERS set NAME = ''XYZ'' where age > 60;';
DECLARE @tableName NVARCHAR(128);

-- 提取UPDATE后的表名
IF @sql LIKE 'UPDATE%'
BEGIN
    SET @sql = LTRIM(SUBSTRING(@sql, CHARINDEX('UPDATE', @sql) + 6, LEN(@sql)));
    SET @tableName = LEFT(@sql, CHARINDEX(' ', @sql) - 1);
    -- 去除可能的方括号
    SET @tableName = REPLACE(REPLACE(@tableName, '[', ''), ']', '');
    PRINT '受影响表：' + @tableName;
END

二、预提取受影响行的主键 #

核心思路是把DML语句转换成查询主键的SELECT语句，执行后拿到所有会被修改/删除的主键值：

步骤1：获取目标表的主键列 #

先通过系统视图查询表的主键字段：

DECLARE @tableName NVARCHAR(128) = 'CUSTOMERS';
DECLARE @pkColumns NVARCHAR(MAX);

SELECT @pkColumns = STRING_AGG(c.COLUMN_NAME, ', ')
FROM INFORMATION_SCHEMA.TABLE_CONSTRAINTS tc
JOIN INFORMATION_SCHEMA.KEY_COLUMN_USAGE c 
    ON tc.CONSTRAINT_NAME = c.CONSTRAINT_NAME
WHERE tc.TABLE_NAME = @tableName 
  AND tc.CONSTRAINT_TYPE = 'PRIMARY KEY';

PRINT '主键列：' + @pkColumns;

步骤2：转换DML为SELECT主键语句 #

以UPDATE为例，提取原语句的WHERE条件，拼接成查询主键的SQL：

DECLARE @originalSql NVARCHAR(MAX) = 'update CUSTOMERS set NAME = ''XYZ'' where age > 60;';
DECLARE @selectPkSql NVARCHAR(MAX);

IF @originalSql LIKE '%WHERE%'
BEGIN
    -- 保留原WHERE条件，拼接查询主键的语句
    SET @selectPkSql = 'SELECT ' + @pkColumns + ' FROM ' + @tableName + ' ' + SUBSTRING(@originalSql, CHARINDEX('WHERE', @originalSql), LEN(@originalSql));
END
ELSE
BEGIN
    -- 无WHERE条件，全表受影响
    SET @selectPkSql = 'SELECT ' + @pkColumns + ' FROM ' + @tableName;
END

-- 执行查询，获取所有即将被影响的主键
EXEC sp_executesql @selectPkSql;

• DELETE语句的转换逻辑和UPDATE完全一致；
• 对于INSERT语句，如果是显式指定主键值的，可以直接从VALUES子句提取；如果是自增主键，执行前无法精准获取（只能预估范围，建议INSERT后用SCOPE_IDENTITY()获取）。

注意事项 #

• 带参数的DML要保留参数，比如update CUSTOMERS set NAME = @name where age > @age;，转换后的SELECT也要用相同参数，避免结果偏差；
• 复杂DML（比如UPDATE ... FROM ... JOIN ...）需要更细致的语法解析，确保JOIN和WHERE条件完全保留；
• 操作时需要对应表的SELECT权限。

内容的提问来源于stack exchange，提问作者Vijay