generationtux/brizo

package migrations

import (
    "log"

    "github.com/generationtux/brizo/database"
    "github.com/rubenv/sql-migrate"
)

// All gets all migrations to run
func All() []*migrate.Migration {
    return []*migrate.Migration{
        createUsersTable(),
        createApplicationsTable(),
        createEnvironmentsTable(),
        createVersionsTable(),
        createEnvironmentConfigsTable(),
        createAccessTokensTable(),
        addUUIDColumnToEnvironmentConfigsTable(),
        updateUUIDColumnForEnvironmentConfigsTable(),
        addUUIDColumnUniqueConstraintToEnvironmentConfigsTable(),
        removeVersionsToEnvironmentsConstraint(),
        addApplicationUUIDColumnToVersionsTable(),
        addApplicationAndVersionColumnsToEnvironmentsTable(),
        addApplicationAndEnvironmentColumnsToVersionsTable(),
        updateApplicationsUUIDInEnvironmentsTable(),
        updateVersionsIDInEnvironmentsTable(),
        updateVersionsUUIDInEnvironmentsTable(),
        updateApplicationIDInVersionsTable(),
        updateApplicationUUIDInVersionsTable(),
        updateEnvironmentUUIDInVersionsTable(),
        addRawArgumentsColumnToVersionsTable(),
    }
}

// Run execute migrations forward
func Run() error {
    conn, err := database.Connect()
    if err != nil {
        return err
    }

    m := &migrate.MemoryMigrationSource{Migrations: All()}
    n, err := migrate.Exec(conn.DB(), "mysql", m, migrate.Up)
    if err != nil {
        return err
    }

    log.Printf("%v migrations applied", n)

    return nil
}

// RunDown execute migrations backwards
func RunDown() error {
    conn, err := database.Connect()
    if err != nil {
        return err
    }

    m := &migrate.MemoryMigrationSource{Migrations: All()}
    n, err := migrate.Exec(conn.DB(), "mysql", m, migrate.Down)
    if err != nil {
        return err
    }

    log.Printf("%v migrations rolled back", n)

    return nil
}