Handle unknown migrations in command_status

ericklaus-wf · ericklaus-wf · commit e03d6789676d · 2016-10-25T15:52:43.000-05:00
diff --git a/sql-migrate/command_status.go b/sql-migrate/command_status.go
@@ -83,8 +83,13 @@ func (c *StatusCommand) Run(args []string) int {
 	}
 
 	for _, r := range records {
-		rows[r.Id].Migrated = true
-		rows[r.Id].AppliedAt = r.AppliedAt
+		status, ok := rows[r.Id]
+		if !ok {
+			ui.Warn(fmt.Sprintf("Migration in database %q unknown to sql-migrate; it will not be possible to migrate down this migration", r.Id))
+			continue
+		}
+		status.Migrated = true
+		status.AppliedAt = r.AppliedAt
 	}
 
 	for _, m := range migrations {

Original file line number	Diff line number	Diff line change
`@@ -83,8 +83,13 @@ func (c *StatusCommand) Run(args []string) int {`
`83`	`83`	`}`
`84`	`84`
`85`	`85`	`for _, r := range records {`
`86`		`- rows[r.Id].Migrated = true`
`87`		`- rows[r.Id].AppliedAt = r.AppliedAt`
	`86`	`+ status, ok := rows[r.Id]`
	`87`	`+ if !ok {`
	`88`	`+ ui.Warn(fmt.Sprintf("Migration in database %q unknown to sql-migrate; it will not be possible to migrate down this migration", r.Id))`
	`89`	`+ continue`
	`90`	`+ }`
	`91`	`+ status.Migrated = true`
	`92`	`+ status.AppliedAt = r.AppliedAt`
`88`	`93`	`}`
`89`	`94`
`90`	`95`	`for _, m := range migrations {`