86 lines
2.6 KiB
Go
86 lines
2.6 KiB
Go
// applysql —— 开发辅助工具：把 SQL 文件应用到 z_xk 库
|
||
//
|
||
// 为什么需要它：Windows 开发机没有 mysql 客户端，而 Go 项目本身带着
|
||
// GORM 依赖与 config.Load() 的 DSN 解析逻辑，直接复用即可执行 DDL，
|
||
// 且凭据不经过命令行/控制台（从 manifest/config/config.yaml 读取）。
|
||
//
|
||
// 用法（必须在 nl-tcm-agent 项目根目录执行，保证能读到配置文件）：
|
||
// go run ./tools/applysql <sql文件1> [sql文件2 ...]
|
||
//
|
||
// 说明：
|
||
// - 按 ";\n" 粗粒度拆分语句（我们自己的 SQL 文件都是规范的分号+换行结尾）
|
||
// - "--" 开头的注释行会被剥掉
|
||
// - 任一语句失败立即退出（DDL 顺序有依赖，跳过会导致后续更混乱）
|
||
package main
|
||
|
||
import (
|
||
"fmt"
|
||
"log"
|
||
"os"
|
||
"strings"
|
||
|
||
"tcm-agent/internal/config"
|
||
"tcm-agent/internal/dao"
|
||
)
|
||
|
||
func main() {
|
||
if len(os.Args) < 2 {
|
||
log.Fatal("用法: go run ./tools/applysql <sql文件...>")
|
||
}
|
||
cfg := config.Load()
|
||
if cfg.DB.DSN == "" {
|
||
log.Fatal("配置缺少 db.dsn（检查 manifest/config/config.yaml）")
|
||
}
|
||
if err := dao.Init(cfg.DB.DSN); err != nil {
|
||
log.Fatalf("连接数据库失败: %v", err)
|
||
}
|
||
for _, path := range os.Args[1:] {
|
||
raw, err := os.ReadFile(path)
|
||
if err != nil {
|
||
log.Fatalf("读取 %s 失败: %v", path, err)
|
||
}
|
||
stmts := splitStatements(string(raw))
|
||
fmt.Printf("== %s：%d 条语句 ==\n", path, len(stmts))
|
||
for i, stmt := range stmts {
|
||
if err := dao.DB.Exec(stmt).Error; err != nil {
|
||
log.Fatalf("第 %d 条语句执行失败: %v\n语句片段: %.120s", i+1, err, stmt)
|
||
}
|
||
fmt.Printf(" [%d/%d] OK %.60s...\n", i+1, len(stmts), firstLine(stmt))
|
||
}
|
||
}
|
||
fmt.Println("全部执行完成")
|
||
}
|
||
|
||
// splitStatements 把 SQL 文本拆成可执行语句：剥注释行后按分号结尾切分
|
||
func splitStatements(sqlText string) []string {
|
||
// Windows 编辑器写出的文件是 CRLF，先统一成 LF，否则 ";\n" 切分匹配不到
|
||
sqlText = strings.ReplaceAll(sqlText, "\r\n", "\n")
|
||
var sb strings.Builder
|
||
for _, line := range strings.Split(sqlText, "\n") {
|
||
trimmed := strings.TrimSpace(line)
|
||
// 剥掉整行注释（字段内注释在行中不受影响）
|
||
if strings.HasPrefix(trimmed, "--") || trimmed == "" {
|
||
continue
|
||
}
|
||
sb.WriteString(line)
|
||
sb.WriteString("\n")
|
||
}
|
||
parts := strings.Split(sb.String(), ";\n")
|
||
out := make([]string, 0, len(parts))
|
||
for _, p := range parts {
|
||
p = strings.TrimSpace(p)
|
||
if p != "" {
|
||
out = append(out, p)
|
||
}
|
||
}
|
||
return out
|
||
}
|
||
|
||
// firstLine 取语句首行用于进度展示
|
||
func firstLine(stmt string) string {
|
||
if i := strings.Index(stmt, "\n"); i > 0 {
|
||
return stmt[:i]
|
||
}
|
||
return stmt
|
||
}
|