如何在 Go 中安全高效地执行多语句 SQL 脚本(含文件与字符串)

go 的 `database/sql` 标准库不支持单次 `exec()` 执行含多个分号的 sql 脚本,需手动拆分语句并逐条执行(推荐在事务中完成),或使用成熟的迁移工具如 goose 或 rambler 实现可维护、可回滚的数据库初始化。

在 Go 应用安装或初始化阶段,常需执行一整套 DDL(建表)和 DML(初始化数据)SQL 脚本。但直接将多语句拼接为单个字符串(如 "CREATE TABLE users(...); INSERT INTO users VALUES (...);")并调用 db.Exec(sqlStr) 会触发错误:

pq: cannot insert multiple commands into a prepared statement

这是因为 database/sql(及其底层驱动,如 lib/pq 或 pgx)明确禁止在单次查询中执行多条 SQL 语句——这是出于安全(防注入)、协议兼容性及事务语义清晰性的设计考量,并非 PostgreSQL 驱动特有,而是标准库层面的统一限制。

✅ 推荐方案:事务内安全拆分执行

最轻量、可控且无需引入外部依赖的方式是:以分号 ; 为界拆分语句,在显式事务中逐条执行。注意需严格处理空白、换行和注释,避免空语句触发错误:

func executeSQLScript(db *sql.DB, script string) error {
    tx, err := db.Begin()
    if err != nil {
        return fmt.Errorf("failed to begin transaction: %w", err)
    }
    defer func() {
        if tx != nil {
            tx.Rollback() // 若未 commit,确保回滚
        }
    }()

    // 拆分语句:支持 ;、;\n、;\r\n,忽略空行和纯空白
    statements := strings.FieldsFunc(script, func(r rune) bool {
        return r == ';' || r == '\n' || r == '\r'
    })

    for i, stmt := range statements {
        stmt = strings.TrimSpace(stmt)
        if stmt == "" {
            continue // 跳过空语句
        }

        _, err := tx.Exec(stmt)
        if err != nil {
            return fmt.Errorf("failed to execute statement #%d (%q): %w", i+1, stmt[:min(len(stmt), 50)], err)
        }
    }

    if err := tx.Commit(); err != nil {
        return fmt.Errorf("failed to commit transaction: %w", err)
    }
    tx = nil // 标记已提交
    return nil
}

? 关键注意事项:

  • 使用 strings.FieldsFunc 或正则 regexp.MustCompile((?m)^\s;?\s$) 更鲁棒地处理边界;
  • 务必在 defer 中检查 tx != nil 再调用 Rollback(),防止已提交后误回滚;
  • 日志中截断长语句(如 stmt[:min(len(stmt),50)])便于调试;
  • 不建议用 strings.Split(script, ";") —— 它无法处理语句内含分号(如字符串字面量 'a;b')或注释中的分号,生产环境应改用 SQL 解析器(见下文进阶方案)。

? 进阶方案:采用专业迁移工具

对于中大型项目,硬编码 SQL 字符串或手动拆分脚本难以维护、不可审计、缺乏版本控制与回滚能力。推荐使用成熟迁移工具:

工具 特点 适用场景
goose 支持 Go 和 SQL 迁移,带版本号、up/down、CLI 与 API,社区活跃 需要灵活逻辑(如条件初始化)、团队协作
rambler 纯 SQL 文件驱动,轻量( 简单项目,偏好“SQL 即代码”,拒绝 Go 逻辑侵入

示例(goose):

goose postgres "user=me dbname=test sslmode=disable" up

配合 20250101120000_init_schema.sql 文件,自动执行且记录迁移状态。

✅ 总结

  • ❌ 不要尝试绕过 database/sql 多语句限制(如修改驱动);
  • ✅ 简单场景:用事务 + 安全拆分(过滤空语句)执行内联 SQL 字符串;
  • ✅ 生产环境:务必使用 goose / rambler 等迁移工具,实

    现可重复、可追溯、可回滚的数据库 Schema 管理;
  • ? 永远避免拼接用户输入到 SQL 脚本中——迁移工具通常也要求 SQL 文件静态化,天然规避注入风险。