-
Notifications
You must be signed in to change notification settings - Fork 136
/
dynakubes.go
114 lines (88 loc) · 2.91 KB
/
dynakubes.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
108
109
110
111
112
113
114
package mapper
import (
"context"
dynatracev1beta1 "github.com/Dynatrace/dynatrace-operator/pkg/api/v1beta1/dynakube"
dtwebhook "github.com/Dynatrace/dynatrace-operator/pkg/webhook"
"github.com/pkg/errors"
corev1 "k8s.io/api/core/v1"
"sigs.k8s.io/controller-runtime/pkg/client"
)
// DynakubeMapper manages the mapping creation from the dynakube's side
type DynakubeMapper struct {
ctx context.Context
client client.Client
apiReader client.Reader
dk *dynatracev1beta1.DynaKube
operatorNs string
}
func NewDynakubeMapper(ctx context.Context, clt client.Client, apiReader client.Reader, operatorNs string, dk *dynatracev1beta1.DynaKube) DynakubeMapper {
return DynakubeMapper{ctx: ctx, client: clt, apiReader: apiReader, operatorNs: operatorNs, dk: dk}
}
// MapFromDynakube checks all the namespaces to all the dynakubes
// updates the labels on the namespaces if necessary,
// finds conflicting dynakubes (2 dynakube with codeModules on the same namespace)
func (dm DynakubeMapper) MapFromDynakube() error {
modifiedNs, err := dm.MatchingNamespaces()
if err != nil {
return errors.Cause(err)
}
return dm.updateNamespaces(modifiedNs)
}
func (dm DynakubeMapper) MatchingNamespaces() ([]*corev1.Namespace, error) {
nsList := &corev1.NamespaceList{}
if err := dm.apiReader.List(dm.ctx, nsList); err != nil {
return nil, errors.Cause(err)
}
dkList := &dynatracev1beta1.DynaKubeList{}
if err := dm.apiReader.List(dm.ctx, dkList, &client.ListOptions{Namespace: dm.operatorNs}); err != nil {
return nil, errors.Cause(err)
}
return dm.mapFromDynakube(nsList, dkList)
}
func (dm DynakubeMapper) UnmapFromDynaKube(namespaces []corev1.Namespace) error {
for _, ns := range namespaces {
delete(ns.Labels, dtwebhook.InjectionInstanceLabel)
ns := ns
setUpdatedViaDynakubeAnnotation(&ns)
if err := dm.client.Update(dm.ctx, &ns); err != nil {
return errors.WithMessagef(err, "failed to remove label %s from namespace %s", dtwebhook.InjectionInstanceLabel, ns.Name)
}
}
return nil
}
func (dm DynakubeMapper) mapFromDynakube(nsList *corev1.NamespaceList, dkList *dynatracev1beta1.DynaKubeList) ([]*corev1.Namespace, error) {
var updated bool
var err error
var modifiedNs []*corev1.Namespace
replaced := false
for i := range dkList.Items {
if dkList.Items[i].Name == dm.dk.Name {
dkList.Items[i] = *dm.dk
replaced = true
break
}
}
if !replaced {
dkList.Items = append(dkList.Items, *dm.dk)
}
for i := range nsList.Items {
namespace := &nsList.Items[i]
updated, err = updateNamespace(namespace, dkList)
if err != nil {
return nil, err
}
if updated {
modifiedNs = append(modifiedNs, namespace)
}
}
return modifiedNs, err
}
func (dm DynakubeMapper) updateNamespaces(modifiedNs []*corev1.Namespace) error {
for _, ns := range modifiedNs {
setUpdatedViaDynakubeAnnotation(ns)
if err := dm.client.Update(dm.ctx, ns); err != nil {
return err
}
}
return nil
}