-
Notifications
You must be signed in to change notification settings - Fork 136
/
resource_query.go
117 lines (105 loc) · 3.74 KB
/
resource_query.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
115
116
117
package support_archive
import (
"reflect"
"github.com/Dynatrace/dynatrace-operator/src/api/v1alpha1"
"github.com/Dynatrace/dynatrace-operator/src/api/v1alpha1/edgeconnect"
dynatracev1beta1 "github.com/Dynatrace/dynatrace-operator/src/api/v1beta1"
dynakubev1beta1 "github.com/Dynatrace/dynatrace-operator/src/api/v1beta1/dynakube"
"github.com/Dynatrace/dynatrace-operator/src/kubeobjects"
"github.com/Dynatrace/dynatrace-operator/src/webhook"
appsv1 "k8s.io/api/apps/v1"
corev1 "k8s.io/api/core/v1"
"k8s.io/apimachinery/pkg/fields"
"k8s.io/apimachinery/pkg/runtime/schema"
"sigs.k8s.io/controller-runtime/pkg/client"
)
type resourceQueryGroup struct {
resources []schema.GroupVersionKind
filters []client.ListOption
}
type resourceQuery struct {
groupVersionKind schema.GroupVersionKind
filters []client.ListOption
}
func getQueries(namespace string, appName string) []resourceQuery {
allQueries := make([]resourceQuery, 0)
allQueries = append(allQueries, getInjectedNamespaceQueryGroup().getQueries()...)
allQueries = append(allQueries, getOperatorNamespaceQueryGroup(namespace).getQueries()...)
allQueries = append(allQueries, getComponentsQueryGroup(namespace, appName, kubeobjects.AppNameLabel).getQueries()...)
allQueries = append(allQueries, getComponentsQueryGroup(namespace, appName, kubeobjects.AppManagedByLabel).getQueries()...)
allQueries = append(allQueries, getCustomResourcesQueryGroup(namespace).getQueries()...)
return allQueries
}
func getInjectedNamespaceQueryGroup() resourceQueryGroup {
return resourceQueryGroup{
resources: []schema.GroupVersionKind{
toGroupVersionKind(corev1.SchemeGroupVersion, corev1.Namespace{}),
},
filters: []client.ListOption{
client.HasLabels{
webhook.InjectionInstanceLabel,
},
},
}
}
func getOperatorNamespaceQueryGroup(namespace string) resourceQueryGroup {
return resourceQueryGroup{
resources: []schema.GroupVersionKind{
toGroupVersionKind(corev1.SchemeGroupVersion, corev1.Namespace{}),
},
filters: []client.ListOption{
&client.ListOptions{
FieldSelector: fields.OneTermEqualSelector("metadata.name", namespace),
},
},
}
}
func getComponentsQueryGroup(namespace string, appName string, labelKey string) resourceQueryGroup {
return resourceQueryGroup{
resources: []schema.GroupVersionKind{
toGroupVersionKind(appsv1.SchemeGroupVersion, appsv1.Deployment{}),
toGroupVersionKind(appsv1.SchemeGroupVersion, appsv1.StatefulSet{}),
toGroupVersionKind(appsv1.SchemeGroupVersion, appsv1.DaemonSet{}),
toGroupVersionKind(appsv1.SchemeGroupVersion, appsv1.ReplicaSet{}),
toGroupVersionKind(corev1.SchemeGroupVersion, corev1.Service{}),
toGroupVersionKind(corev1.SchemeGroupVersion, corev1.Pod{}),
},
filters: []client.ListOption{
client.MatchingLabels{
labelKey: appName,
},
client.InNamespace(namespace),
},
}
}
func getCustomResourcesQueryGroup(namespace string) resourceQueryGroup {
return resourceQueryGroup{
resources: []schema.GroupVersionKind{
toGroupVersionKind(dynatracev1beta1.GroupVersion, dynakubev1beta1.DynaKube{}),
toGroupVersionKind(v1alpha1.GroupVersion, edgeconnect.EdgeConnect{}),
},
filters: []client.ListOption{
client.InNamespace(namespace),
},
}
}
func toGroupVersionKind(groupVersion schema.GroupVersion, resource any) schema.GroupVersionKind {
typ := reflect.TypeOf(resource)
typ.Name()
gvk := schema.GroupVersionKind{
Group: groupVersion.Group,
Version: groupVersion.Version,
Kind: typ.Name(),
}
return gvk
}
func (q resourceQueryGroup) getQueries() []resourceQuery {
queries := make([]resourceQuery, 0, len(q.resources))
for _, resource := range q.resources {
queries = append(queries, resourceQuery{
groupVersionKind: resource,
filters: q.filters,
})
}
return queries
}