feat(migrations): Adding steps count for the migrations (#25)

Adding steps count for the migrations

cmd/schema/cmd_create.go

@@ -56,7 +56,7 @@ func (c *createCmd) Execute(_ context.Context, _ *flag.FlagSet, _ ...interface{}
 	// The timestamp is the current time in the format YYYYMMDDHHMMSS
 	// The name is the name of the migration with spaces as underscores
 
-	now := time.Now()
+	now := time.Now().UTC()
 	name := fmt.Sprintf("%s_%s", now.Format(migrations.FilePrefix), strings.TrimSpace(c.name))
 	name = strings.ReplaceAll(name, " ", "_")
 

cmd/schema/cmd_migrate.go

@@ -21,6 +21,9 @@ type migrateCmd struct {
 
 	// migrationLocation is where the migrations are located.
 	migrationLocation string
+
+	// steps is the number of steps to migrate.
+	steps int
 }
 
 func (m *migrateCmd) Name() string {
@@ -41,6 +44,7 @@ func (m *migrateCmd) SetFlags(f *flag.FlagSet) {
 	f.BoolVar(&m.up, "up", false, "Migrate up.")
 	f.BoolVar(&m.down, "down", false, "Migrate down.")
 	f.StringVar(&m.migrationLocation, "loc", "./migrations", "The location of the migrations.")
+	f.IntVar(&m.steps, "steps", 0, "The number of steps to migrate (0 means all).")
 }
 
 func (m *migrateCmd) Execute(ctx context.Context, _ *flag.FlagSet, _ ...interface{}) subcommands.ExitStatus {
@@ -71,12 +75,12 @@ func (m *migrateCmd) Execute(ctx context.Context, _ *flag.FlagSet, _ ...interfac
 
 	switch {
 	case m.up:
-		if err := migrations.NewVersioning(db, absPath).MigrateUp(); err != nil {
+		if err := migrations.NewVersioning(db, absPath, m.steps).MigrateUp(); err != nil {
 			slog.Error("Error migrating up", slog.String("error", err.Error()))
 			return subcommands.ExitFailure
 		}
 	case m.down:
-		if err := migrations.NewVersioning(db, absPath).MigrateDown(); err != nil {
+		if err := migrations.NewVersioning(db, absPath, m.steps).MigrateDown(); err != nil {
 			slog.Error("Error migrating down", slog.String("error", err.Error()))
 			return subcommands.ExitFailure
 		}

example/database/migrations/20241019214408_testing_two.down.sql

@@ -0,0 +1 @@
+ALTER TABLE test DROP COLUMN age;

example/database/migrations/20241019214408_testing_two.up.sql

@@ -0,0 +1 @@
+ALTER TABLE test ADD COLUMN age INT;

pkg/migrations/down.go

@@ -42,6 +42,7 @@ func (v *versioning) MigrateDown() error {
 	}
 
 	// Migrate down.
+	count := 0
 	for _, f := range orderedFiles {
 		slog.Debug(fmt.Sprintf("Migrating up: %s", f.Name()))
 
@@ -71,6 +72,11 @@ func (v *versioning) MigrateDown() error {
 			continue
 		}
 
+		if v.steps > 0 && count == v.steps {
+			break
+		}
+		count++
+
 		// Migrate down.
 		if err := v.migrate(f, down); err != nil {
 			return fmt.Errorf("error migrating down: %w", err)

pkg/migrations/up.go

@@ -46,6 +46,7 @@ func (v *versioning) MigrateUp() error {
 	}
 
 	// Migrate up.
+	count := 0
 	for _, f := range orderedFiles {
 		slog.Debug(fmt.Sprintf("Migrating up: %s", f.Name()))
 
@@ -67,11 +68,16 @@ func (v *versioning) MigrateUp() error {
 				return fmt.Errorf("error parsing current version: %w", err)
 			}
 
-			if parsed.Before(currentParsed) {
+			if parsed.Before(currentParsed) || parsed.Equal(currentParsed) {
 				continue
 			}
 		}
 
+		if v.steps > 0 && count == v.steps {
+			break
+		}
+		count++
+
 		// Migrate up.
 		if err := v.migrate(f, up); err != nil {
 			return fmt.Errorf("error migrating up: %w", err)

pkg/migrations/versioning.go

@@ -37,12 +37,16 @@ type versioning struct {
 
 	// migrationLocation is the location of the migrations.
 	migrationLocation string
+
+	// steps is the number of steps to migrate.
+	steps int
 }
 
-func NewVersioning(db *sqlx.DB, migrationLocation string) Versioning {
+func NewVersioning(db *sqlx.DB, migrationLocation string, steps int) Versioning {
 	return &versioning{
 		db:                db,
 		migrationLocation: migrationLocation,
+		steps:             steps,
 	}
 }