mirror of https://github.com/pressly/goose.git
94 lines
2.4 KiB
Go
94 lines
2.4 KiB
Go
package main
|
|
|
|
import (
|
|
"database/sql"
|
|
"io/ioutil"
|
|
"log"
|
|
"path/filepath"
|
|
"strings"
|
|
)
|
|
|
|
// Run a migration specified in raw SQL.
|
|
//
|
|
// Sections of the script can be annotated with a special comment,
|
|
// starting with "-- +goose" to specify whether the section should
|
|
// be applied during an Up or Down migration
|
|
//
|
|
// All statements following an Up or Down directive are grouped together
|
|
// until another direction directive is found.
|
|
func runSQLMigration(conf *DBConf, db *sql.DB, script string, v int64, direction bool) error {
|
|
|
|
txn, err := db.Begin()
|
|
if err != nil {
|
|
log.Fatal("db.Begin:", err)
|
|
}
|
|
|
|
f, err := ioutil.ReadFile(script)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
// track the count of each section
|
|
// so we can diagnose scripts with no annotations
|
|
upSections := 0
|
|
downSections := 0
|
|
|
|
// ensure we don't apply a query until we're sure it's going
|
|
// in the direction we're interested in
|
|
directionIsActive := false
|
|
|
|
// find each statement, checking annotations for up/down direction
|
|
// and execute each of them in the current transaction
|
|
stmts := strings.Split(string(f), ";")
|
|
|
|
for _, query := range stmts {
|
|
|
|
query = strings.TrimSpace(query)
|
|
|
|
if strings.HasPrefix(query, "-- +goose Up") {
|
|
directionIsActive = direction == true
|
|
upSections++
|
|
} else if strings.HasPrefix(query, "-- +goose Down") {
|
|
directionIsActive = direction == false
|
|
downSections++
|
|
}
|
|
|
|
if !directionIsActive || query == "" {
|
|
continue
|
|
}
|
|
|
|
if _, err = txn.Exec(query); err != nil {
|
|
txn.Rollback()
|
|
log.Fatalf("FAIL %s (%v), quitting migration.", filepath.Base(script), err)
|
|
return err
|
|
}
|
|
}
|
|
|
|
if upSections == 0 && downSections == 0 {
|
|
txn.Rollback()
|
|
log.Fatalf(`ERROR: no Up/Down annotations found in %s, so no statements were executed.
|
|
See https://bitbucket.org/liamstask/goose/overview for details.`,
|
|
filepath.Base(script))
|
|
}
|
|
|
|
if err = finalizeMigration(conf, txn, direction, v); err != nil {
|
|
log.Fatalf("error finalizing migration %s, quitting. (%v)", filepath.Base(script), err)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// Update the version table for the given migration,
|
|
// and finalize the transaction.
|
|
func finalizeMigration(conf *DBConf, txn *sql.Tx, direction bool, v int64) error {
|
|
|
|
// XXX: drop goose_db_version table on some minimum version number?
|
|
d := conf.Driver.Dialect
|
|
if _, err := txn.Exec(d.insertVersionSql(), v, direction); err != nil {
|
|
txn.Rollback()
|
|
return err
|
|
}
|
|
|
|
return txn.Commit()
|
|
}
|