50 lines
935 B
Go
50 lines
935 B
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, dbConn *bun.DB) error {
|
|
// Add your migration code here
|
|
_, err := dbConn.NewCreateTable().
|
|
Model((*db.Team)(nil)).
|
|
Exec(ctx)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
_, err = dbConn.NewCreateTable().
|
|
Model((*db.TeamParticipation)(nil)).
|
|
Exec(ctx)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
},
|
|
// DOWN migration
|
|
func(ctx context.Context, dbConn *bun.DB) error {
|
|
// Add your rollback code here
|
|
_, err := dbConn.NewDropTable().
|
|
Model((*db.TeamParticipation)(nil)).
|
|
IfExists().
|
|
Exec(ctx)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
_, err = dbConn.NewDropTable().
|
|
Model((*db.Team)(nil)).
|
|
IfExists().
|
|
Exec(ctx)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
},
|
|
)
|
|
}
|