mirror of
https://github.com/c9s/bbgo.git
synced 2024-09-22 17:21:10 +00:00
45 lines
925 B
Go
45 lines
925 B
Go
|
package migrations
|
||
|
|
||
|
import (
|
||
|
"database/sql"
|
||
|
"context"
|
||
|
|
||
|
"github.com/c9s/rockhopper"
|
||
|
)
|
||
|
|
||
|
func init() {
|
||
|
rockhopper.AddMigration(upOrdersAddIndex, downOrdersAddIndex)
|
||
|
}
|
||
|
|
||
|
func upOrdersAddIndex(ctx context.Context, tx *sql.Tx) (err error) {
|
||
|
// This code is executed when the migration is applied.
|
||
|
|
||
|
_, err = tx.ExecContext(ctx, "CREATE INDEX orders_symbol ON orders (exchange, symbol);")
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
_, err = tx.ExecContext(ctx, "CREATE UNIQUE INDEX orders_order_id ON orders (order_id, exchange);")
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
func downOrdersAddIndex(ctx context.Context, tx *sql.Tx) (err error) {
|
||
|
// This code is executed when the migration is rolled back.
|
||
|
|
||
|
_, err = tx.ExecContext(ctx, "DROP INDEX orders_symbol ON orders;")
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
_, err = tx.ExecContext(ctx, "DROP INDEX orders_order_id ON orders;")
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
return err
|
||
|
}
|