-
Notifications
You must be signed in to change notification settings - Fork 339
/
elasticsearch.go
67 lines (54 loc) · 1.54 KB
/
elasticsearch.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
package inventory
import (
"fmt"
esv1 "github.com/openshift/elasticsearch-operator/apis/logging/v1"
"github.com/jaegertracing/jaeger-operator/pkg/util"
)
// Elasticsearch represents the elastic search inventory based on the current and desired states
type Elasticsearch struct {
Create []esv1.Elasticsearch
Update []esv1.Elasticsearch
Delete []esv1.Elasticsearch
}
// ForElasticsearches builds a new elastic search inventory based on the existing and desired states
func ForElasticsearches(existing []esv1.Elasticsearch, desired []esv1.Elasticsearch) Elasticsearch {
update := []esv1.Elasticsearch{}
mcreate := esMap(desired)
mdelete := esMap(existing)
for k, v := range mcreate {
if t, ok := mdelete[k]; ok {
tp := t.DeepCopy()
util.InitObjectMeta(tp)
tp.Spec = v.Spec
tp.ObjectMeta.OwnerReferences = v.ObjectMeta.OwnerReferences
for k, v := range v.ObjectMeta.Annotations {
tp.ObjectMeta.Annotations[k] = v
}
for k, v := range v.ObjectMeta.Labels {
tp.ObjectMeta.Labels[k] = v
}
update = append(update, *tp)
delete(mcreate, k)
delete(mdelete, k)
}
}
return Elasticsearch{
Create: esList(mcreate),
Update: update,
Delete: esList(mdelete),
}
}
func esMap(deps []esv1.Elasticsearch) map[string]esv1.Elasticsearch {
m := map[string]esv1.Elasticsearch{}
for _, d := range deps {
m[fmt.Sprintf("%s.%s", d.Namespace, d.Name)] = d
}
return m
}
func esList(m map[string]esv1.Elasticsearch) []esv1.Elasticsearch {
l := []esv1.Elasticsearch{}
for _, v := range m {
l = append(l, v)
}
return l
}