如何在Golang中执行PostgreSQL SQL文件中的多段SQL语句？

嘿，这个问题我正好碰到过，给你几个实用的解决办法！

如果你的SQL文件里都是像建表这样的简单DDL语句，语句之间用分号清晰分隔，没有嵌套在字符串里的分号，那直接读取整个文件内容传给db.Exec就可以了，非常省心。

示例代码：

package main

import (
	"database/sql"
	"os"
	_ "github.com/lib/pq"
)

func main() {
	// 先建立数据库连接
	db, err := sql.Open("postgres", "postgres://user:password@host:port/dbname?sslmode=disable")
	if err != nil {
		panic(err)
	}
	defer db.Close()

	// 读取database.sql文件内容
	sqlContent, err := os.ReadFile("database.sql")
	if err != nil {
		panic(err)
	}

	// 执行整个SQL内容
	_, err = db.Exec(string(sqlContent))
	if err != nil {
		panic(err)
	}

	// 执行成功，启动你的服务器...
}

注意：这个方法依赖postgres驱动（比如lib/pq）支持多语句执行，它默认是支持的，但如果你的SQL里有分号出现在字符串常量里（比如INSERT INTO posts(content) VALUES('hello; world')），这种简单的整段执行可能会出错，这时候就需要用下面的方法。

如果你的SQL文件里有注释、空行，或者存在字符串里的分号，那可以先把SQL内容分割成单个语句，再逐个执行。这里给你一个简单的分割函数，能处理注释和空行：

示例代码：

package main

import (
	"database/sql"
	"os"
	"regexp"
	"strings"
	_ "github.com/lib/pq"
)

// 分割SQL语句，处理注释和空行
func splitSQLStatements(sql string) []string {
	// 去掉单行注释
	re := regexp.MustCompile(`--.*$`)
	sql = re.ReplaceAllString(sql, "")
	// 按分号分割语句
	statements := strings.Split(sql, ";")
	// 清理空语句和多余空格
	var cleanedStatements []string
	for _, stmt := range statements {
		trimmed := strings.TrimSpace(stmt)
		if trimmed != "" {
			cleanedStatements = append(cleanedStatements, trimmed)
		}
	}
	return cleanedStatements
}

func main() {
	db, err := sql.Open("postgres", "your_connection_string")
	if err != nil {
		panic(err)
	}
	defer db.Close()

	sqlContent, err := os.ReadFile("database.sql")
	if err != nil {
		panic(err)
	}

	// 分割成单个语句
	statements := splitSQLStatements(string(sqlContent))

	// 逐个执行
	for _, stmt := range statements {
		_, err := db.Exec(stmt)
		if err != nil {
			panic(err)
		}
	}
}

如果你的SQL非常复杂（比如有存储过程、多行字符串里的分号），那这个简单的正则可能不够用，这时候可以用专门的SQL解析库，比如github.com/xwb1989/sqlparser来精准分割语句。

如果你用的是pgx（PostgreSQL的高性能Go驱动），可以用它的Batch功能来批量执行多个SQL语句，效率更高，也更可靠：

示例代码：

package main

import (
	"context"
	"os"
	"regexp"
	"strings"

	"github.com/jackc/pgx/v5"
	"github.com/jackc/pgx/v5/pgconn"
)

func splitSQLStatements(sql string) []string {
	re := regexp.MustCompile(`--.*$`)
	sql = re.ReplaceAllString(sql, "")
	statements := strings.Split(sql, ";")
	var cleaned []string
	for _, s := range statements {
		trimmed := strings.TrimSpace(s)
		if trimmed != "" {
			cleaned = append(cleaned, trimmed)
		}
	}
	return cleaned
}

func main() {
	ctx := context.Background()
	// 建立pgx连接
	conn, err := pgx.Connect(ctx, "postgres://user:password@host:port/dbname?sslmode=disable")
	if err != nil {
		panic(err)
	}
	defer conn.Close(ctx)

	sqlContent, err := os.ReadFile("database.sql")
	if err != nil {
		panic(err)
	}

	statements := splitSQLStatements(string(sqlContent))

	// 创建Batch对象，添加所有语句
	batch := &pgx.Batch{}
	for _, stmt := range statements {
		batch.Queue(stmt)
	}

	// 发送批量请求
	results := conn.SendBatch(ctx, batch)
	defer results.Close()

	// 处理每个结果
	for range statements {
		_, err := results.Exec()
		if err != nil {
			if pgErr, ok := err.(*pgconn.PgError); ok {
				panic("SQL执行错误: " + pgErr.Message)
			}
			panic(err)
		}
	}
}

根据你的实际情况选一个方法就好，一般建表这种场景用方法一就足够啦！

内容的提问来源于stack exchange，提问作者Arthur Brave