diff --git a/cli/app/rollback.go b/cli/app/rollback.go index cadd4d47..941b2fcd 100644 --- a/cli/app/rollback.go +++ b/cli/app/rollback.go @@ -180,6 +180,7 @@ recipes. } config.ExposeAllEnv(stackName, compose, app.Env) config.SetRecipeLabel(compose, stackName, app.Recipe) + config.SetChaosLabel(compose, stackName, internal.Chaos) if !internal.Force { if err := internal.NewVersionOverview(app, deployedVersion, chosenDowngrade, ""); err != nil { diff --git a/cli/app/upgrade.go b/cli/app/upgrade.go index 5037a490..a2138f27 100644 --- a/cli/app/upgrade.go +++ b/cli/app/upgrade.go @@ -191,6 +191,7 @@ recipes. } config.ExposeAllEnv(stackName, compose, app.Env) config.SetRecipeLabel(compose, stackName, app.Recipe) + config.SetChaosLabel(compose, stackName, internal.Chaos) if err := internal.NewVersionOverview(app, deployedVersion, chosenUpgrade, releaseNotes); err != nil { logrus.Fatal(err) diff --git a/cli/internal/deploy.go b/cli/internal/deploy.go index 68a007e5..3ae9d8e0 100644 --- a/cli/internal/deploy.go +++ b/cli/internal/deploy.go @@ -136,6 +136,7 @@ func DeployAction(c *cli.Context) error { } config.ExposeAllEnv(app.StackName(), compose, app.Env) config.SetRecipeLabel(compose, app.StackName(), app.Recipe) + config.SetChaosLabel(compose, app.StackName(), Chaos) if err := DeployOverview(app, version, "continue with deployment?"); err != nil { logrus.Fatal(err) diff --git a/pkg/config/app.go b/pkg/config/app.go index bb06e10d..f7d338f1 100644 --- a/pkg/config/app.go +++ b/pkg/config/app.go @@ -5,6 +5,7 @@ import ( "io/ioutil" "os" "path" + "strconv" "strings" "github.com/schollz/progressbar/v3" @@ -476,3 +477,14 @@ func SetRecipeLabel(compose *composetypes.Config, stackName string, recipe strin } } } + +// add a label to signal that a deploy is a chaos deploy +func SetChaosLabel(compose *composetypes.Config, stackName string, chaos bool) { + for _, service := range compose.Services { + if service.Name == "app" { + logrus.Debugf("set chaos label to %s", chaos) + labelKey := fmt.Sprintf("coop-cloud.%s.chaos", stackName) + service.Deploy.Labels[labelKey] = strconv.FormatBool(chaos) + } + } +}