如何在 Go 中安全执行 MSSQL 脚本文件-Golang-PHP中文网

如何在 Go 中安全执行 MSSQL 脚本文件

本文介绍在 Go 中读取并执行本地 .sql 文件（如 SQL Server 脚本）的正确方法，重点解决因不可见控制字符、BOM 或扩展 ASCII 导致的 Incorrect syntax near '' 错误，并提供健壮的字符串清洗与分批执行方案。

本文介绍在 go 中读取并执行本地 `.sql` 文件（如 sql server 脚本）的正确方法，重点解决因不可见控制字符、bom 或扩展 ascii 导致的 `incorrect syntax near ''` 错误，并提供健壮的字符串清洗与分批执行方案。

在 Go 中直接通过 db.Exec() 执行从文件读取的 SQL 脚本看似简单，但实践中极易因文件编码、隐藏字符或批量语句分隔问题而失败。你遇到的 Incorrect syntax near '' 错误，绝大多数情况下并非语法错误，而是脚本内容中混入了不可见字符（如 UTF-8 BOM、Windows 换行符 \r\n 的异常处理、零宽空格、或非 ASCII 控制字符），导致 SQL Server 解析器在首字节或分号前意外截断。

以下是一个生产就绪的解决方案，包含三重保障：
✅ 安全读取（自动跳过 BOM）
✅ 严格清洗（移除所有控制字符及扩展 ASCII）
✅ 智能分批（按 GO 批处理分隔符拆分，兼容 SQL Server 语义）

package main

import (
    "bufio"
    "bytes"
    "database/sql"
    "fmt"
    "os"
    "regexp"
    "strings"

    _ "github.com/denisenkom/go-mssqldb"
)

// stripCtlAndExtFromBytes 移除字符串中所有控制字符（ASCII  126）
func stripCtlAndExtFromBytes(s string) string {
    b := make([]byte, 0, len(s))
    for i := 0; i = 32 && c  0 {
                batches = append(batches, strings.TrimSpace(currentBatch.String()))
                currentBatch.Reset()
            }
        } else {
            currentBatch.WriteString(line + "\n")
        }
    }
    // 添加最后一个批次（无 GO 结尾时）
    if currentBatch.Len() > 0 {
        batches = append(batches, strings.TrimSpace(currentBatch.String()))
    }
    return batches
}

func ExecuteSQLScriptFromFile(filePath string, connectionString string) error {
    // 1. 安全读取文件（自动处理 BOM）
    data, err := os.ReadFile(filePath)
    if err != nil {
        return fmt.Errorf("failed to read script file: %w", err)
    }

    // 2. 清洗不可见/非法字符
    cleaned := stripCtlAndExtFromBytes(string(data))

    // 3. 按 GO 分割为多个语句批次
    batches := splitSQLScriptByGO(cleaned)
    if len(batches) == 0 {
        return fmt.Errorf("no executable SQL batches found in script")
    }

    // 4. 连接数据库
    db, err := sql.Open("mssql", connectionString)
    if err != nil {
        return fmt.Errorf("failed to open database: %w", err)
    }
    defer db.Close()

    // 5. 逐批执行（重要：每批必须是单个完整语句或批处理）
    for i, batch := range batches {
        if strings.TrimSpace(batch) == "" {
            continue // 跳过空批次
        }
        _, err := db.Exec(batch)
        if err != nil {
            return fmt.Errorf("failed to execute batch %d (%d chars): %w", i+1, len(batch), err)
        }
    }

    return nil
}

// 使用示例
func main() {
    err := ExecuteSQLScriptFromFile("./myscript.sql", "server=localhost;user id=sa;password=yourpass;database=mydb;")
    if err != nil {
        fmt.Printf("Error: %v\n", err)
        return
    }
    fmt.Println("SQL script executed successfully.")
}

⚠️ 关键注意事项：

不要用 os.Open + bytes.Buffer.ReadFrom：该方式不处理 BOM，且对换行符敏感；改用 os.ReadFile 更简洁安全。
GO 不是 T-SQL 关键字：它是 SQL Server Management Studio (SSMS) 和 sqlcmd 的客户端批处理分隔符，不能直接传给 db.Exec() —— 必须先按 GO 拆分，再对每个子串单独调用 Exec。
避免长事务风险：大型脚本建议添加事务控制（db.Begin() / tx.Commit()），并在关键批次后加日志。
编码统一：确保 .sql 文件保存为 UTF-8 without BOM（推荐用 VS Code 或 Notepad++ 显式设置）。