本文介绍使用 Go 语言（配合 pgx 或 database/sql 驱动）安全、跨平台地读取并执行本地 .sql 文件，适用于数据库初始化、测试数据填充或迁移场景，避免依赖外部命令行工具。
本文介绍使用 go 语言（配合 pgx 或 database/sql 驱动）安全、跨平台地读取并执行本地 `.sql` 文件，适用于数据库初始化、测试数据填充或迁移场景，避免依赖外部命令行工具。
在 Go 应用中直接执行 SQL 脚本文件（如 schema.sql 或 seed.sql）是一种常见需求，尤其在服务启动时初始化数据库结构或预置测试数据。相比调用 psql -f script.sql 等 shell 命令，纯 Go 实现更可控、可移植性强，且能统一错误处理与连接生命周期管理。
✅ 推荐方案：使用 pgx（推荐）或标准 database/sql
虽然 lib/pq 仍可用，但 pgx 是目前最活跃、性能更优的 PostgreSQL Go 驱动，原生支持批量语句、类型强映射，并对多语句 SQL 字符串有良好兼容性（注意：database/sql 标准接口默认不支持一次执行多条语句，需手动拆分；而 pgx 的 Conn.Exec() 可直接运行含 ; 分隔的完整脚本）。
示例代码（基于 pgx v5）
package main
import (
"context"
"fmt"
"io/ioutil"
"path/filepath"
"github.com/jackc/pgx/v5"
)
func executeSQLFile(connStr, sqlFilePath string) error {
// 1. 安全构建文件路径（跨平台）
path := filepath.Clean(sqlFilePath)
// 2. 读取 SQL 文件内容
content, err := ioutil.ReadFile(path)
if err != nil {
return fmt.Errorf("failed to read SQL file %s: %w", path, err)
}
// 3. 建立数据库连接
ctx := context.Background()
conn, err := pgx.Connect(ctx, connStr)
if err != nil {
return fmt.Errorf("failed to connect to PostgreSQL: %w", err)
}
defer conn.Close(ctx)
// 4. 执行整个 SQL 脚本（pgx 支持多语句）
_, err = conn.Exec(ctx, string(content))
if err != nil {
return fmt.Errorf("failed to execute SQL file %s: %w", path, err)
}
fmt.Printf("✅ Successfully executed %s\n", path)
return nil
}
// 使用示例
func main() {
connStr := "postgres://user:pass@localhost:5432/mydb?sslmode=disable"
if err := executeSQLFile(connStr, "./migrations/init.sql"); err != nil {
panic(err)
}
}⚠️ 重要注意事项：
- SQL 注入风险：此方式仅适用于可信的本地文件（如项目内建的 migration 脚本），切勿用于动态拼接或用户上传的 SQL 内容。
- 事务控制：上述示例未启用事务。若脚本需原子性执行，应显式开启事务：
tx, err := conn.Begin(ctx) if err != nil { /* ... */ } _, err = tx.Exec(ctx, string(content)) if err != nil { tx.Rollback(ctx); return err } return tx.Commit(ctx)- 语句分隔与兼容性：pgx.Conn.Exec() 可处理以 ; 结尾的多语句（如 CREATE TABLE...; INSERT INTO...;），但不支持 \ 元命令（如 \dt, -- 注释会被忽略）。确保 SQL 文件为纯 DDL/DML。
- 替代方案（兼容 lib/pq）：若坚持使用 lib/pq 或标准 database/sql，需手动按 ; 拆分语句并逐条执行（注意空行、注释和字符串内分号的边界处理），推荐改用 pgx 简化逻辑。
✅ 最佳实践建议
- 将 SQL 文件置于 ./migrations/ 或 ./sql/ 目录，通过 embed.FS（Go 1.16+）编译进二进制，提升部署可靠性；
- 对关键脚本添加校验（如 SHA256 哈希比对），防止意外篡改；
- 在 CI/CD 或 main() 启动阶段封装为幂等操作（例如先检查表是否存在再执行 CREATE TABLE IF NOT EXISTS）。
通过以上方式，你可以在 Go 项目中以简洁、健壮、可维护的方式完成 SQL 文件的自动化执行，真正实现“数据库即代码”（Database-as-Code）的工程实践。
