bt: Add retries to stack component config

DavidGamba · Oct 2, 2024 · 85fdd31 · 85fdd31
1 parent 12c9852
commit 85fdd31
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 1 deletion.
diff --git a/bt/stack/config/schema.cue b/bt/stack/config/schema.cue
@@ -13,6 +13,7 @@ package bt_stacks
 	depends_on: [...#Component.id]
 	variables: [...#Variable]
 	workspaces: [...string]
+	retries: int | *0
 }
 
 #Stack: {

diff --git a/bt/stack/config/schema.go b/bt/stack/config/schema.go
@@ -15,6 +15,7 @@ type Component struct {
 	DependsOn  []string   `json:"depends_on"`
 	Variables  []Variable `json:"variables"`
 	Workspaces []string   `json:"workspaces"`
+	Retries    int        `json:"retries"`
 }
 
 type ID string
@@ -30,7 +31,7 @@ type Stack struct {
 }
 
 func (c Component) String() string {
-	return fmt.Sprintf("id: %s, path: %s, depends_on: %v, variables: %v, workspaces: %v", c.ID, c.Path, c.DependsOn, c.Variables, c.Workspaces)
+	return fmt.Sprintf("id: %s, path: %s, depends_on: %v, variables: %v, workspaces: %v, retries: %d", c.ID, c.Path, c.DependsOn, c.Variables, c.Workspaces, c.Retries)
 }
 
 func (s Stack) String() string {

diff --git a/bt/stack/dag.go b/bt/stack/dag.go
@@ -84,12 +84,18 @@ func generateDAG(opt *getoptions.GetOpt, id string, cfg *sconfig.Config, normal
 				} else {
 					g.TaskDependsOn(tm.Get(wID), tm.Get(cID))
 				}
+				if c.Retries > 0 {
+					g.TaskRetries(tm.Get(wID), c.Retries)
+				}
 			}
 		} else {
 			// normal mode
 			tm.Add(cID, wsFn(cID, c.Path, "", variables))
 			Logger.Printf("adding task %s on %s vars: %v\n", cID, c.Path, variables)
 			g.AddTask(tm.Get(cID))
+			if c.Retries > 0 {
+				g.TaskRetries(tm.Get(cID), c.Retries)
+			}
 		}
 	}