58 lines
1.1 KiB
Go
58 lines
1.1 KiB
Go
|
package migrations
|
||
|
|
|
||
|
|
import (
|
||
|
|
"database/sql"
|
||
|
|
"flag"
|
||
|
|
"os"
|
||
|
|
|
||
|
|
"github.com/pressly/goose"
|
||
|
|
log "github.com/sirupsen/logrus"
|
||
|
|
|
||
|
|
_ "github.com/go-sql-driver/mysql"
|
||
|
|
)
|
||
|
|
|
||
|
|
var (
|
||
|
|
flags = flag.NewFlagSet("goose", flag.ExitOnError)
|
||
|
|
dir = flags.String("dir", ".", "directory with migration files")
|
||
|
|
)
|
||
|
|
|
||
|
|
func init() {
|
||
|
|
_ = flags.Parse(os.Args[1:])
|
||
|
|
args := flags.Args()
|
||
|
|
|
||
|
|
if len(args) < 1 {
|
||
|
|
flags.Usage()
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
command := args[0]
|
||
|
|
|
||
|
|
var dbUrl string
|
||
|
|
var exists bool
|
||
|
|
if dbUrl, exists = os.LookupEnv("DB_URL"); !exists {
|
||
|
|
log.Fatalf("define database URL in environment variable DB_URL")
|
||
|
|
}
|
||
|
|
db, err := sql.Open("mysql", dbUrl)
|
||
|
|
if err != nil {
|
||
|
|
log.Fatalf("goose: failt to open DB: %v\n", err)
|
||
|
|
}
|
||
|
|
if err = goose.SetDialect("mysql"); err != nil {
|
||
|
|
log.Fatalf("failed to set dialect: %v\n", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
defer func() {
|
||
|
|
if err := db.Close(); err != nil {
|
||
|
|
log.Fatalf("goose: failed to close DB: %v\n", err)
|
||
|
|
}
|
||
|
|
}()
|
||
|
|
|
||
|
|
arguments := make([]string, 0)
|
||
|
|
if len(args) > 3 {
|
||
|
|
arguments = append(arguments, args[3:]...)
|
||
|
|
}
|
||
|
|
|
||
|
|
if err := goose.Run(command, db, *dir, arguments...); err != nil {
|
||
|
|
log.Fatalf("goose %v: %v", command, err)
|
||
|
|
}
|
||
|
|
}
|