Collectd Operator Unit Tests
[demo.git] / vnfs / DAaaS / microservices / collectd-operator / pkg / controller / collectdglobal / collectdglobal_controller.go
index a43afdc..d1bd5c6 100644 (file)
@@ -1,3 +1,16 @@
+/*
+Copyright 2019 Intel Corporation.
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+    http://www.apache.org/licenses/LICENSE-2.0
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+*/
+
 package collectdglobal
 
 import (
@@ -12,6 +25,7 @@ import (
 
        onapv1alpha1 "collectd-operator/pkg/apis/onap/v1alpha1"
        collectdutils "collectd-operator/pkg/controller/utils"
+       dspredicate "collectd-operator/pkg/controller/utils"
        dsutils "collectd-operator/pkg/controller/utils"
 
        appsv1 "k8s.io/api/apps/v1"
@@ -69,11 +83,12 @@ func add(mgr manager.Manager, r reconcile.Reconciler) error {
                                        log.Error(err, "Failed to get watch labels, continuing with default label")
                                }
                                rcp := r.(*ReconcileCollectdGlobal)
-                               // Select the Daemonset with labelSelector (Defautl  is app=collectd)
+                               // Select the Daemonset with labelSelector (Default  is app=collectd)
                                if a.Meta.GetLabels()[labels[0]] == labels[1] {
                                        var requests []reconcile.Request
                                        cg, err := collectdutils.GetCollectdGlobal(rcp.client, a.Meta.GetNamespace())
                                        if err != nil || cg == nil {
+                                               log.V(1).Info("No CollectdGlobal CR instance Exist")
                                                return nil
                                        }
                                        requests = append(requests, reconcile.Request{
@@ -82,7 +97,7 @@ func add(mgr manager.Manager, r reconcile.Reconciler) error {
                                }
                                return nil
                        }),
-               })
+               }, dspredicate.DaemonSetStatusChangedPredicate{})
        if err != nil {
                return err
        }
@@ -139,7 +154,7 @@ func (r *ReconcileCollectdGlobal) Reconcile(request reconcile.Request) (reconcil
                if err := r.addFinalizer(reqLogger, instance); err != nil {
                        return reconcile.Result{}, err
                }
-               return reconcile.Result{}, nil
+               //return reconcile.Result{}, nil
        }
        // Handle the reconciliation for CollectdGlobal.
        // At this stage the Status of the CollectdGlobal should NOT be ""
