Add team_owner for PrometheusDiscover

tsuru · Feb 10, 2023 · 07b3b76 · 07b3b76
1 parent dce71a0
commit 07b3b76
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/pkg/controllerapi/controllerapi.go b/pkg/controllerapi/controllerapi.go
@@ -61,6 +61,7 @@ func rpaasTargetGroups(svcMap map[sigsk8sclient.ObjectKey]*coreV1.Service, nginx
 
 		serviceInstance := svc.Labels["rpaas.extensions.tsuru.io/instance-name"]
 		service := svc.Labels["rpaas.extensions.tsuru.io/service-name"]
+		teamOwner := svc.Labels["rpaas.extensions.tsuru.io/team-owner"]
 
 		targetGroups = append(targetGroups, TargetGroup{
 			Targets: []string{
@@ -69,6 +70,7 @@ func rpaasTargetGroups(svcMap map[sigsk8sclient.ObjectKey]*coreV1.Service, nginx
 			Labels: map[string]string{
 				"service_instance": serviceInstance,
 				"service":          service,
+				"team_owner":       teamOwner,
 			},
 		})
 
@@ -82,6 +84,7 @@ func rpaasTargetGroups(svcMap map[sigsk8sclient.ObjectKey]*coreV1.Service, nginx
 						"service_instance": serviceInstance,
 						"service":          service,
 						"servername":       host,
+						"team_owner":       teamOwner,
 					},
 				})
 			}

diff --git a/pkg/controllerapi/controllerapi_test.go b/pkg/controllerapi/controllerapi_test.go
@@ -51,6 +51,7 @@ func TestPrometheusDiscover(t *testing.T) {
 				"nginx.tsuru.io/app":                      "nginx",
 				"rpaas.extensions.tsuru.io/instance-name": "test",
 				"rpaas.extensions.tsuru.io/service-name":  "rpaasv2",
+				"rpaas.extensions.tsuru.io/team-owner":    "team01",
 			},
 		},
 		Status: coreV1.ServiceStatus{
@@ -84,6 +85,7 @@ func TestPrometheusDiscover(t *testing.T) {
 			Labels: map[string]string{
 				"service":          "rpaasv2",
 				"service_instance": "test",
+				"team_owner":       "team01",
 			},
 		},
 		{
@@ -92,6 +94,7 @@ func TestPrometheusDiscover(t *testing.T) {
 				"servername":       "test.internal",
 				"service":          "rpaasv2",
 				"service_instance": "test",
+				"team_owner":       "team01",
 			},
 		},
 		{
@@ -100,6 +103,7 @@ func TestPrometheusDiscover(t *testing.T) {
 				"servername":       "hello.globo",
 				"service":          "rpaasv2",
 				"service_instance": "test",
+				"team_owner":       "team01",
 			},
 		},
 	}, l)