Update

Author: realkarych

pkg/dbmate/db.go

@@ -741,8 +741,8 @@ func (db *DB) Rollback() error {
 	return nil
 }
 
-// MigrateOnly applies exactly one pending migration that matches the given version.
-func (db *DB) MigrateOnly(version string) error {
+// migrateOnly applies exactly one pending migration that matches the given version.
+func (db *DB) migrateOnly(version string) error {
 	drv, err := db.Driver()
 	if err != nil {
 		return err
@@ -813,8 +813,8 @@ func (db *DB) MigrateOnly(version string) error {
 	return nil
 }
 
-// RollbackOnly rolls back exactly one applied migration that matches the given version.
-func (db *DB) RollbackOnly(version string) error {
+// rollbackOnly rolls back exactly one applied migration that matches the given version.
+func (db *DB) rollbackOnly(version string) error {
 	drv, err := db.Driver()
 	if err != nil {
 		return err
@@ -878,6 +878,43 @@ func (db *DB) RollbackOnly(version string) error {
 	return nil
 }
 
+// MigrateTo applies every pending migration whose version ≤ target (inclusive).
+func (db *DB) MigrateTo(target string) error {
+	migs, err := db.FindMigrations()
+	if err != nil {
+		return err
+	}
+
+	for _, m := range migs {
+		if m.Applied || m.Version > target {
+			continue
+		}
+		if err := db.migrateOnly(m.Version); err != nil {
+			return err
+		}
+	}
+	return nil
+}
+
+// RollbackTo rolls back until version == target remains applied (exclusive).
+func (db *DB) RollbackTo(target string) error {
+	migs, err := db.FindMigrations()
+	if err != nil {
+		return err
+	}
+	// process latest-first
+	for i := len(migs) - 1; i >= 0; i-- {
+		m := migs[i]
+		if !m.Applied || m.Version == target {
+			continue
+		}
+		if err := db.rollbackOnly(m.Version); err != nil {
+			return err
+		}
+	}
+	return nil
+}
+
 // Status shows the status of all migrations
 func (db *DB) Status(quiet bool) (int, error) {
 	results, err := db.FindMigrations()

pkg/dbmate/db_test.go

@@ -593,7 +593,7 @@ func TestRollbackAll(t *testing.T) {
 	})
 }
 
-func TestMigrateOnly(t *testing.T) {
+func TestMigrateTo(t *testing.T) {
 	testEachURL(t, func(t *testing.T, u *url.URL) {
 		const v1 = "20151129054053" // users
 		const v2 = "20200227231541" // posts
@@ -605,7 +605,7 @@ func TestMigrateOnly(t *testing.T) {
 		require.NoError(t, db.Drop())
 		require.NoError(t, db.Create())
 
-		require.NoError(t, db.MigrateOnly(v1))
+		require.NoError(t, db.MigrateTo(v1))
 
 		sqlDB, err := drv.Open()
 		require.NoError(t, err)
@@ -619,20 +619,20 @@ func TestMigrateOnly(t *testing.T) {
 		require.NoError(t, sqlDB.QueryRow("select count(*) from users").Scan(&cnt))
 		require.Error(t, sqlDB.QueryRow("select count(*) from posts").Scan(&cnt))
 
-		require.NoError(t, db.MigrateOnly(v1))
+		require.NoError(t, db.MigrateTo(v1))
 
-		err = db.MigrateOnly("99999999999999")
+		err = db.MigrateTo("99999999999999")
 		require.ErrorIs(t, err, dbmate.ErrMigrationNotFound)
 
 		db.Strict = true
-		err = db.MigrateOnly(v2)
+		err = db.MigrateTo(v2)
 		require.NoError(t, err)
 
-		require.NoError(t, db.MigrateOnly(v1))
+		require.NoError(t, db.MigrateTo(v1))
 	})
 }
 
-func TestRollbackOnly(t *testing.T) {
+func TestRollbackTo(t *testing.T) {
 	testEachURL(t, func(t *testing.T, u *url.URL) {
 		const v1 = "20151129054053" // users
 		const v2 = "20200227231541" // posts
@@ -652,7 +652,7 @@ func TestRollbackOnly(t *testing.T) {
 		applied, _ := drv.SelectMigrations(sqlDB, -1)
 		require.Equal(t, map[string]bool{v1: true, v2: true}, applied)
 
-		require.NoError(t, db.RollbackOnly(v2))
+		require.NoError(t, db.RollbackTo(v2))
 
 		applied, _ = drv.SelectMigrations(sqlDB, -1)
 		require.Equal(t, map[string]bool{v1: true}, applied)
@@ -661,14 +661,14 @@ func TestRollbackOnly(t *testing.T) {
 		require.NoError(t, sqlDB.QueryRow("select count(*) from users").Scan(&cnt))
 		require.Error(t, sqlDB.QueryRow("select count(*) from posts").Scan(&cnt))
 
-		err = db.RollbackOnly(v2)
+		err = db.RollbackTo(v2)
 		require.Error(t, err)
 		require.Contains(t, err.Error(), "not applied")
 
-		err = db.RollbackOnly("99999999999999")
+		err = db.RollbackTo("99999999999999")
 		require.ErrorIs(t, err, dbmate.ErrMigrationNotFound)
 
-		require.NoError(t, db.RollbackOnly(v1))
+		require.NoError(t, db.RollbackTo(v1))
 		applied, _ = drv.SelectMigrations(sqlDB, -1)
 		require.Empty(t, applied)
 	})