add owner tags for jobs and pods (tailwarden#1064)

* add cronJob support Signed-off-by: shubhindia <[email protected]> * drop fmt.Printf() Signed-off-by: shubhindia <[email protected]> * don't use separate cronJob collector Signed-off-by: shubhindia <[email protected]> * use owner_kind instead of owner Signed-off-by: shubhindia <[email protected]> * add owner tags for pods as well Signed-off-by: shubhindia <[email protected]> --------- Signed-off-by: shubhindia <[email protected]> Co-authored-by: Azanul Haque <[email protected]>
kavibritto · Oct 10, 2023 · 476b273 · 476b273
1 parent 5513ebe
commit 476b273
Show file tree

Hide file tree

Showing 2 changed files with 31 additions and 1 deletion.
diff --git a/providers/k8s/core/jobs.go b/providers/k8s/core/jobs.go
@@ -41,6 +41,21 @@ func Jobs(ctx context.Context, client providers.ProviderClient) ([]models.Resour
 				})
 			}
 
+			if len(job.OwnerReferences) > 0 {
+				// we use the owner kind of first owner only as the owner tag
+				ownerTags := []models.Tag{
+					{
+						Key:   "owner_kind",
+						Value: job.OwnerReferences[0].Kind,
+					},
+					{
+						Key:   "owner_name",
+						Value: job.OwnerReferences[0].Name,
+					},
+				}
+				tags = append(tags, ownerTags...)
+			}
+
 			cost := 0.0
 			if opencostEnabled {
 				cost = jobsCost[job.Name].TotalCost
@@ -70,7 +85,7 @@ func Jobs(ctx context.Context, client providers.ProviderClient) ([]models.Resour
 	log.WithFields(log.Fields{
 		"provider":  "Kubernetes",
 		"account":   client.Name,
-		"service":   "DaemonSet",
+		"service":   "Job",
 		"resources": len(resources),
 	}).Info("Fetched resources")
 	return resources, nil

diff --git a/providers/k8s/core/pods.go b/providers/k8s/core/pods.go
@@ -41,6 +41,21 @@ func Pods(ctx context.Context, client providers.ProviderClient) ([]models.Resour
 				})
 			}
 
+			if len(pod.OwnerReferences) > 0 {
+				// we use the owner kind of first owner only as the owner tag
+				ownerTags := []models.Tag{
+					{
+						Key:   "owner_kind",
+						Value: pod.OwnerReferences[0].Kind,
+					},
+					{
+						Key:   "owner_name",
+						Value: pod.OwnerReferences[0].Name,
+					},
+				}
+				tags = append(tags, ownerTags...)
+			}
+
 			cost := 0.0
 			if opencostEnabled {
 				cost = podsCost[pod.Name].TotalCost