Use TargetDSP on SCRIPT_RUN_ROLLBACK stage (#5215)

Signed-off-by: Yoshiki Fujikane <[email protected]>
pipe-cd · Sep 19, 2024 · 0e7a177 · 0e7a177
1 parent 9fcf50d
commit 0e7a177
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/pkg/app/piped/executor/kubernetes/rollback.go b/pkg/app/piped/executor/kubernetes/rollback.go
@@ -195,6 +195,13 @@ func (e *rollbackExecutor) ensureScriptRunRollback(ctx context.Context) model.St
 		return model.StageStatus_STAGE_SUCCESS
 	}
 
+	ds, err := e.TargetDSP.Get(ctx, e.LogPersister)
+	if err != nil {
+		e.LogPersister.Errorf("Failed to prepare target deploy source data (%v)", err)
+		return model.StageStatus_STAGE_FAILURE
+	}
+	e.appDir = ds.AppDir
+
 	envStr, ok := e.Stage.Metadata["env"]
 	env := make(map[string]string, 0)
 	if ok {