add priority labeling

Author: w0ng952

kube/config/templates/5xx-rate.tmpl

@@ -45,3 +45,6 @@ spec:
         {{if .Sensitivity}}
         sensitivity: {{.Sensitivity}}
         {{end}}
+        {{if .Priority}}
+        priority: {{.Priority}}
+        {{end}}

kube/config/templates/replicas-availability-deployment.tmpl

@@ -37,3 +37,6 @@ spec:
         {{if .Sensitivity}}
         sensitivity: {{.Sensitivity}}
         {{end}}
+        {{if .Priority}}
+        priority: {{.Priority}}
+        {{end}}

pkg/templates/ingress.go

@@ -29,6 +29,7 @@ type templateParameterIngress struct {
 	Criticality    string
 	Sensitivity    string
 	BackendService string
+	Priority       string
 }
 
 // CreateFromIngress
@@ -46,6 +47,7 @@ func (a *PrometheusRuleTemplateManager) CreateFromIngress(ingress *networkingv1.
 		Environment: ingress.GetAnnotations()[environmentAnnotation],
 		Criticality: ingress.GetAnnotations()[criticalityAnnotation],
 		Sensitivity: ingress.GetAnnotations()[sensitivityAnnotation],
+		Priority:    ingress.GetAnnotations()[priorityAnnotation],
 	}
 
 	prometheusRules := map[string]*monitoringv1.PrometheusRule{}

pkg/templates/ingress_test.go

@@ -69,11 +69,12 @@ var (
 			Name:      "testDefaultBackend",
 			Namespace: "testNamespace",
 			Annotations: map[string]string{
-				"com.uswitch.heimdall/5xx-rate": "0.001",
-				ownerAnnotation:                 "testIngressOwner",
-				environmentAnnotation:           "testing",
-				criticalityAnnotation:           "low",
-				sensitivityAnnotation:           "public",
+				"com.uswitch.heimdall/5xx-rate":       "0.001",
+				"com.uswitch.heimdall/label-priority": "p1",
+				ownerAnnotation:                       "testIngressOwner",
+				environmentAnnotation:                 "testing",
+				criticalityAnnotation:                 "low",
+				sensitivityAnnotation:                 "public",
 			},
 		},
 		Spec: networkingv1.IngressSpec{
@@ -147,6 +148,10 @@ func TestIngressAnnotationsDefaultBackend(t *testing.T) {
 	assert.Assert(t, is.Len(promrules, 1))
 	assert.Equal(t, promrules[0].Spec.Groups[0].Rules[0].Expr.StrVal, expr)
 	assert.Equal(t, promrules[0].Spec.Groups[0].Rules[0].Labels["owner"], "testIngressOwner")
+	assert.Equal(t, promrules[0].Spec.Groups[0].Rules[0].Labels["environment"], "testing")
+	assert.Equal(t, promrules[0].Spec.Groups[0].Rules[0].Labels["criticality"], "low")
+	assert.Equal(t, promrules[0].Spec.Groups[0].Rules[0].Labels["sensitivity"], "public")
+	assert.Equal(t, promrules[0].Spec.Groups[0].Rules[0].Labels["priority"], "p1")
 }
 
 func TestIngressAnnotationsRuleBackend(t *testing.T) {

pkg/templates/promrules.go

@@ -21,6 +21,7 @@ const (
 	environmentAnnotation = "service.rvu.co.uk/environment"
 	criticalityAnnotation = "service.rvu.co.uk/criticality"
 	sensitivityAnnotation = "service.rvu.co.uk/sensitivity"
+	priorityAnnotation    = "com.uswitch.heimdall/label-priority"
 )
 
 // ClientSetI