-
Notifications
You must be signed in to change notification settings - Fork 339
/
kafka.go
85 lines (70 loc) · 1.82 KB
/
kafka.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
package inventory
import (
"fmt"
"sigs.k8s.io/controller-runtime/pkg/log"
"github.com/jaegertracing/jaeger-operator/pkg/kafka/v1beta2"
"github.com/jaegertracing/jaeger-operator/pkg/util"
)
// Kafka represents the inventory of kafkas based on the current and desired states
type Kafka struct {
Create []v1beta2.Kafka
Update []v1beta2.Kafka
Delete []v1beta2.Kafka
}
// ForKafkas builds an inventory of kafkas based on the existing and desired states
func ForKafkas(existing []v1beta2.Kafka, desired []v1beta2.Kafka) Kafka {
update := []v1beta2.Kafka{}
mcreate := kafkaMap(desired)
mdelete := kafkaMap(existing)
for _, k := range existing {
log.Log.V(-1).Info(
"existing",
"kafka", k.GetName(),
"namespace", k.GetNamespace(),
)
}
for _, k := range desired {
log.Log.V(-1).Info(
"desired",
"kafka", k.GetName(),
"namespace", k.GetNamespace(),
)
}
for k, v := range mcreate {
if t, ok := mdelete[k]; ok {
tp := t.DeepCopy()
util.InitObjectMeta(tp)
// we can't blindly DeepCopyInto, so, we select what we bring from the new to the old object
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 Kafka{
Create: kafkaList(mcreate),
Update: update,
Delete: kafkaList(mdelete),
}
}
func kafkaMap(deps []v1beta2.Kafka) map[string]v1beta2.Kafka {
m := map[string]v1beta2.Kafka{}
for _, d := range deps {
m[fmt.Sprintf("%s.%s", d.Namespace, d.Name)] = d
}
return m
}
func kafkaList(m map[string]v1beta2.Kafka) []v1beta2.Kafka {
l := []v1beta2.Kafka{}
for _, v := range m {
l = append(l, v)
}
return l
}