Add new function to get only external repos urls

openshift · Aug 3, 2023 · 4af787f · 4af787f
1 parent 5c08ac5
commit 4af787f
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/pkg/gatherers/workloads/gather_workloads_info.go b/pkg/gatherers/workloads/gather_workloads_info.go
@@ -60,7 +60,7 @@ const (
 // None
 //
 // ### Changes
-// None
+// - Image repository is now collected if it comes from outside the Red Hat domain
 func (g *Gatherer) GatherWorkloadInfo(ctx context.Context) ([]record.Record, []error) {
 	gatherKubeClient, err := kubernetes.NewForConfig(g.gatherProtoKubeConfig)
 	if err != nil {
@@ -509,8 +509,11 @@ func calculateWorkloadInfo(h hash.Hash, image *imagev1.Image) workloadImage {
 	for _, layer := range image.DockerImageLayers {
 		layers = append(layers, layer.Name)
 	}
+
 	info := workloadImage{
 		LayerIDs: layers,
+		//RepoURL: "quay.io/openshift-release-dev/ocp-v4.0-art-dev@sha256:5710554c08735126986b7c553cdb9a31bf97071c7adceda20f7aa116f35e867f",
+		Repository: getExternalImageRepo(image.DockerImageReference),
 	}
 
 	if err := imageutil.ImageWithMetadata(image); err != nil {
@@ -587,3 +590,11 @@ func workloadImageAdd(imageID string, image workloadImage) {
 	defer workloadSizeLock.Unlock()
 	workloadImageLRU.Add(imageID, image)
 }
+
+// getExternalImageRepo returns image URLs if they are not from Red Hat domain
+func getExternalImageRepo(s string) (repo string) {
+	if !strings.Contains(s, "redhat") {
+		repo = s
+	}
+	return
+}