add to example as well

2018-11-06 14:14:09 -05:00 · 2018-11-06 14:14:09 -05:00 · a4a53bdf99
parent 3c2c9d9076
commit a4a53bdf99
1 changed files with 8 additions and 0 deletions
--- a/examples/go-migrations/main.go
+++ b/examples/go-migrations/main.go
@ -33,6 +33,14 @@ func main() {
 		return
 	}

+	// TODO clean up arg/flag parsing flow
+	if args[0] == "fix" {
+		if err := goose.Run("fix", nil, *dir); err != nil {
+			log.Fatalf("goose run: %v", err)
+		}
+		return
+	}
+
 	if len(args) < 3 {
 		flags.Usage()
 		return