1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
|
package utils
import (
"context"
"crypto/sha256"
"fmt"
"os"
"sync"
"github.com/go-logr/logr"
onapv1alpha1 "collectd-operator/pkg/apis/onap/v1alpha1"
appsv1 "k8s.io/api/apps/v1"
corev1 "k8s.io/api/core/v1"
"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 *appsv1.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 := &appsv1.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
}
|
