72 lines
1.4 KiB
Go
72 lines
1.4 KiB
Go
package migrations
|
|
|
|
import (
|
|
"context"
|
|
|
|
"git.haelnorr.com/h/oslstats/internal/db"
|
|
"github.com/uptrace/bun"
|
|
)
|
|
|
|
func init() {
|
|
Migrations.MustRegister(
|
|
// UP migration
|
|
func(ctx context.Context, conn *bun.DB) error {
|
|
// Add is_forfeit column
|
|
_, err := conn.NewAddColumn().
|
|
Model((*db.PlayoffSeries)(nil)).
|
|
ColumnExpr("is_forfeit BOOLEAN NOT NULL DEFAULT false").
|
|
IfNotExists().
|
|
Exec(ctx)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// Add forfeit_team_id column
|
|
_, err = conn.NewAddColumn().
|
|
Model((*db.PlayoffSeries)(nil)).
|
|
ColumnExpr("forfeit_team_id INTEGER REFERENCES teams(id)").
|
|
IfNotExists().
|
|
Exec(ctx)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// Add forfeit_reason column
|
|
_, err = conn.NewAddColumn().
|
|
Model((*db.PlayoffSeries)(nil)).
|
|
ColumnExpr("forfeit_reason VARCHAR").
|
|
IfNotExists().
|
|
Exec(ctx)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
},
|
|
// DOWN migration
|
|
func(ctx context.Context, conn *bun.DB) error {
|
|
_, err := conn.NewDropColumn().
|
|
Model((*db.PlayoffSeries)(nil)).
|
|
ColumnExpr("forfeit_reason").
|
|
Exec(ctx)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
_, err = conn.NewDropColumn().
|
|
Model((*db.PlayoffSeries)(nil)).
|
|
ColumnExpr("forfeit_team_id").
|
|
Exec(ctx)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
_, err = conn.NewDropColumn().
|
|
Model((*db.PlayoffSeries)(nil)).
|
|
ColumnExpr("is_forfeit").
|
|
Exec(ctx)
|
|
return err
|
|
},
|
|
)
|
|
}
|