129 lines
3.4 KiB
Go
129 lines
3.4 KiB
Go
// dbtool 数据库初始化工具：读取 sql/init.sql 并逐条执行
|
||
// 用法（在 nl-game-api-gin 目录下）：go run ./cmd/dbtool -f ../sql/init.sql
|
||
// 连接参数复用 config.yaml 中的 mysql 配置（不指定库名，脚本自带 CREATE DATABASE）
|
||
package main
|
||
|
||
import (
|
||
"database/sql"
|
||
"flag"
|
||
"fmt"
|
||
"log"
|
||
"os"
|
||
"strings"
|
||
|
||
_ "github.com/go-sql-driver/mysql"
|
||
|
||
"nl-game-api-gin/internal/config"
|
||
)
|
||
|
||
func main() {
|
||
file := flag.String("f", "../sql/init.sql", "SQL 脚本路径")
|
||
conf := flag.String("c", "config.yaml", "配置文件路径")
|
||
flag.Parse()
|
||
if err := config.Load(*conf); err != nil {
|
||
log.Fatalln("配置加载失败:", err)
|
||
}
|
||
raw, err := os.ReadFile(*file)
|
||
if err != nil {
|
||
log.Fatalln("读取 SQL 文件失败:", err)
|
||
}
|
||
// 服务器级 DSN（不带库名），multiStatements 关闭、逐条执行便于定位错误
|
||
dsn := fmt.Sprintf("%s:%s@tcp(%s:%d)/?charset=utf8mb4&parseTime=false",
|
||
config.C.MySQL.User, config.C.MySQL.Password, config.C.MySQL.Host, config.C.MySQL.Port)
|
||
db, err := sql.Open("mysql", dsn)
|
||
if err != nil {
|
||
log.Fatalln("连接 MySQL 失败:", err)
|
||
}
|
||
defer db.Close()
|
||
// 限制单连接：保证 USE xiaoyouxi 对后续语句生效
|
||
db.SetMaxOpenConns(1)
|
||
if err := db.Ping(); err != nil {
|
||
log.Fatalln("MySQL 无法连通，请确认服务已启动且账号密码正确:", err)
|
||
}
|
||
stmts := splitSQL(string(raw))
|
||
log.Printf("共解析出 %d 条语句，开始执行…\n", len(stmts))
|
||
for i, s := range stmts {
|
||
if _, err := db.Exec(s); err != nil {
|
||
head := s
|
||
if len(head) > 80 {
|
||
head = head[:80] + "…"
|
||
}
|
||
log.Fatalf("第 %d 条语句执行失败: %v\n语句开头: %s", i+1, err, head)
|
||
}
|
||
}
|
||
log.Println("✅ 数据库初始化完成（库名 xiaoyouxi，含种子数据与超管 liqi）")
|
||
}
|
||
|
||
// splitSQL 把脚本按分号拆成独立语句：跟踪引号与注释状态，避免误切字符串内的分号
|
||
func splitSQL(text string) []string {
|
||
var stmts []string
|
||
var sb strings.Builder
|
||
inSingle, inDouble, inBacktick, inLineComment := false, false, false, false
|
||
runes := []rune(text)
|
||
for i := 0; i < len(runes); i++ {
|
||
ch := runes[i]
|
||
// 行注释：遇到换行结束（注释内容不写入语句）
|
||
if inLineComment {
|
||
if ch == '\n' {
|
||
inLineComment = false
|
||
}
|
||
continue
|
||
}
|
||
// 检测行注释起始（不在任何引号内）
|
||
if !inSingle && !inDouble && !inBacktick && ch == '-' && i+1 < len(runes) && runes[i+1] == '-' {
|
||
inLineComment = true
|
||
i++
|
||
continue
|
||
}
|
||
// 引号状态翻转（MySQL 字符串内以 \' 转义，需要跳过转义符）
|
||
switch {
|
||
case inSingle:
|
||
if ch == '\\' && i+1 < len(runes) {
|
||
sb.WriteRune(ch)
|
||
i++
|
||
sb.WriteRune(runes[i])
|
||
continue
|
||
}
|
||
if ch == '\'' {
|
||
inSingle = false
|
||
}
|
||
case inDouble:
|
||
if ch == '\\' && i+1 < len(runes) {
|
||
sb.WriteRune(ch)
|
||
i++
|
||
sb.WriteRune(runes[i])
|
||
continue
|
||
}
|
||
if ch == '"' {
|
||
inDouble = false
|
||
}
|
||
case inBacktick:
|
||
if ch == '`' {
|
||
inBacktick = false
|
||
}
|
||
default:
|
||
if ch == '\'' {
|
||
inSingle = true
|
||
} else if ch == '"' {
|
||
inDouble = true
|
||
} else if ch == '`' {
|
||
inBacktick = true
|
||
} else if ch == ';' {
|
||
// 语句结束
|
||
stmt := strings.TrimSpace(sb.String())
|
||
if stmt != "" {
|
||
stmts = append(stmts, stmt)
|
||
}
|
||
sb.Reset()
|
||
continue
|
||
}
|
||
}
|
||
sb.WriteRune(ch)
|
||
}
|
||
// 收尾：最后一条可能没有分号
|
||
if tail := strings.TrimSpace(sb.String()); tail != "" {
|
||
stmts = append(stmts, tail)
|
||
}
|
||
return stmts
|
||
}
|