62 lines
1.6 KiB
Go
62 lines
1.6 KiB
Go
package mysql
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"errors"
|
|
"fmt"
|
|
|
|
"github.com/golang-migrate/migrate/v4"
|
|
migratemysql "github.com/golang-migrate/migrate/v4/database/mysql"
|
|
"github.com/golang-migrate/migrate/v4/source/iofs"
|
|
|
|
"overnight-trading-bot/internal/repository/migrations"
|
|
)
|
|
|
|
func ApplyMigrations(ctx context.Context, db *sql.DB) error {
|
|
if err := ctx.Err(); err != nil {
|
|
return err
|
|
}
|
|
driver, err := migratemysql.WithInstance(db, &migratemysql.Config{})
|
|
if err != nil {
|
|
return fmt.Errorf("create mysql migration driver: %w", err)
|
|
}
|
|
source, err := iofs.New(migrations.FS, ".")
|
|
if err != nil {
|
|
return fmt.Errorf("create iofs migration source: %w", err)
|
|
}
|
|
m, err := migrate.NewWithInstance("iofs", source, "mysql", driver)
|
|
if err != nil {
|
|
return fmt.Errorf("create migrate instance: %w", err)
|
|
}
|
|
defer func() {
|
|
_, _ = m.Close()
|
|
}()
|
|
if err := m.Up(); err != nil && !errors.Is(err, migrate.ErrNoChange) {
|
|
return fmt.Errorf("apply migrations: %w", err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func RollbackAll(db *sql.DB) error {
|
|
driver, err := migratemysql.WithInstance(db, &migratemysql.Config{})
|
|
if err != nil {
|
|
return fmt.Errorf("create mysql migration driver: %w", err)
|
|
}
|
|
source, err := iofs.New(migrations.FS, ".")
|
|
if err != nil {
|
|
return fmt.Errorf("create iofs migration source: %w", err)
|
|
}
|
|
m, err := migrate.NewWithInstance("iofs", source, "mysql", driver)
|
|
if err != nil {
|
|
return fmt.Errorf("create migrate instance: %w", err)
|
|
}
|
|
defer func() {
|
|
_, _ = m.Close()
|
|
}()
|
|
if err := m.Down(); err != nil && !errors.Is(err, migrate.ErrNoChange) {
|
|
return fmt.Errorf("rollback migrations: %w", err)
|
|
}
|
|
return nil
|
|
}
|