Fix labeling clusters with multiple labels

When using `pgo label` or `pgo delete label` with a list of labels,
PgCluster was getting different labels than its Deployments.

Issue: [ch8740]
See: 88cd75f

Author: cbandy

internal/apiserver/labelservice/labelimpl.go

@@ -125,7 +125,7 @@ func addLabels(items []crv1.Pgcluster, DryRun bool, LabelCmdLabel string, newLab
 			log.Debug("dry run only")
 		} else {
 			log.Debugf("adding label to cluster %s", items[i].Spec.Name)
-			err := PatchPgcluster(LabelCmdLabel, items[i], ns)
+			err := PatchPgcluster(newLabels, items[i], ns)
 			if err != nil {
 				log.Error(err.Error())
 			}
@@ -224,19 +224,18 @@ func updateLabels(deployment *v1.Deployment, clusterName string, newLabels map[s
 
 }
 
-func PatchPgcluster(newLabel string, oldCRD crv1.Pgcluster, ns string) error {
+func PatchPgcluster(newLabels map[string]string, oldCRD crv1.Pgcluster, ns string) error {
 
-	fields := strings.Split(newLabel, "=")
-	labelKey := fields[0]
-	labelValue := fields[1]
 	oldData, err := json.Marshal(oldCRD)
 	if err != nil {
 		return err
 	}
 	if oldCRD.ObjectMeta.Labels == nil {
 		oldCRD.ObjectMeta.Labels = make(map[string]string)
 	}
-	oldCRD.ObjectMeta.Labels[labelKey] = labelValue
+	for key, value := range newLabels {
+		oldCRD.ObjectMeta.Labels[key] = value
+	}
 	var newData, patchBytes []byte
 	newData, err = json.Marshal(oldCRD)
 	if err != nil {
@@ -377,7 +376,7 @@ func deleteLabels(items []crv1.Pgcluster, LabelCmdLabel string, labelsMap map[st
 
 	for i := 0; i < len(items); i++ {
 		log.Debugf("deleting label from %s", items[i].Spec.Name)
-		err = deletePatchPgcluster(LabelCmdLabel, items[i], ns)
+		err = deletePatchPgcluster(labelsMap, items[i], ns)
 		if err != nil {
 			log.Error(err.Error())
 			return err
@@ -406,19 +405,18 @@ func deleteLabels(items []crv1.Pgcluster, LabelCmdLabel string, labelsMap map[st
 	return err
 }
 
-func deletePatchPgcluster(newLabel string, oldCRD crv1.Pgcluster, ns string) error {
+func deletePatchPgcluster(labelsMap map[string]string, oldCRD crv1.Pgcluster, ns string) error {
 
-	fields := strings.Split(newLabel, "=")
-	labelKey := fields[0]
-	//labelValue := fields[1]
 	oldData, err := json.Marshal(oldCRD)
 	if err != nil {
 		return err
 	}
 	if oldCRD.ObjectMeta.Labels == nil {
 		oldCRD.ObjectMeta.Labels = make(map[string]string)
 	}
-	delete(oldCRD.ObjectMeta.Labels, labelKey)
+	for k := range labelsMap {
+		delete(oldCRD.ObjectMeta.Labels, k)
+	}
 
 	var newData, patchBytes []byte
 	newData, err = json.Marshal(oldCRD)

internal/apiserver/policyservice/policyimpl.go

@@ -270,7 +270,7 @@ func ApplyPolicy(request *msgs.ApplyPolicyRequest, ns, pgouser string) msgs.Appl
 		}
 
 		//update the pgcluster crd labels with the new policy
-		err = labelservice.PatchPgcluster(request.Name+"="+config.LABEL_PGPOLICY, cl, ns)
+		err = labelservice.PatchPgcluster(map[string]string{request.Name: config.LABEL_PGPOLICY}, cl, ns)
 		if err != nil {
 			log.Error(err)
 		}