code-utils/tools/applyinit/main.go

// Applies nl-pms-api/init.sql to MySQL.
// Usage (from view repo root):
//
//	go run ./tools/applyinit -dsn "user:pass@tcp(host:3306)/"
//
// Or set MYSQL_DSN. Schema source of truth is ../nl-pms-api/init.sql.
package main

import (
	"database/sql"
	"flag"
	"fmt"
	"os"
	"path/filepath"
	"strings"

	_ "github.com/go-sql-driver/mysql"
)

func main() {
	dsnFlag := flag.String("dsn", "", "MySQL DSN without database (or set MYSQL_DSN)")
	sqlPath := flag.String("sql", "", "path to init.sql (default: ../nl-pms-api/init.sql)")
	flag.Parse()
	dsn := strings.TrimSpace(*dsnFlag)
	if dsn == "" {
		dsn = strings.TrimSpace(os.Getenv("MYSQL_DSN"))
	}
	if dsn == "" {
		fmt.Fprintln(os.Stderr, "缺少 -dsn 或 MYSQL_DSN（例: root:root@tcp(127.0.0.1:3306)/）")
		os.Exit(2)
	}
	if !strings.Contains(dsn, "multiStatements") {
		if strings.Contains(dsn, "?") {
			dsn += "&multiStatements=false&charset=utf8mb4"
		} else {
			dsn += "?multiStatements=false&charset=utf8mb4"
		}
	}
	path := *sqlPath
	if path == "" {
		path = filepath.Join("..", "nl-pms-api", "init.sql")
	}
	raw, err := os.ReadFile(path)
	if err != nil {
		fmt.Fprintln(os.Stderr, err)
		os.Exit(1)
	}
	db, err := sql.Open("mysql", dsn)
	if err != nil {
		panic(err)
	}
	defer db.Close()
	db.SetMaxOpenConns(1)

	var kept []string
	for _, line := range strings.Split(string(raw), "\n") {
		if t := strings.TrimSpace(line); strings.HasPrefix(t, "--") {
			continue
		}
		kept = append(kept, line)
	}
	for _, stmt := range strings.Split(strings.Join(kept, "\n"), ";") {
		s := strings.TrimSpace(stmt)
		if s == "" {
			continue
		}
		if _, err := db.Exec(s); err != nil {
			fmt.Println("ERR:", err, "stmt:", s[:min(80, len(s))])
			os.Exit(1)
		}
	}
	var n int
	if err := db.QueryRow("SELECT COUNT(*) FROM code_count.users").Scan(&n); err != nil {
		fmt.Println("verify failed:", err)
		os.Exit(1)
	}
	fmt.Println("ok, users rows:", n, "sql:", path)
}

func min(a, b int) int {
	if a < b {
		return a
	}
	return b
}