如何移除SQL变量中的GO命令 使脚本可被sp_executesql执行

实现思路 #

利用GO作为批处理命令时的格式特征：仅单独出现在行中，前后仅允许存在空格、制表符等空白字符，不会被引号包裹。我们可以通过标准化换行符、匹配独立行GO的规则来做精准替换，不会误删字符串内容中的GO字符。

具体实现代码 #

DECLARE @originalSQL NVARCHAR(MAX) = N'你读取到的.sql文件内容' -- 替换为你实际读取到的脚本内容
DECLARE @processedSQL NVARCHAR(MAX)

-- 第一步：统一换行符为\n，兼容不同操作系统生成的sql文件
SET @processedSQL = REPLACE(REPLACE(@originalSQL, CHAR(13) + CHAR(10), CHAR(10)), CHAR(13), CHAR(10))

-- 第二步：匹配所有独立行的GO（允许前后带空白字符）
WHILE CHARINDEX(CHAR(10) + 'GO' + CHAR(10), @processedSQL) > 0 
    OR CHARINDEX(CHAR(10) + 'GO ', @processedSQL) > 0
    OR CHARINDEX(' GO' + CHAR(10), @processedSQL) > 0
BEGIN
    -- 替换前后带空白的独立行GO
    SET @processedSQL = REPLACE(@processedSQL, CHAR(10) + 'GO' + CHAR(10), CHAR(10))
    SET @processedSQL = REPLACE(@processedSQL, CHAR(10) + 'GO ', CHAR(10))
    SET @processedSQL = REPLACE(@processedSQL, ' GO' + CHAR(10), CHAR(10))
    -- 处理带制表符的情况
    SET @processedSQL = REPLACE(@processedSQL, CHAR(10) + CHAR(9) + 'GO' + CHAR(10), CHAR(10))
END

-- 第三步：处理开头或结尾单独出现的GO
IF LEFT(@processedSQL, 2) = 'GO' AND (LEN(@processedSQL) = 2 OR SUBSTRING(@processedSQL, 3, 1) = CHAR(10))
    SET @processedSQL = STUFF(@processedSQL, 1, 2, '')
IF RIGHT(@processedSQL, 2) = 'GO' AND (LEN(@processedSQL) = 2 OR SUBSTRING(@processedSQL, LEN(@processedSQL)-2, 1) = CHAR(10))
    SET @processedSQL = STUFF(@processedSQL, LEN(@processedSQL)-1, 2, '')

-- 处理完成后执行
PRINT @processedSQL
EXEC sp_executesql @processedSQL

测试用例适配效果 #

将你给出的示例SQL代入后，会保留VALUES中的''GO''字符串、''Show must go on''中的go字符，仅移除最后单独占一行的GO命令，执行不会报错。

注意事项 #

• 如果需要兼容小写的go命令，可以将匹配逻辑改为大小写不敏感的匹配，或者先把待匹配的子串转为大写判断
• 如果需要支持GO 数字（重复执行N次）的语法，可以扩展匹配规则，匹配GO后面跟数字和空白的场景单独处理

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