-
Notifications
You must be signed in to change notification settings - Fork 396
/
collector.go
102 lines (86 loc) · 3.46 KB
/
collector.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
// Copyright The OpenTelemetry Authors
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package collector
import (
"context"
"fmt"
"strconv"
appsv1 "k8s.io/api/apps/v1"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
"sigs.k8s.io/controller-runtime/pkg/client"
"github.com/open-telemetry/opentelemetry-operator/apis/v1beta1"
"github.com/open-telemetry/opentelemetry-operator/internal/manifests/collector"
"github.com/open-telemetry/opentelemetry-operator/internal/manifests/manifestutils"
"github.com/open-telemetry/opentelemetry-operator/internal/naming"
"github.com/open-telemetry/opentelemetry-operator/internal/version"
)
func UpdateCollectorStatus(ctx context.Context, cli client.Client, changed *v1beta1.OpenTelemetryCollector) error {
if changed.Status.Version == "" {
// a version is not set, otherwise let the upgrade mechanism take care of it!
changed.Status.Version = version.OpenTelemetryCollector()
}
mode := changed.Spec.Mode
if mode == v1beta1.ModeSidecar {
changed.Status.Scale.Replicas = 0
changed.Status.Scale.Selector = ""
return nil
}
name := naming.Collector(changed.Name)
// Set the scale selector
labels := manifestutils.Labels(changed.ObjectMeta, name, changed.Spec.Image, collector.ComponentOpenTelemetryCollector, []string{})
selector, err := metav1.LabelSelectorAsSelector(&metav1.LabelSelector{MatchLabels: labels})
if err != nil {
return fmt.Errorf("failed to get selector for labelSelector: %w", err)
}
changed.Status.Scale.Selector = selector.String()
// Set the scale replicas
objKey := client.ObjectKey{
Namespace: changed.GetNamespace(),
Name: naming.Collector(changed.Name),
}
var replicas int32
var readyReplicas int32
var statusReplicas string
var statusImage string
switch mode { // nolint:exhaustive
case v1beta1.ModeDeployment:
obj := &appsv1.Deployment{}
if err := cli.Get(ctx, objKey, obj); err != nil {
return fmt.Errorf("failed to get deployment status.replicas: %w", err)
}
replicas = obj.Status.Replicas
readyReplicas = obj.Status.ReadyReplicas
statusReplicas = strconv.Itoa(int(readyReplicas)) + "/" + strconv.Itoa(int(replicas))
statusImage = obj.Spec.Template.Spec.Containers[0].Image
case v1beta1.ModeStatefulSet:
obj := &appsv1.StatefulSet{}
if err := cli.Get(ctx, objKey, obj); err != nil {
return fmt.Errorf("failed to get statefulSet status.replicas: %w", err)
}
replicas = obj.Status.Replicas
readyReplicas = obj.Status.ReadyReplicas
statusReplicas = strconv.Itoa(int(readyReplicas)) + "/" + strconv.Itoa(int(replicas))
statusImage = obj.Spec.Template.Spec.Containers[0].Image
case v1beta1.ModeDaemonSet:
obj := &appsv1.DaemonSet{}
if err := cli.Get(ctx, objKey, obj); err != nil {
return fmt.Errorf("failed to get daemonSet status.replicas: %w", err)
}
statusImage = obj.Spec.Template.Spec.Containers[0].Image
}
changed.Status.Scale.Replicas = replicas
changed.Status.Image = statusImage
changed.Status.Scale.StatusReplicas = statusReplicas
return nil
}