68 lines
1.3 KiB
Go
68 lines
1.3 KiB
Go
package migrations
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/uptrace/bun"
|
|
|
|
"git.haelnorr.com/h/oslstats/internal/db"
|
|
)
|
|
|
|
func init() {
|
|
Migrations.MustRegister(
|
|
// UP migration
|
|
func(ctx context.Context, conn *bun.DB) error {
|
|
// Add slap_version column to seasons table
|
|
_, err := conn.NewAddColumn().
|
|
Model((*db.Season)(nil)).
|
|
ColumnExpr("slap_version VARCHAR NOT NULL DEFAULT 'rebound'").
|
|
IfNotExists().
|
|
Exec(ctx)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// Create leagues table
|
|
_, err = conn.NewCreateTable().
|
|
Model((*db.League)(nil)).
|
|
Exec(ctx)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// Create season_leagues join table
|
|
_, err = conn.NewCreateTable().
|
|
Model((*db.SeasonLeague)(nil)).
|
|
Exec(ctx)
|
|
return err
|
|
},
|
|
// DOWN migration
|
|
func(ctx context.Context, conn *bun.DB) error {
|
|
// Drop season_leagues join table first
|
|
_, err := conn.NewDropTable().
|
|
Model((*db.SeasonLeague)(nil)).
|
|
IfExists().
|
|
Exec(ctx)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// Drop leagues table
|
|
_, err = conn.NewDropTable().
|
|
Model((*db.League)(nil)).
|
|
IfExists().
|
|
Exec(ctx)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// Remove slap_version column from seasons table
|
|
_, err = conn.NewDropColumn().
|
|
Model((*db.Season)(nil)).
|
|
ColumnExpr("slap_version").
|
|
Exec(ctx)
|
|
return err
|
|
},
|
|
)
|
|
}
|