diff options
author | 2019-08-07 00:14:07 -0700 | |
---|---|---|
committer | 2019-08-11 21:54:15 +0000 | |
commit | 787f229616b7d26d4a712288018b83c8de6d900c (patch) | |
tree | 7f464f4e4d7a7ff2980fde552269e57b127ad36d /vnfs/DAaaS/microservices/collectd-operator/pkg/controller/utils/collectdutils.go | |
parent | 4c13499b5915842b3c6b7c493310b185d89ae040 (diff) |
Collectd Global CRD
This patch introduces an enhancement to the existing Collectd Operator.
It has now 2 CRDS CollectdGlobal and CollectdPlugin and corresponding
controllers. CollectdGlobal is used to specify the global options of
collectd conf and CollectdPlugin is used to load/unload plugins.
Added synchronization when both controllers try to fetch the same
resources. Refactored some of the common code to collectdutils.
Issue-ID: ONAPARC-461
Signed-off-by: Dileep Ranganathan <dileep.ranganathan@intel.com>
Change-Id: Id2edf6fa6de56d9d7216ebb258b3017661c6a3dd
Diffstat (limited to 'vnfs/DAaaS/microservices/collectd-operator/pkg/controller/utils/collectdutils.go')
-rw-r--r-- | vnfs/DAaaS/microservices/collectd-operator/pkg/controller/utils/collectdutils.go | 218 |
1 files changed, 218 insertions, 0 deletions
diff --git a/vnfs/DAaaS/microservices/collectd-operator/pkg/controller/utils/collectdutils.go b/vnfs/DAaaS/microservices/collectd-operator/pkg/controller/utils/collectdutils.go new file mode 100644 index 00000000..0eb4e322 --- /dev/null +++ b/vnfs/DAaaS/microservices/collectd-operator/pkg/controller/utils/collectdutils.go @@ -0,0 +1,218 @@ +package utils + +import ( + "context" + "crypto/sha256" + "fmt" + "os" + "sync" + + "github.com/go-logr/logr" + + onapv1alpha1 "demo/vnfs/DAaaS/microservices/collectd-operator/pkg/apis/onap/v1alpha1" + + corev1 "k8s.io/api/core/v1" + extensionsv1beta1 "k8s.io/api/extensions/v1beta1" + "k8s.io/apimachinery/pkg/api/errors" + "sigs.k8s.io/controller-runtime/pkg/client" +) + +// Define the collectdPlugin finalizer for handling deletion +const ( + defaultWatchLabel = "app=collectd" + CollectdFinalizer = "finalizer.collectd.onap.org" + + // WatchLabelsEnvVar is the constant for env variable WATCH_LABELS + // which is the labels where the watch activity happens. + // this value is empty if the operator is running with clusterScope. + WatchLabelsEnvVar = "WATCH_LABELS" +) + +var lock sync.Mutex + +// ResourceMap to hold objects to update/reload +type ResourceMap struct { + ConfigMap *corev1.ConfigMap + DaemonSet *extensionsv1beta1.DaemonSet + CollectdPlugins *[]onapv1alpha1.CollectdPlugin +} + +// ComputeSHA256 returns hash of data as string +func ComputeSHA256(data []byte) string { + hash := sha256.Sum256(data) + return fmt.Sprintf("%x", hash) +} + +// Contains checks if a string is contained in a list of strings +func Contains(list []string, s string) bool { + for _, v := range list { + if v == s { + return true + } + } + return false +} + +// Remove removes a string from a list of string +func Remove(list []string, s string) []string { + for i, v := range list { + if v == s { + list = append(list[:i], list[i+1:]...) + } + } + return list +} + +// GetWatchLabels returns the labels the operator should be watching for changes +func GetWatchLabels() (string, error) { + labelSelector, found := os.LookupEnv(WatchLabelsEnvVar) + if !found { + return defaultWatchLabel, fmt.Errorf("%s must be set", WatchLabelsEnvVar) + } + return labelSelector, nil +} + +// FindResourceMapForCR returns the configMap, collectd Daemonset and list of Collectd Plugins +func FindResourceMapForCR(rc client.Client, reqLogger logr.Logger, ns string) (*ResourceMap, error) { + lock.Lock() + defer lock.Unlock() + cmList := &corev1.ConfigMapList{} + opts := &client.ListOptions{} + rmap := &ResourceMap{} + + // Select ConfigMaps with label + labelSelector, err := GetWatchLabels() + if err != nil { + reqLogger.Error(err, "Failed to get watch labels, continuing with default label") + } + opts.SetLabelSelector(labelSelector) + opts.InNamespace(ns) + + err = rc.List(context.TODO(), opts, cmList) + if err != nil { + return rmap, err + } + + if cmList.Items == nil || len(cmList.Items) == 0 { + return rmap, errors.NewNotFound(corev1.Resource("configmap"), "ConfigMap") + } + + // Select DaemonSets with label + dsList := &extensionsv1beta1.DaemonSetList{} + err = rc.List(context.TODO(), opts, dsList) + if err != nil { + return rmap, err + } + + if dsList.Items == nil || len(dsList.Items) == 0 { + return rmap, errors.NewNotFound(corev1.Resource("daemonset"), "DaemonSet") + } + + rmap.ConfigMap = &cmList.Items[0] + rmap.DaemonSet = &dsList.Items[0] + + return rmap, err +} + +// GetCollectdPluginList returns the list of CollectdPlugin instances in the namespace ns +func GetCollectdPluginList(rc client.Client, ns string) (*onapv1alpha1.CollectdPluginList, error) { + // Get all collectd plugins in the current namespace to rebuild conf. + collectdPlugins := &onapv1alpha1.CollectdPluginList{} + cpOpts := &client.ListOptions{} + cpOpts.InNamespace(ns) + err := rc.List(context.TODO(), cpOpts, collectdPlugins) + if err != nil { + return nil, err + } + return collectdPlugins, nil +} + +// GetCollectdGlobal returns the CollectdGlobal instance in the namespace ns +func GetCollectdGlobal(rc client.Client, ns string) (*onapv1alpha1.CollectdGlobal, error) { + // Get the CollectdGlobal instance in current namespace to rebuild conf. + cgList := &onapv1alpha1.CollectdGlobalList{} + cpOpts := &client.ListOptions{} + cpOpts.InNamespace(ns) + err := rc.List(context.TODO(), cpOpts, cgList) + if err != nil { + return nil, err + } + if cgList.Items == nil || len(cgList.Items) == 0 { + return nil, err + } + collectdGlobals := &cgList.Items[0] + return collectdGlobals, nil +} + +// GetPodList returns the list of pods in the namespace ns +func GetPodList(rc client.Client, ns string) ([]string, error) { + var pods []string + podList := &corev1.PodList{} + opts := &client.ListOptions{} + // Select ConfigMaps with label + labelSelector, _ := GetWatchLabels() + opts.SetLabelSelector(labelSelector) + opts.InNamespace(ns) + err := rc.List(context.TODO(), opts, podList) + if err != nil { + return nil, err + } + + if podList.Items == nil || len(podList.Items) == 0 { + return nil, err + } + + for _, pod := range podList.Items { + pods = append(pods, pod.Name) + } + return pods, nil +} + +// RebuildCollectdConf Get all collectd plugins and reconstruct, compute Hash and check for changes +func RebuildCollectdConf(rc client.Client, ns string, isDelete bool, delPlugin string) (string, error) { + var collectdConf, collectdGlobalConf string + // Get the collectd global in the current namespace to rebuild conf. + cg, err := GetCollectdGlobal(rc, ns) + if err != nil { + return "", err + } + + if cg != nil { + collectdGlobalConf += cg.Spec.GlobalOptions + "\n" + } + + // Get all collectd plugins in the current namespace to rebuild conf. + cp, err := GetCollectdPluginList(rc, ns) + if err != nil { + return "", err + } + cpList := &cp.Items + loadPlugin := make(map[string]string) + if *cpList != nil && len(*cpList) > 0 { + for _, cp := range *cpList { + // using CollectdPlugin to set global options. If CollectdGlobal CRD is defined do not check for this + if cp.Spec.PluginName == "global" { + if cg == nil { + collectdGlobalConf += cp.Spec.PluginConf + "\n" + } + } else { + loadPlugin[cp.Spec.PluginName] = cp.Spec.PluginConf + } + } + } + + if isDelete { + delete(loadPlugin, delPlugin) + } else { + collectdConf += collectdGlobalConf + } + + for cpName, cpConf := range loadPlugin { + collectdConf += "LoadPlugin" + " " + cpName + "\n" + collectdConf += cpConf + "\n" + } + + collectdConf += "#Last line (collectd requires ā\\nā at the last line)\n" + + return collectdConf, nil +} |