Reorganized API + added db migrations
Read the README file for more informations
This commit is contained in:
219
backend/cmd/migrate/main.go
Normal file
219
backend/cmd/migrate/main.go
Normal file
@@ -0,0 +1,219 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"database/sql"
|
||||
"fmt"
|
||||
"log"
|
||||
"os"
|
||||
"strings"
|
||||
|
||||
"fr.latosa-escrima/api/core"
|
||||
"fr.latosa-escrima/cmd/migrate/migrations"
|
||||
"github.com/joho/godotenv"
|
||||
"github.com/uptrace/bun/dialect/pgdialect"
|
||||
"github.com/uptrace/bun/driver/pgdriver"
|
||||
"github.com/uptrace/bun/extra/bundebug"
|
||||
"github.com/uptrace/bun/migrate"
|
||||
|
||||
"github.com/urfave/cli/v2"
|
||||
|
||||
"github.com/uptrace/bun"
|
||||
)
|
||||
|
||||
func main() {
|
||||
err := godotenv.Load()
|
||||
if err != nil {
|
||||
log.Fatalf("Error loading .env file: %v", err)
|
||||
}
|
||||
environ := os.Getenv("ENVIRONMENT")
|
||||
|
||||
hostname := os.Getenv("DATABASE_HOSTNAME")
|
||||
postgres_port := os.Getenv("POSTGRES_DOCKER_PORT")
|
||||
if environ == "DEV" {
|
||||
hostname = "localhost"
|
||||
postgres_port = os.Getenv("POSTGRES_PORT")
|
||||
}
|
||||
|
||||
dsn := core.DSN{
|
||||
Hostname: hostname,
|
||||
Port: postgres_port,
|
||||
DBName: os.Getenv("POSTGRES_DB"),
|
||||
User: os.Getenv("POSTGRES_USER"),
|
||||
Password: os.Getenv("POSTGRES_PASSWORD"),
|
||||
}
|
||||
fmt.Println(dsn.ToString())
|
||||
sqldb := sql.OpenDB(pgdriver.NewConnector(pgdriver.WithDSN(dsn.ToString())))
|
||||
db := bun.NewDB(sqldb, pgdialect.New())
|
||||
|
||||
db.AddQueryHook(bundebug.NewQueryHook(bundebug.WithVerbose(true)))
|
||||
|
||||
defer db.Close()
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
db.AddQueryHook(bundebug.NewQueryHook(
|
||||
bundebug.WithEnabled(false),
|
||||
bundebug.FromEnv(),
|
||||
))
|
||||
|
||||
app := &cli.App{
|
||||
Name: "bun",
|
||||
|
||||
Commands: []*cli.Command{
|
||||
newDBCommand(migrate.NewMigrator(db, migrations.Migrations)),
|
||||
},
|
||||
}
|
||||
if err := app.Run(os.Args); err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
}
|
||||
|
||||
func newDBCommand(migrator *migrate.Migrator) *cli.Command {
|
||||
return &cli.Command{
|
||||
Name: "db",
|
||||
Usage: "database migrations",
|
||||
Subcommands: []*cli.Command{
|
||||
{
|
||||
Name: "init",
|
||||
Usage: "create migration tables",
|
||||
Action: func(c *cli.Context) error {
|
||||
return migrator.Init(c.Context)
|
||||
},
|
||||
},
|
||||
{
|
||||
Name: "migrate",
|
||||
Usage: "migrate database",
|
||||
Action: func(c *cli.Context) error {
|
||||
if err := migrator.Lock(c.Context); err != nil {
|
||||
return err
|
||||
}
|
||||
defer migrator.Unlock(c.Context) //nolint:errcheck
|
||||
|
||||
group, err := migrator.Migrate(c.Context)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
if group.IsZero() {
|
||||
fmt.Printf("there are no new migrations to run (database is up to date)\n")
|
||||
return nil
|
||||
}
|
||||
fmt.Printf("migrated to %s\n", group)
|
||||
return nil
|
||||
},
|
||||
},
|
||||
{
|
||||
Name: "rollback",
|
||||
Usage: "rollback the last migration group",
|
||||
Action: func(c *cli.Context) error {
|
||||
if err := migrator.Lock(c.Context); err != nil {
|
||||
return err
|
||||
}
|
||||
defer migrator.Unlock(c.Context) //nolint:errcheck
|
||||
|
||||
group, err := migrator.Rollback(c.Context)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
if group.IsZero() {
|
||||
fmt.Printf("there are no groups to roll back\n")
|
||||
return nil
|
||||
}
|
||||
fmt.Printf("rolled back %s\n", group)
|
||||
return nil
|
||||
},
|
||||
},
|
||||
{
|
||||
Name: "lock",
|
||||
Usage: "lock migrations",
|
||||
Action: func(c *cli.Context) error {
|
||||
return migrator.Lock(c.Context)
|
||||
},
|
||||
},
|
||||
{
|
||||
Name: "unlock",
|
||||
Usage: "unlock migrations",
|
||||
Action: func(c *cli.Context) error {
|
||||
return migrator.Unlock(c.Context)
|
||||
},
|
||||
},
|
||||
{
|
||||
Name: "create_go",
|
||||
Usage: "create Go migration",
|
||||
Action: func(c *cli.Context) error {
|
||||
name := strings.Join(c.Args().Slice(), "_")
|
||||
mf, err := migrator.CreateGoMigration(c.Context, name)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
fmt.Printf("created migration %s (%s)\n", mf.Name, mf.Path)
|
||||
return nil
|
||||
},
|
||||
},
|
||||
{
|
||||
Name: "create_sql",
|
||||
Usage: "create up and down SQL migrations",
|
||||
Action: func(c *cli.Context) error {
|
||||
name := strings.Join(c.Args().Slice(), "_")
|
||||
files, err := migrator.CreateSQLMigrations(c.Context, name)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
for _, mf := range files {
|
||||
fmt.Printf("created migration %s (%s)\n", mf.Name, mf.Path)
|
||||
}
|
||||
|
||||
return nil
|
||||
},
|
||||
},
|
||||
{
|
||||
Name: "create_tx_sql",
|
||||
Usage: "create up and down transactional SQL migrations",
|
||||
Action: func(c *cli.Context) error {
|
||||
name := strings.Join(c.Args().Slice(), "_")
|
||||
files, err := migrator.CreateTxSQLMigrations(c.Context, name)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
for _, mf := range files {
|
||||
fmt.Printf("created transaction migration %s (%s)\n", mf.Name, mf.Path)
|
||||
}
|
||||
|
||||
return nil
|
||||
},
|
||||
},
|
||||
{
|
||||
Name: "status",
|
||||
Usage: "print migrations status",
|
||||
Action: func(c *cli.Context) error {
|
||||
ms, err := migrator.MigrationsWithStatus(c.Context)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
fmt.Printf("migrations: %s\n", ms)
|
||||
fmt.Printf("unapplied migrations: %s\n", ms.Unapplied())
|
||||
fmt.Printf("last migration group: %s\n", ms.LastGroup())
|
||||
return nil
|
||||
},
|
||||
},
|
||||
{
|
||||
Name: "mark_applied",
|
||||
Usage: "mark migrations as applied without actually running them",
|
||||
Action: func(c *cli.Context) error {
|
||||
group, err := migrator.Migrate(c.Context, migrate.WithNopMigration())
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
if group.IsZero() {
|
||||
fmt.Printf("there are no new migrations to mark as applied\n")
|
||||
return nil
|
||||
}
|
||||
fmt.Printf("marked as applied %s\n", group)
|
||||
return nil
|
||||
},
|
||||
},
|
||||
},
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1 @@
|
||||
ALTER TABLE users DROP COLUMN attributes;
|
||||
@@ -0,0 +1 @@
|
||||
ALTER TABLE users ADD COLUMN attributes jsonb;
|
||||
11
backend/cmd/migrate/migrations/main.go
Normal file
11
backend/cmd/migrate/migrations/main.go
Normal file
@@ -0,0 +1,11 @@
|
||||
package migrations
|
||||
|
||||
import "github.com/uptrace/bun/migrate"
|
||||
|
||||
var Migrations = migrate.NewMigrations()
|
||||
|
||||
func init() {
|
||||
if err := Migrations.DiscoverCaller(); err != nil {
|
||||
panic(err)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user