58 lines
1.1 KiB
Go
58 lines
1.1 KiB
Go
package migrations
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"git.handmade.network/hmn/hmn/src/migration/types"
|
|
"github.com/jackc/pgx/v4"
|
|
)
|
|
|
|
func init() {
|
|
registerMigration(AddTwitchEnded{})
|
|
}
|
|
|
|
type AddTwitchEnded struct{}
|
|
|
|
func (m AddTwitchEnded) Version() types.MigrationVersion {
|
|
return types.MigrationVersion(time.Date(2022, 10, 18, 6, 28, 31, 0, time.UTC))
|
|
}
|
|
|
|
func (m AddTwitchEnded) Name() string {
|
|
return "AddTwitchEnded"
|
|
}
|
|
|
|
func (m AddTwitchEnded) Description() string {
|
|
return "Add stream_ended to twitch history"
|
|
}
|
|
|
|
func (m AddTwitchEnded) Up(ctx context.Context, tx pgx.Tx) error {
|
|
_, err := tx.Exec(ctx,
|
|
`
|
|
ALTER TABLE twitch_stream_history
|
|
ADD COLUMN stream_ended BOOLEAN NOT NULL DEFAULT FALSE;
|
|
`,
|
|
)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
_, err = tx.Exec(ctx,
|
|
`
|
|
UPDATE twitch_stream_history
|
|
SET stream_ended = TRUE
|
|
WHERE ended_at > TIMESTAMP '2000-01-01 00:00:00';
|
|
`,
|
|
)
|
|
return err
|
|
}
|
|
|
|
func (m AddTwitchEnded) Down(ctx context.Context, tx pgx.Tx) error {
|
|
_, err := tx.Exec(ctx,
|
|
`
|
|
ALTER TABLE twitch_stream_history
|
|
DROP COLUMN stream_ended BOOLEAN NOT NULL DEFAULT FALSE;
|
|
`,
|
|
)
|
|
return err
|
|
}
|