55 lines
985 B
Go
55 lines
985 B
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
"time"
|
|
|
|
"allapp-go/internal/bootstrap/boots"
|
|
"allapp-go/internal/config"
|
|
)
|
|
|
|
func main() {
|
|
if len(os.Args) != 2 {
|
|
fmt.Fprintln(os.Stderr, "usage: go run ./cmd/apply_sql <sql-file>")
|
|
os.Exit(1)
|
|
}
|
|
|
|
sqlPath, err := filepath.Abs(os.Args[1])
|
|
if err != nil {
|
|
exitErr("resolve sql path", err)
|
|
}
|
|
|
|
sqlBytes, err := os.ReadFile(sqlPath)
|
|
if err != nil {
|
|
exitErr("read sql file", err)
|
|
}
|
|
|
|
cfg, err := config.Load()
|
|
if err != nil {
|
|
exitErr("load config", err)
|
|
}
|
|
|
|
pool, err := boots.InitPgsql(cfg)
|
|
if err != nil {
|
|
exitErr("connect postgres", err)
|
|
}
|
|
defer pool.Close()
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), 60*time.Second)
|
|
defer cancel()
|
|
|
|
if _, err := pool.Exec(ctx, string(sqlBytes)); err != nil {
|
|
exitErr("execute sql", err)
|
|
}
|
|
|
|
fmt.Printf("applied sql: %s\n", sqlPath)
|
|
}
|
|
|
|
func exitErr(step string, err error) {
|
|
fmt.Fprintf(os.Stderr, "%s failed: %v\n", step, err)
|
|
os.Exit(1)
|
|
}
|