colocationprofile: support mutating pod labels and annotations with m…

…apping (#1781) Signed-off-by: saintube <[email protected]>
koordinator-sh · Dec 28, 2023 · 3b10d2a · 3b10d2a
1 parent a404e7f
commit 3b10d2a
Show file tree

Hide file tree

Showing 5 changed files with 487 additions and 0 deletions.
diff --git a/apis/config/v1alpha1/cluster_colocation_profile_types.go b/apis/config/v1alpha1/cluster_colocation_profile_types.go
@@ -74,6 +74,16 @@ type ClusterColocationProfileSpec struct {
 	// +optional
 	Annotations map[string]string `json:"annotations,omitempty"`
 
+	// LabelKeysMapping describes the labels that needs to inject into Pod.Labels with the same values.
+	// It sets the Pod.Labels[LabelsToLabels[k]] = Pod.Labels[k] for each key k.
+	// +optional
+	LabelKeysMapping map[string]string `json:"labelKeysMapping,omitempty"`
+
+	// AnnotationKeysMapping describes the annotations that needs to inject into Pod.Annotations with the same values.
+	// It sets the Pod.Annotations[AnnotationsToAnnotations[k]] = Pod.Annotations[k] for each key k.
+	// +optional
+	AnnotationKeysMapping map[string]string `json:"annotationKeysMapping,omitempty"`
+
 	// If specified, the pod will be dispatched by specified scheduler.
 	// +optional
 	SchedulerName string `json:"schedulerName,omitempty"`

diff --git a/apis/config/v1alpha1/zz_generated.deepcopy.go b/apis/config/v1alpha1/zz_generated.deepcopy.go
diff --git a/config/crd/bases/config.koordinator.sh_clustercolocationprofiles.yaml b/config/crd/bases/config.koordinator.sh_clustercolocationprofiles.yaml
@@ -36,6 +36,14 @@ spec:
           spec:
             description: ClusterColocationProfileSpec is a description of a ClusterColocationProfile.
             properties:
+              annotationKeysMapping:
+                additionalProperties:
+                  type: string
+                description: AnnotationKeysMapping describes the annotations that
+                  needs to inject into Pod.Annotations with the same values. It sets
+                  the Pod.Annotations[AnnotationsToAnnotations[k]] = Pod.Annotations[k]
+                  for each key k.
+                type: object
               annotations:
                 additionalProperties:
                   type: string
@@ -51,6 +59,13 @@ spec:
                   priority.
                 format: int32
                 type: integer
+              labelKeysMapping:
+                additionalProperties:
+                  type: string
+                description: LabelKeysMapping describes the labels that needs to inject
+                  into Pod.Labels with the same values. It sets the Pod.Labels[LabelsToLabels[k]]
+                  = Pod.Labels[k] for each key k.
+                type: object
               labels:
                 additionalProperties:
                   type: string

diff --git a/pkg/webhook/pod/mutating/cluster_colocation_profile.go b/pkg/webhook/pod/mutating/cluster_colocation_profile.go
@@ -173,6 +173,24 @@ func (h *PodMutatingHandler) doMutateByColocationProfile(ctx context.Context, po
 		}
 	}
 
+	if len(profile.Spec.LabelKeysMapping) > 0 {
+		if pod.Labels == nil {
+			pod.Labels = make(map[string]string)
+		}
+		for keyOld, keyNew := range profile.Spec.LabelKeysMapping {
+			pod.Labels[keyNew] = pod.Labels[keyOld]
+		}
+	}
+
+	if len(profile.Spec.AnnotationKeysMapping) > 0 {
+		if pod.Annotations == nil {
+			pod.Annotations = make(map[string]string)
+		}
+		for keyOld, keyNew := range profile.Spec.AnnotationKeysMapping {
+			pod.Annotations[keyNew] = pod.Annotations[keyOld]
+		}
+	}
+
 	if profile.Spec.SchedulerName != "" {
 		pod.Spec.SchedulerName = profile.Spec.SchedulerName
 	}