dir string before extra args

pull/5/head
Josh Fyne 2016-06-29 14:54:13 -04:00
parent ba86be6db6
commit db38a98723
2 changed files with 4 additions and 7 deletions

View File

@ -57,12 +57,12 @@ func main() {
log.Fatalf("-dbstring=%q: %v\n", dbstring, err)
}
arguments := []string{*dir}
arguments := []string{}
if len(args) > 3 {
arguments = append(arguments, args[3:]...)
}
if err := goose.Run(command, db, arguments...); err != nil {
if err := goose.Run(command, db, *dir, arguments...); err != nil {
log.Fatalf("goose run: %v", err)
}
}

View File

@ -5,10 +5,7 @@ import (
"fmt"
)
func Run(command string, db *sql.DB, args ...string) error {
dir := args[0]
additional := args[0:]
func Run(command string, db *sql.DB, dir string, args ...string) error {
switch command {
case "up":
if err := Up(db, dir); err != nil {
@ -19,7 +16,7 @@ func Run(command string, db *sql.DB, args ...string) error {
return err
}
case "create":
if err := Create(db, additional[0], additional[1], dir); err != nil {
if err := Create(db, args[0], args[1], dir); err != nil {
return err
}
case "down":