Check if labels are nil before assigning

davide-pasquero · Oct 13, 2020 · 0685861 · 0685861
1 parent 02ebc44
commit 0685861
Show file tree

Hide file tree

Showing 4 changed files with 33 additions and 3 deletions.
diff --git a/pkg/controllers/management/auth/crtb_handler.go b/pkg/controllers/management/auth/crtb_handler.go
@@ -227,6 +227,9 @@ func (c *crtbLifecycle) reconcileLabels(binding *v3.ClusterRoleTemplateBinding)
 			if updateErr != nil {
 				return updateErr
 			}
+			if crbToUpdate.Labels == nil {
+				crbToUpdate.Labels = make(map[string]string)
+			}
 			crbToUpdate.Labels[bindingKey] = membershipBindingOwner
 			crbToUpdate.Labels[rtbLabelUpdated] = "true"
 			_, err := c.mgr.crbClient.Update(crbToUpdate)
@@ -249,6 +252,9 @@ func (c *crtbLifecycle) reconcileLabels(binding *v3.ClusterRoleTemplateBinding)
 			if updateErr != nil {
 				return updateErr
 			}
+			if rbToUpdate.Labels == nil {
+				rbToUpdate.Labels = make(map[string]string)
+			}
 			rbToUpdate.Labels[bindingKey] = crtbInProjectBindingOwner
 			rbToUpdate.Labels[rtbLabelUpdated] = "true"
 			_, err := c.mgr.rbClient.Update(rbToUpdate)
@@ -267,7 +273,10 @@ func (c *crtbLifecycle) reconcileLabels(binding *v3.ClusterRoleTemplateBinding)
 		if updateErr != nil {
 			return updateErr
 		}
-		binding.Labels[rtbCrbRbLabelsUpdated] = "true"
+		if crtbToUpdate.Labels == nil {
+			crtbToUpdate.Labels = make(map[string]string)
+		}
+		crtbToUpdate.Labels[rtbCrbRbLabelsUpdated] = "true"
 		_, err := c.mgr.crtbs.Update(crtbToUpdate)
 		return err
 	})

diff --git a/pkg/controllers/management/auth/prtb_handler.go b/pkg/controllers/management/auth/prtb_handler.go
@@ -240,6 +240,9 @@ func (p *prtbLifecycle) reconcileLabels(binding *v3.ProjectRoleTemplateBinding)
 			if updateErr != nil {
 				return updateErr
 			}
+			if crbToUpdate.Labels == nil {
+				crbToUpdate.Labels = make(map[string]string)
+			}
 			crbToUpdate.Labels[bindingKey] = membershipBindingOwner
 			crbToUpdate.Labels[rtbLabelUpdated] = "true"
 			_, err := p.mgr.crbClient.Update(crbToUpdate)
@@ -262,6 +265,9 @@ func (p *prtbLifecycle) reconcileLabels(binding *v3.ProjectRoleTemplateBinding)
 				if updateErr != nil {
 					return updateErr
 				}
+				if rbToUpdate.Labels == nil {
+					rbToUpdate.Labels = make(map[string]string)
+				}
 				rbToUpdate.Labels[bindingKey] = prtbLabel
 				rbToUpdate.Labels[rtbLabelUpdated] = "true"
 				_, err := p.mgr.rbClient.Update(rbToUpdate)
@@ -281,7 +287,10 @@ func (p *prtbLifecycle) reconcileLabels(binding *v3.ProjectRoleTemplateBinding)
 		if updateErr != nil {
 			return updateErr
 		}
-		binding.Labels[rtbCrbRbLabelsUpdated] = "true"
+		if prtbToUpdate.Labels == nil {
+			prtbToUpdate.Labels = make(map[string]string)
+		}
+		prtbToUpdate.Labels[rtbCrbRbLabelsUpdated] = "true"
 		_, err := p.mgr.prtbs.Update(prtbToUpdate)
 		return err
 	})

diff --git a/pkg/controllers/managementuser/rbac/crtb_handler.go b/pkg/controllers/managementuser/rbac/crtb_handler.go
@@ -121,6 +121,9 @@ func (c *crtbLifecycle) reconcileCRTBUserClusterLabels(binding *v3.ClusterRoleTe
 			if updateErr != nil {
 				return updateErr
 			}
+			if crbToUpdate.Labels == nil {
+				crbToUpdate.Labels = make(map[string]string)
+			}
 			crbToUpdate.Labels[rtbOwnerLabel] = bindingValue
 			crbToUpdate.Labels[rtbLabelUpdated] = "true"
 			_, err := c.m.clusterRoleBindings.Update(crbToUpdate)

diff --git a/pkg/controllers/managementuser/rbac/prtb_handler.go b/pkg/controllers/managementuser/rbac/prtb_handler.go
@@ -356,6 +356,9 @@ func (p *prtbLifecycle) reconcilePRTBUserClusterLabels(binding *v3.ProjectRoleTe
 			if updateErr != nil {
 				return updateErr
 			}
+			if crbToUpdate.Labels == nil {
+				crbToUpdate.Labels = make(map[string]string)
+			}
 			crbToUpdate.Labels[bindingLabel] = owner
 			crbToUpdate.Labels[rtbLabelUpdated] = "true"
 			_, err := p.m.clusterRoleBindings.Update(crbToUpdate)
@@ -381,6 +384,9 @@ func (p *prtbLifecycle) reconcilePRTBUserClusterLabels(binding *v3.ProjectRoleTe
 			if updateErr != nil {
 				return updateErr
 			}
+			if rbToUpdate.Labels == nil {
+				rbToUpdate.Labels = make(map[string]string)
+			}
 			rbToUpdate.Labels[rtbOwnerLabel] = bindingLabel
 			rbToUpdate.Labels[rtbLabelUpdated] = "true"
 			_, err := p.m.roleBindings.Update(rbToUpdate)
@@ -400,7 +406,10 @@ func (p *prtbLifecycle) reconcilePRTBUserClusterLabels(binding *v3.ProjectRoleTe
 		if updateErr != nil {
 			return updateErr
 		}
-		binding.Labels[rtbCrbRbLabelsUpdated] = "true"
+		if crtbToUpdate.Labels == nil {
+			crtbToUpdate.Labels = make(map[string]string)
+		}
+		crtbToUpdate.Labels[rtbCrbRbLabelsUpdated] = "true"
 		_, err := p.m.prtbs.Update(crtbToUpdate)
 		return err
 	})