Ensure labels are initialized in deployment and pod mutations

Author: chenzhiguo

pkg/mutation/mutation_deploy.go

@@ -87,6 +87,22 @@ func injectionDeploy(request *admissionv1.AdmissionRequest) (*admissionv1.Admiss
 		target := deploy.DeepCopy()
 		added := false
 
+		if deploy.Labels == nil {
+			deploy.Labels = make(map[string]string)
+		}
+
+		if deploy.Spec.Template.Labels == nil {
+			deploy.Spec.Template.Labels = make(map[string]string)
+		}
+
+		if target.Labels == nil {
+			target.Labels = make(map[string]string)
+		}
+
+		if target.Spec.Template.Labels == nil {
+			target.Spec.Template.Labels = make(map[string]string)
+		}
+
 		if enhanceType, ok := deploy.Labels[config.EnhanceTypeLabel]; ok && enhanceType == config.EnhanceTypeSidecar {
 			log.Infof("[mutation] /injection-deploy: add label %s to deployment %s/%s", config.SidecarEnhanceLabel, deploy.Name, deploy.Namespace)
 			target.Spec.Template.Labels[config.SidecarEnhanceLabel] = "true"

pkg/mutation/mutation_pod.go

@@ -66,6 +66,15 @@ func injectionPod(request *admissionv1.AdmissionRequest) (*admissionv1.Admission
 
 		envs := makePodEnvs(&pod)
 		targetPod := pod.DeepCopy()
+
+		if pod.Labels == nil {
+			pod.Labels = make(map[string]string)
+		}
+
+		if targetPod.Labels == nil {
+			targetPod.Labels = make(map[string]string)
+		}
+
 		rs := resource.GetResource()
 		deploymentName, err := rs.GetDeploymentName(&pod, request.Namespace)
 		if err != nil {

pkg/resource/control.go

@@ -30,12 +30,11 @@ func GetApplicationEnvironments(labels map[string]string) (map[string]string, er
 	envMaps := make(map[string]string)
 	if len(serviceSpace) != 0 && len(application) != 0 {
 		url := fmt.Sprintf(
-			"%s/ns/%s/application/%s/environments",
-			config.ControlPlaneUrl, serviceSpace, application,
+			"%s/ns/%s/application/%s/environments", config.ControlPlaneUrl, serviceSpace, application,
 		)
 		resp, err := http.Get(url)
 		if err != nil {
-			return nil, err
+			return nil, nil
 		}
 		defer resp.Body.Close()