Add label app:nvidia-dra-driver-gpu to all project components

Signed-off-by: Carlos Eduardo Arango Gutierrez <[email protected]>

Author: ArangoGutierrez

cmd/compute-domain-controller/controller.go

@@ -29,6 +29,9 @@ import (
 // It contains essential fields for driver identification, Kubernetes client access,
 // and work queue management.
 type ManagerConfig struct {
+	// chartName is the Helm chart name to use for the app label value
+	chartName string
+
 	// driverName is the unique identifier for this DRA driver
 	driverName string
 
@@ -63,6 +66,7 @@ func (c *Controller) Run(ctx context.Context) error {
 	workQueue := workqueue.New(workqueue.DefaultControllerRateLimiter())
 
 	managerConfig := &ManagerConfig{
+		chartName:       c.config.flags.chartName,
 		driverName:      c.config.driverName,
 		driverNamespace: c.config.flags.namespace,
 		imageName:       c.config.flags.imageName,

cmd/compute-domain-controller/daemonset.go

@@ -47,6 +47,7 @@ type DaemonSetTemplateData struct {
 	Finalizer                 string
 	ComputeDomainLabelKey     string
 	ComputeDomainLabelValue   types.UID
+	AppLabelValue             string
 	ResourceClaimTemplateName string
 	ImageName                 string
 }
@@ -180,6 +181,7 @@ func (m *DaemonSetManager) Create(ctx context.Context, namespace string, cd *nva
 		Finalizer:                 computeDomainFinalizer,
 		ComputeDomainLabelKey:     computeDomainLabelKey,
 		ComputeDomainLabelValue:   cd.UID,
+		AppLabelValue:             m.config.chartName,
 		ResourceClaimTemplateName: rct.Name,
 		ImageName:                 m.config.imageName,
 	}

cmd/compute-domain-controller/main.go

@@ -47,6 +47,7 @@ const (
 )
 
 type Flags struct {
+	chartName        string
 	kubeClientConfig flags.KubeClientConfig
 	loggingConfig    *flags.LoggingConfig
 
@@ -78,6 +79,14 @@ func newApp() *cli.App {
 		loggingConfig: flags.NewLoggingConfig(),
 	}
 	cliFlags := []cli.Flag{
+		&cli.StringFlag{
+			Name:        "chart-name",
+			Usage:       "The Helm chart name to use for the app label value.",
+			Required:    true,
+			Destination: &flags.chartName,
+			Value:       "nvidia-dra-driver-gpu",
+			EnvVars:     []string{"HELM_CHART_NAME"},
+		},
 		&cli.StringFlag{
 			Name:        "pod-name",
 			Usage:       "The name of the pod this controller is running in.",

cmd/compute-domain-controller/resourceclaimtemplate.go

@@ -49,6 +49,7 @@ type ResourceClaimTemplateTemplateData struct {
 	Finalizer               string
 	ComputeDomainLabelKey   string
 	ComputeDomainLabelValue types.UID
+	AppLabelValue           string
 	TargetLabelKey          string
 	TargetLabelValue        string
 	DeviceClassName         string
@@ -298,6 +299,7 @@ func (m *DaemonSetResourceClaimTemplateManager) Create(ctx context.Context, name
 		Finalizer:               computeDomainFinalizer,
 		ComputeDomainLabelKey:   computeDomainLabelKey,
 		ComputeDomainLabelValue: cd.UID,
+		AppLabelValue:           m.config.chartName,
 		TargetLabelKey:          computeDomainResourceClaimTemplateTargetLabelKey,
 		TargetLabelValue:        computeDomainResourceClaimTemplateTargetDaemon,
 		DeviceClassName:         computeDomainDaemonDeviceClass,
@@ -363,6 +365,7 @@ func (m *WorkloadResourceClaimTemplateManager) Create(ctx context.Context, names
 		DeviceClassName:         computeDomainDefaultChannelDeviceClass,
 		DriverName:              DriverName,
 		ChannelConfig:           channelConfig,
+		AppLabelValue:           m.config.chartName,
 	}
 
 	rct, err := m.BaseResourceClaimTemplateManager.Create(ctx, WorkloadResourceClaimTemplateTemplatePath, &templateData)

deployments/helm/nvidia-dra-driver-gpu/templates/controller.yaml

@@ -20,6 +20,7 @@ metadata:
   name: {{ include "nvidia-dra-driver-gpu.name" . }}-controller
   namespace: {{ include "nvidia-dra-driver-gpu.namespace" . }}
   labels:
+    app: {{ .Chart.Name }}
     {{- include "nvidia-dra-driver-gpu.labels" . | nindent 4 }}
 spec:
   replicas: 1
@@ -66,6 +67,8 @@ spec:
               fieldPath: metadata.namespace
         - name: IMAGE_NAME
           value: {{ include "nvidia-dra-driver-gpu.fullimage" . }}
+        - name: HELM_CHART_NAME
+          value: {{ .Chart.Name }}
       {{- with .Values.controller.nodeSelector }}
       nodeSelector:
         {{- toYaml . | nindent 8 }}

deployments/helm/nvidia-dra-driver-gpu/templates/kubeletplugin.yaml

@@ -20,6 +20,7 @@ metadata:
   name: {{ include "nvidia-dra-driver-gpu.name" . }}-kubelet-plugin
   namespace: {{ include "nvidia-dra-driver-gpu.namespace" . }}
   labels:
+    app: {{ .Chart.Name }}
     {{- include "nvidia-dra-driver-gpu.labels" . | nindent 4 }}
 spec:
   selector:

templates/compute-domain-daemon-claim-template.tmpl.yaml

@@ -8,6 +8,7 @@ metadata:
     - {{ .Finalizer }}
   labels:
     {{ .ComputeDomainLabelKey }}: {{ .ComputeDomainLabelValue }}
+    app: {{ .AppLabelValue }}
     {{ .TargetLabelKey }}: {{ .TargetLabelValue }}
 spec:
   spec:

templates/compute-domain-daemon.tmpl.yaml

@@ -8,6 +8,7 @@ metadata:
     - {{ .Finalizer }}
   labels:
     {{ .ComputeDomainLabelKey }}: {{ .ComputeDomainLabelValue }}
+    app: {{ .AppLabelValue }}
 spec:
   selector:
     matchLabels:

templates/compute-domain-workload-claim-template.tmpl.yaml

@@ -8,6 +8,7 @@ metadata:
     - {{ .Finalizer }}
   labels:
     {{ .ComputeDomainLabelKey }}: {{ .ComputeDomainLabelValue }}
+    app: {{ .AppLabelValue }}
     {{ .TargetLabelKey }}: {{ .TargetLabelValue }}
 spec:
   spec: