51 lines
1.2 KiB
Go
51 lines
1.2 KiB
Go
package postgres
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"fmt"
|
|
"log/slog"
|
|
|
|
_ "github.com/jackc/pgx/v5/stdlib" // registra o driver "pgx" no database/sql
|
|
"github.com/pressly/goose/v3"
|
|
|
|
"botpontos/db/migrations"
|
|
"botpontos/internal/config"
|
|
)
|
|
|
|
// Migrate executa o goose com as migrations embutidas no binário.
|
|
// Deve rodar com o usuário dono do banco, não com o usuário da aplicação.
|
|
func Migrate(ctx context.Context, cfg config.DB, command string, logger *slog.Logger) error {
|
|
db, err := sql.Open("pgx", cfg.DSN())
|
|
if err != nil {
|
|
return fmt.Errorf("abrir conexão para migrations: %w", err)
|
|
}
|
|
defer db.Close()
|
|
|
|
if err := waitFor(ctx, logger, db.PingContext); err != nil {
|
|
return err
|
|
}
|
|
|
|
goose.SetBaseFS(migrations.FS)
|
|
if err := goose.SetDialect("postgres"); err != nil {
|
|
return err
|
|
}
|
|
|
|
switch command {
|
|
case "up":
|
|
err = goose.UpContext(ctx, db, ".")
|
|
case "status":
|
|
err = goose.StatusContext(ctx, db, ".")
|
|
case "down":
|
|
err = goose.DownContext(ctx, db, ".")
|
|
default:
|
|
return fmt.Errorf("subcomando de migrate desconhecido: %s (use up, status ou down)", command)
|
|
}
|
|
if err != nil {
|
|
return fmt.Errorf("migrate %s: %w", command, err)
|
|
}
|
|
|
|
logger.Info("migrations concluídas", "command", command)
|
|
return nil
|
|
}
|