feat(hatchery/marathon): allow to set uris for new applications (#5910)

ovh · Aug 31, 2021 · de71b22 · de71b22
1 parent bbae9de
commit de71b22
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/engine/hatchery/marathon/marathon.go b/engine/hatchery/marathon/marathon.go
@@ -320,6 +320,10 @@ func (h *HatcheryMarathon) SpawnWorker(ctx context.Context, spawnArgs hatchery.S
 		Labels:    &h.marathonLabels,
 	}
 
+	if h.Config.MarathonApplicationURIs != nil && len(h.Config.MarathonApplicationURIs) > 0 {
+		application.Uris = &h.Config.MarathonApplicationURIs
+	}
+
 	_, next := telemetry.Span(ctx, "marathonClient.CreateApplication")
 	if _, err := h.marathonClient.CreateApplication(application); err != nil {
 		next()

diff --git a/engine/hatchery/marathon/types.go b/engine/hatchery/marathon/types.go
@@ -37,6 +37,9 @@ type HatcheryConfiguration struct {
 
 	// WorkerSpawnTimeout Worker Timeout Spawning (seconds)
 	WorkerSpawnTimeout int `mapstructure:"workerSpawnTimeout" toml:"workerSpawnTimeout" default:"120" commented:"false" comment:"Worker Timeout Spawning (seconds)" json:"workerSpawnTimeout"`
+
+	// MarathonApplicationURIs will set "uris" value for each Application
+	MarathonApplicationURIs []string `mapstructure:"applicationURIs" toml:"applicationURIs" commented:"true" comment:"Use this option if you want to add uris on workers spawned by this hatchery." json:"-"`
 }
 
 // HatcheryMarathon implements HatcheryMode interface for mesos mode