Update

Author: realkarych

pkg/dbmate/db.go

@@ -742,17 +742,12 @@ func (db *DB) Rollback() error {
 }
 
 // migrateOnly applies exactly one pending migration that matches the given version.
-func (db *DB) migrateOnly(version string) error {
+func (db *DB) migrateOnly(migrations []Migration, version string) error {
 	drv, err := db.Driver()
 	if err != nil {
 		return err
 	}
 
-	migrations, err := db.FindMigrations()
-	if err != nil {
-		return err
-	}
-
 	var target *Migration
 	highestApplied := ""
 	for i, m := range migrations {
@@ -814,17 +809,12 @@ func (db *DB) migrateOnly(version string) error {
 }
 
 // rollbackOnly rolls back exactly one applied migration that matches the given version.
-func (db *DB) rollbackOnly(version string) error {
+func (db *DB) rollbackOnly(migrations []Migration, version string) error {
 	drv, err := db.Driver()
 	if err != nil {
 		return err
 	}
 
-	migrations, err := db.FindMigrations()
-	if err != nil {
-		return err
-	}
-
 	var target *Migration
 	for i, m := range migrations {
 		if m.Version == version {
@@ -880,16 +870,16 @@ func (db *DB) rollbackOnly(version string) error {
 
 // MigrateTo applies every pending migration whose version ≤ target (inclusive).
 func (db *DB) MigrateTo(target string) error {
-	migs, err := db.FindMigrations()
+	migrations, err := db.FindMigrations()
 	if err != nil {
 		return err
 	}
 
-	for _, m := range migs {
+	for _, m := range migrations {
 		if m.Applied || m.Version > target {
 			continue
 		}
-		if err := db.migrateOnly(m.Version); err != nil {
+		if err := db.migrateOnly(migrations, m.Version); err != nil {
 			return err
 		}
 	}
@@ -898,17 +888,17 @@ func (db *DB) MigrateTo(target string) error {
 
 // RollbackTo rolls back until version == target remains applied (exclusive).
 func (db *DB) RollbackTo(target string) error {
-	migs, err := db.FindMigrations()
+	migrations, err := db.FindMigrations()
 	if err != nil {
 		return err
 	}
 	// process latest-first
-	for i := len(migs) - 1; i >= 0; i-- {
-		m := migs[i]
+	for i := len(migrations) - 1; i >= 0; i-- {
+		m := migrations[i]
 		if !m.Applied || m.Version == target {
 			continue
 		}
-		if err := db.rollbackOnly(m.Version); err != nil {
+		if err := db.rollbackOnly(migrations, m.Version); err != nil {
 			return err
 		}
 	}