Template
61 lines
1.1 KiB
Go
61 lines
1.1 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"flag"
|
|
"log"
|
|
"os"
|
|
"path/filepath"
|
|
|
|
"github.com/panelhosting/panel/internal/config"
|
|
"github.com/panelhosting/panel/internal/database"
|
|
)
|
|
|
|
func main() {
|
|
down := flag.Bool("down", false, "rollback all migrations")
|
|
flag.Parse()
|
|
|
|
cfg, err := config.Load()
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
migrationsPath, err := resolveMigrationsPath()
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
if *down {
|
|
if err := database.RollbackMigrations(cfg.DatabaseURL, migrationsPath); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
log.Println("migrations rolled back")
|
|
return
|
|
}
|
|
|
|
if err := database.RunMigrations(cfg.DatabaseURL, migrationsPath); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
log.Println("migrations applied")
|
|
|
|
ctx := context.Background()
|
|
pool, err := database.NewPool(ctx, cfg.DatabaseURL)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
defer pool.Close()
|
|
log.Println("database connection ok")
|
|
}
|
|
|
|
func resolveMigrationsPath() (string, error) {
|
|
if p := os.Getenv("MIGRATIONS_PATH"); p != "" {
|
|
return p, nil
|
|
}
|
|
|
|
wd, err := os.Getwd()
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
return filepath.Join(wd, "migrations"), nil
|
|
}
|