@@ -149,7 +164,8 @@ func (r *ReconcileCollectdGlobal) Reconcile(request reconcile.Request) (reconcil
 
 // handleCollectdGlobal regenerates the collectd conf on CR Create, Update, Delete events
 func (r *ReconcileCollectdGlobal) handleCollectdGlobal(reqLogger logr.Logger, cr *onapv1alpha1.CollectdGlobal, isDelete bool) error {
-       var collectdConf string
+       collectdutils.ReconcileLock.Lock()
+       defer collectdutils.ReconcileLock.Unlock()
 
        retryErr := retry.RetryOnConflict(retry.DefaultRetry, func() error {
                cm, err := collectdutils.GetConfigMap(r.client, reqLogger, cr.Namespace)
@@ -174,13 +190,11 @@ func (r *ReconcileCollectdGlobal) handleCollectdGlobal(reqLogger logr.Logger, cr
                // Update the ConfigMap with new Spec and reload DaemonSets
                reqLogger.Info("Updating the ConfigMap", "ConfigMap.Namespace", cm.Namespace, "ConfigMap.Name", cm.Name)
                updateErr := r.client.Update(context.TODO(), cm)
-               return updateErr
-       })
-       if retryErr != nil {
-               panic(fmt.Errorf("Update failed: %v", retryErr))
-       }
+               if updateErr != nil {
+                       reqLogger.Error(updateErr, "Update ConfigMap failed")
+                       return updateErr
+               }
 
-       retryErr = retry.RetryOnConflict(retry.DefaultRetry, func() error {
                // Retrieve the latest version of Daemonset before attempting update
                // RetryOnConflict uses exponential backoff to avoid exhausting the apiserver
                // Select DaemonSets with label
@@ -208,17 +222,20 @@ func (r *ReconcileCollectdGlobal) handleCollectdGlobal(reqLogger logr.Logger, cr
                        "daaas-random": collectdutils.ComputeSHA256([]byte(collectdConf)),
                })
                r.handleTypesDB(reqLogger, cr, ds, isDelete)
-               updateErr := r.client.Update(context.TODO(), ds)
+               updateErr = r.client.Update(context.TODO(), ds)
                return updateErr
        })
        if retryErr != nil {
                panic(fmt.Errorf("Update failed: %v", retryErr))
        }
 
-       err := r.updateStatus(cr)
-       if err != nil {
-               reqLogger.Error(err, "Unable to update status")
+       retryErr = retry.RetryOnConflict(retry.DefaultRetry, func() error {
+               err := r.updateStatus(cr)
                return err
+       })
+
+       if retryErr != nil {
+               panic(fmt.Errorf("Update failed: %v", retryErr))
        }
        // Reconcile success
        reqLogger.Info("Reconcile success!!")
@@ -257,22 +274,29 @@ func (r *ReconcileCollectdGlobal) handleDelete(reqLogger logr.Logger, cr *onapv1
 }
 
 func (r *ReconcileCollectdGlobal) updateStatus(cr *onapv1alpha1.CollectdGlobal) error {
-       switch cr.Status.Status {
+       // Fetch the CollectdGlobal instance
+       instance := &onapv1alpha1.CollectdGlobal{}
+       key := types.NamespacedName{Namespace: cr.Namespace, Name: cr.Name}
+       err := r.client.Get(context.TODO(), key, instance)
+       if err != nil {
+               return err
+       }
+       switch instance.Status.Status {
        case onapv1alpha1.Initial:
-               cr.Status.Status = onapv1alpha1.Created
+               instance.Status.Status = onapv1alpha1.Created
        case onapv1alpha1.Created, onapv1alpha1.Enabled:
-               pods, err := collectdutils.GetPodList(r.client, cr.Namespace)
+               pods, err := collectdutils.GetPodList(r.client, instance.Namespace)
                if err != nil {
                        return err
                }
-               if !reflect.DeepEqual(pods, cr.Status.CollectdAgents) {
-                       cr.Status.CollectdAgents = pods
-                       cr.Status.Status = onapv1alpha1.Enabled
+               if !reflect.DeepEqual(pods, instance.Status.CollectdAgents) {
+                       instance.Status.CollectdAgents = pods
+                       instance.Status.Status = onapv1alpha1.Enabled
                }
        case onapv1alpha1.Deleting, onapv1alpha1.Deprecated:
                return nil
        }
-       err := r.client.Status().Update(context.TODO(), cr)
+       err = r.client.Status().Update(context.TODO(), instance)
        return err
 }
 
@@ -292,11 +316,12 @@ func (r *ReconcileCollectdGlobal) addFinalizer(reqLogger logr.Logger, cr *onapv1
        // Update status from Initial to Created
        // Since addFinalizer will be executed only once,
        // the status will be changed from Initial state to Created
-       updateErr := r.updateStatus(cr)
-       if updateErr != nil {
-               reqLogger.Error(updateErr, "Failed to update status from Initial state")
-       }
+       // updateErr := r.updateStatus(cr)
+       // if updateErr != nil {
+       //      reqLogger.Error(updateErr, "Failed to update status from Initial state")
+       // }
        // Update CR
+       cr.Status.Status = onapv1alpha1.Created
        err := r.client.Update(context.TODO(), cr)
        if err != nil {
                reqLogger.Error(err, "Failed to update CollectdGlobal with finalizer")