/
controller_shared.go
107 lines (90 loc) · 2.95 KB
/
controller_shared.go
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
package controllers
import (
"bytes"
"context"
"encoding/json"
"time"
"github.com/grafana/grafana-operator/v5/api/v1beta1"
"github.com/grafana/grafana-operator/v5/controllers/model"
"k8s.io/apimachinery/pkg/api/meta"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
v1 "k8s.io/apimachinery/pkg/apis/meta/v1"
"k8s.io/apimachinery/pkg/runtime"
"sigs.k8s.io/controller-runtime/pkg/client"
"sigs.k8s.io/controller-runtime/pkg/event"
"sigs.k8s.io/controller-runtime/pkg/predicate"
)
const grafanaFinalizer = "operator.grafana.com/finalizer"
const (
conditionNoMatchingInstance = "NoMatchingInstance"
conditionNoMatchingFolder = "NoMatchingFolder"
)
func GetMatchingInstances(ctx context.Context, k8sClient client.Client, labelSelector *v1.LabelSelector) (v1beta1.GrafanaList, error) {
if labelSelector == nil {
return v1beta1.GrafanaList{}, nil
}
var list v1beta1.GrafanaList
opts := []client.ListOption{
client.MatchingLabels(labelSelector.MatchLabels),
}
err := k8sClient.List(ctx, &list, opts...)
return list, err
}
func ReconcilePlugins(ctx context.Context, k8sClient client.Client, scheme *runtime.Scheme, grafana *v1beta1.Grafana, plugins v1beta1.PluginList, resource string) error {
pluginsConfigMap := model.GetPluginsConfigMap(grafana, scheme)
selector := client.ObjectKey{
Namespace: pluginsConfigMap.Namespace,
Name: pluginsConfigMap.Name,
}
err := k8sClient.Get(ctx, selector, pluginsConfigMap)
if err != nil {
return err
}
val, err := json.Marshal(plugins.Sanitize())
if err != nil {
return err
}
if pluginsConfigMap.BinaryData == nil {
pluginsConfigMap.BinaryData = make(map[string][]byte)
}
if !bytes.Equal(val, pluginsConfigMap.BinaryData[resource]) {
pluginsConfigMap.BinaryData[resource] = val
return k8sClient.Update(ctx, pluginsConfigMap)
}
return nil
}
func setNoMatchingInstance(conditions *[]metav1.Condition, generation int64, reason, message string) {
meta.SetStatusCondition(conditions, metav1.Condition{
Type: conditionNoMatchingInstance,
Status: "True",
ObservedGeneration: generation,
LastTransitionTime: metav1.Time{
Time: time.Now(),
},
Reason: reason,
Message: message,
})
}
func setNoMatchingFolder(conditions *[]metav1.Condition, generation int64, reason, message string) {
meta.SetStatusCondition(conditions, metav1.Condition{
Type: conditionNoMatchingFolder,
Status: "True",
ObservedGeneration: generation,
LastTransitionTime: metav1.Time{
Time: time.Now(),
},
Reason: reason,
Message: message,
})
}
func removeNoMatchingInstance(conditions *[]metav1.Condition) {
meta.RemoveStatusCondition(conditions, conditionNoMatchingInstance)
}
func ignoreStatusUpdates() predicate.Predicate {
return predicate.Funcs{
UpdateFunc: func(e event.UpdateEvent) bool {
// Ignore updates to CR status in which case metadata.Generation does not change
return e.ObjectOld.GetGeneration() != e.ObjectNew.GetGeneration()
},
}
}