Allow migrating up and down
This commit is contained in:
parent
64e48a5689
commit
9ba477174d
2 changed files with 41 additions and 5 deletions
|
@ -2,6 +2,7 @@ package db
|
|||
|
||||
import (
|
||||
"fmt"
|
||||
"strconv"
|
||||
|
||||
"git.faercol.me/faercol/polyculeconnect/polyculeconnect/cmd/utils"
|
||||
"git.faercol.me/faercol/polyculeconnect/polyculeconnect/config"
|
||||
|
@ -19,13 +20,40 @@ var migrateCmd = &cobra.Command{
|
|||
Long: `Run the database migrations.`,
|
||||
Run: func(cmd *cobra.Command, args []string) {
|
||||
conf := utils.InitConfig("")
|
||||
if err := runMigrations(conf); err != nil {
|
||||
up, nbSteps := parseArgs(args)
|
||||
if err := runMigrations(conf, up, nbSteps); err != nil {
|
||||
utils.Failf("Failed to run migrations: %s", err.Error())
|
||||
}
|
||||
},
|
||||
}
|
||||
|
||||
func runMigrations(conf *config.AppConfig) error {
|
||||
func parseArgs(args []string) (bool, int) {
|
||||
if len(args) == 0 {
|
||||
return true, 0
|
||||
}
|
||||
var actionUp bool
|
||||
switch args[0] {
|
||||
case "up":
|
||||
actionUp = true
|
||||
case "down":
|
||||
actionUp = false
|
||||
default:
|
||||
actionUp = true
|
||||
}
|
||||
|
||||
nbSteps := 0
|
||||
if len(args) > 1 {
|
||||
var err error
|
||||
nbSteps, err = strconv.Atoi(args[1])
|
||||
if err != nil {
|
||||
return actionUp, 0
|
||||
}
|
||||
}
|
||||
|
||||
return actionUp, nbSteps
|
||||
}
|
||||
|
||||
func runMigrations(conf *config.AppConfig, up bool, nbSteps int) error {
|
||||
storage, err := db.New(*conf)
|
||||
if err != nil {
|
||||
return fmt.Errorf("failed to connect to db: %w", err)
|
||||
|
@ -39,10 +67,18 @@ func runMigrations(conf *config.AppConfig) error {
|
|||
if err != nil {
|
||||
return fmt.Errorf("failed to init migrator: %w", err)
|
||||
}
|
||||
if err := m.Up(); err != nil {
|
||||
return fmt.Errorf("failed to run migrations: %w", err)
|
||||
|
||||
if nbSteps > 0 {
|
||||
multiplier := 1
|
||||
if !up {
|
||||
multiplier = -1
|
||||
}
|
||||
return m.Steps(multiplier * nbSteps)
|
||||
}
|
||||
return nil
|
||||
if up {
|
||||
return m.Up()
|
||||
}
|
||||
return m.Down()
|
||||
}
|
||||
|
||||
func init() {
|
||||
|
|
Binary file not shown.
Loading…
Reference in a new issue