Allow haproxy template to list all app env values

services/template/template.go

@@ -2,6 +2,7 @@ package template
 
 import (
 	"bytes"
+	"github.com/QubitProducts/bamboo/services/marathon"
 	"github.com/QubitProducts/bamboo/services/service"
 	"strings"
 	"text/template"
@@ -17,19 +18,42 @@ func getService(data map[string]service.Service, appId string) service.Service {
 	return serviceModel
 }
 
+func getAppEnvValues(envVar string, appList marathon.AppList) (envValues []string) {
+	for _, app := range appList {
+		value, ok := app.Env[envVar]
+		if ok {
+			envValues = append(envValues, value)
+		}
+	}
+	return unique(envValues)
+}
+
+func unique(strings []string) (uniqueStrings []string) {
+	seen := make(map[string]bool)
+	for _, s := range strings {
+		if !seen[s] {
+			uniqueStrings = append(uniqueStrings, s)
+			seen[s] = true
+		}
+	}
+	return uniqueStrings
+}
+
 /*
 	Returns string content of a rendered template
 */
 func RenderTemplate(templateName string, templateContent string, data interface{}) (string, error) {
 	funcMap := template.FuncMap{
-		"hasKey":     hasKey,
-		"getService": getService,
-		"Split":      strings.Split,
-		"Contains":   strings.Contains,
-		"Join":       strings.Join,
-		"Replace":    strings.Replace,
-		"ToUpper":    strings.ToUpper,
-		"ToLower":    strings.ToLower}
+		"hasKey":          hasKey,
+		"getService":      getService,
+		"getAppEnvValues": getAppEnvValues,
+		"Split":           strings.Split,
+		"Contains":        strings.Contains,
+		"Join":            strings.Join,
+		"Replace":         strings.Replace,
+		"ToUpper":         strings.ToUpper,
+		"ToLower":         strings.ToLower,
+		"unique":          unique}
 
 	tpl := template.Must(template.New(templateName).Funcs(funcMap).Parse(templateContent))
 

services/template/template_test.go

@@ -1,6 +1,7 @@
 package template
 
 import (
+	"github.com/QubitProducts/bamboo/services/marathon"
 	. "github.com/smartystreets/goconvey/convey"
 	"testing"
 )
@@ -18,6 +19,60 @@ func TestTemplateWriter(t *testing.T) {
 	})
 }
 
+type TemplateData struct {
+	Apps marathon.AppList
+}
+
+func TestTemplateGetAppEnvValuesFunction(t *testing.T) {
+	Convey("#RenderTemplate", t, func() {
+		templateName := "templateName"
+		apps := marathon.AppList{}
+		apps = append(apps, marathon.App{
+			Id: "app1",
+			Env: map[string]string{
+				"BAMBOO_TCP_PORT": "10001",
+			},
+		})
+		apps = append(apps, marathon.App{
+			Id: "app2",
+			Env: map[string]string{
+				"BAMBOO_TCP_PORT": "10001",
+			},
+		})
+		apps = append(apps, marathon.App{
+			Id: "app3",
+			Env: map[string]string{
+				"BAMBOO_TCP_PORT": "10002",
+			},
+		})
+		apps = append(apps, marathon.App{
+			Id: "app4",
+			Env: map[string]string{
+				"BAMBOO_TCP_PORT": "10001",
+			},
+		})
+		apps = append(apps, marathon.App{
+			Id: "app5",
+			Env: map[string]string{
+				"BAMBOO_TCP_PORT": "10003",
+			},
+		})
+		apps = append(apps, marathon.App{
+			Id: "app6",
+			Env: map[string]string{
+				"BAMBOO_TCP_PORT": "10002",
+			},
+		})
+
+		templateData := TemplateData{Apps: apps}
+		templateContent := "{{ range $port := .Apps | getAppEnvValues \"BAMBOO_TCP_PORT\" }}{{ $port }} {{ end }}"
+		Convey("should list the three unique ports", func() {
+			content, _ := RenderTemplate(templateName, templateContent, templateData)
+			So(content, ShouldEqual, "10001 10002 10003 ")
+		})
+	})
+}
+
 func TestTemplateSplitFunction(t *testing.T) {
 	Convey("#RenderTemplate", t, func() {
 		templateName := "templateName"
@@ -31,6 +86,20 @@ func TestTemplateSplitFunction(t *testing.T) {
 	})
 }
 
+func TestTemplateUniqueFunction(t *testing.T) {
+	Convey("#RenderTemplate", t, func() {
+		templateName := "templateName"
+		domains := []string{"example.com", "example.com", "example.org", "example.com", "example.org", "example.net"}
+		params := map[string][]string{"domains": domains}
+
+		templateContent := "{{ range unique .domains }}{{ . }} {{ end }}"
+		Convey("should create a list of three distinct domains", func() {
+			content, _ := RenderTemplate(templateName, templateContent, params)
+			So(content, ShouldEqual, "example.com example.org example.net ")
+		})
+	})
+}
+
 func TestTemplateContainsFunction(t *testing.T) {
 	Convey("#RenderTemplate", t, func() {
 		templateName := "templateName"