Rollback operation on Start failure

xataio · Aug 31, 2023 · 3d3b69c · 3d3b69c
1 parent ac990e6
commit 3d3b69c
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/pkg/roll/execute.go b/pkg/roll/execute.go
@@ -2,6 +2,7 @@ package roll
 
 import (
 	"context"
+	"errors"
 	"fmt"
 	"strings"
 
@@ -40,7 +41,11 @@ func (m *Roll) Start(ctx context.Context, migration *migrations.Migration) error
 	for _, op := range migration.Operations {
 		err := op.Start(ctx, m.pgConn, m.state.Schema(), newSchema)
 		if err != nil {
-			return fmt.Errorf("unable to execute start operation: %w", err)
+			errRollback := m.Rollback(ctx)
+
+			return errors.Join(
+				fmt.Errorf("unable to execute start operation: %w", err),
+				errRollback)
 		}
 
 		if _, ok := op.(migrations.RequiresSchemaRefreshOperation); ok {