fix: fix always update configmap

Signed-off-by: Rory Z <[email protected]>
emqx · Feb 21, 2024 · f840106 · f840106
1 parent a3ec997
commit f840106
Showing 1 changed file with 20 additions and 23 deletions.
diff --git a/controllers/apps/v2beta1/sync_emqx_config.go b/controllers/apps/v2beta1/sync_emqx_config.go
@@ -15,6 +15,7 @@ import (
 	k8sErrors "k8s.io/apimachinery/pkg/api/errors"
 	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
 	"k8s.io/apimachinery/pkg/types"
+	ctrl "sigs.k8s.io/controller-runtime"
 )
 
 type syncConfig struct {
@@ -32,8 +33,12 @@ func (s *syncConfig) reconcile(ctx context.Context, logger logr.Logger, instance
 		Namespace: instance.Namespace,
 	}, configMap); err != nil {
 		if k8sErrors.IsNotFound(err) {
-			if err := s.update(ctx, logger, instance, confStr); err != nil {
-				return subResult{err: emperror.Wrap(err, "failed to update emqx config")}
+			configMap = generateConfigMap(instance, confStr)
+			if err := ctrl.SetControllerReference(instance, configMap, s.Scheme); err != nil {
+				return subResult{err: emperror.Wrap(err, "failed to set controller reference for configMap")}
+			}
+			if err := s.Client.Create(ctx, configMap); err != nil {
+				return subResult{err: emperror.Wrap(err, "failed to create configMap")}
 			}
 			return subResult{}
 		}
@@ -42,8 +47,12 @@ func (s *syncConfig) reconcile(ctx context.Context, logger logr.Logger, instance
 
 	lastConfigStr, ok := instance.Annotations[appsv2beta1.AnnotationsLastEMQXConfigKey]
 	if !ok {
-		if err := s.update(ctx, logger, instance, confStr); err != nil {
-			return subResult{err: emperror.Wrap(err, "failed to update emqx config")}
+		if instance.Annotations == nil {
+			instance.Annotations = map[string]string{}
+		}
+		instance.Annotations[appsv2beta1.AnnotationsLastEMQXConfigKey] = confStr
+		if err := s.Client.Update(ctx, instance); err != nil {
+			return subResult{err: emperror.Wrap(err, "failed to update emqx instance annotation")}
 		}
 		return subResult{}
 	}
@@ -78,8 +87,13 @@ func (s *syncConfig) reconcile(ctx context.Context, logger logr.Logger, instance
 			return subResult{err: emperror.Wrap(err, "failed to put emqx config")}
 		}
 
-		if err := s.update(ctx, logger, instance, confStr); err != nil {
-			return subResult{err: emperror.Wrap(err, "failed to update emqx config")}
+		if err := s.Client.Update(ctx, generateConfigMap(instance, confStr)); err != nil {
+			return subResult{err: emperror.Wrap(err, "failed to update configMap")}
+		}
+
+		instance.Annotations[appsv2beta1.AnnotationsLastEMQXConfigKey] = confStr
+		if err := s.Client.Update(ctx, instance); err != nil {
+			return subResult{err: emperror.Wrap(err, "failed to update emqx instance annotation")}
 		}
 
 		return subResult{}
@@ -88,23 +102,6 @@ func (s *syncConfig) reconcile(ctx context.Context, logger logr.Logger, instance
 	return subResult{}
 }
 
-func (s *syncConfig) update(ctx context.Context, logger logr.Logger, instance *appsv2beta1.EMQX, confStr string) error {
-	configMap := generateConfigMap(instance, confStr)
-	if err := s.Handler.CreateOrUpdate(ctx, s.Scheme, logger, instance, configMap); err != nil {
-		return emperror.Wrap(err, "failed to create or update configMap")
-	}
-
-	if instance.Annotations == nil {
-		instance.Annotations = map[string]string{}
-	}
-	instance.Annotations[appsv2beta1.AnnotationsLastEMQXConfigKey] = confStr
-	if err := s.Client.Update(ctx, instance); err != nil {
-		return emperror.Wrap(err, "failed to update emqx instance annotation")
-	}
-
-	return nil
-}
-
 func generateConfigMap(instance *appsv2beta1.EMQX, data string) *corev1.ConfigMap {
 	return &corev1.ConfigMap{
 		TypeMeta: metav1.TypeMeta{