通过正则表达式判断 SQL 命令是否为 UPDATE 命令

Question

我有很多来自实体框架拦截器的 SQL 命令，例如：

UPDATE [dbo].[Products]
SET [Name] = @0
WHERE ([Id] = @1)

如何确定该命令是UPDATE 命令？

我写了以下正则表达式：

(update)? [a-z0-9-]+ (set)

但它不支持多行命令且命令文本应为一行，并且不支持表名中带有模式和[]字符的表名。

只支持简单的SQL命令如下：

UPDATE Products SET [Name] = @0 WHERE ([Id] = @1)

注意： 我正在使用实体框架的这个正则表达式和IDbCommandInterceptor 功能为UPDATE 命令向SQL 服务器提供WITH (ROWLOCK) 表提示，所以我需要一个从UPDATE 关键字到SET 关键字的正则表达式，因为@987654331 @ 应该插入到 SET 关键字之前。

考虑以下代码：

String updateCommandRegularExpression = "(update)? [a-z0-9-]+ (set)";

Boolean isUpdateCommand = Regex.IsMatch(commandText, updateCommandRegularExpression, RegexOptions.IgnoreCase | RegexOptions.Multiline); // You may use better regular expression pattern here.

if (isUpdateCommand)
{
    Boolean isSnapshotIsolationTransaction = sqlCommand.Transaction != null && sqlCommand.Transaction.IsolationLevel == IsolationLevel.Snapshot;
    // Transactions with Snapshot isolation level have more complicated table hints to enable rowlock.

    String tableHintToAdd = isSnapshotIsolationTransaction ? " with (rowlock , updlock) set " : " with (rowlock) set ";

    commandText = Regex.Replace(commandText, updateCommandRegularExpression, (match) =>
    {
        return Regex.Replace(match.Value, " SET ", tableHintToAdd, RegexOptions.IgnoreCase); // You may use better regular expression here.
    }, RegexOptions.IgnoreCase | RegexOptions.Multiline);

    command.CommandText = commandText;
}

Answer 1

这样的事情怎么样？

'[; \t]*[Uu][Pp][Dd][Aa][Tt][Ee][ \t]'

基本上，更新命令是第一个单词是update 的命令——至少对于简单命令是这样。如果允许公用表表达式（with 语句），则逻辑更具挑战性。

编辑：

我建议采用以下方法。使用上面的表达式来确定命令是否是更新。然后，将“set”替换为“with (ROWLOCK) set”。这将处理表可能包含特殊字符或使用别名的情况。一个缺点是如果 table 包含“set”将无济于事。

由于您是在 C# 中执行此操作，因此您可以测试该集合前后的字符是否为空白字符而不是字母数字。

注意：如果update 已经有提示，这将无法正常工作。您可以通过在“set”之前查找“with”来查找。