Skip to content

Commit 62a3d26

Browse files
committed
Move label definition to constants package
1 parent 2553634 commit 62a3d26

5 files changed

+11
-11
lines changed

controllers/opentelemetrycollector_controller.go

+3-5
Original file line numberDiff line numberDiff line change
@@ -48,11 +48,10 @@ import (
4848
"github.com/open-telemetry/opentelemetry-operator/internal/manifests/collector"
4949
"github.com/open-telemetry/opentelemetry-operator/internal/manifests/manifestutils"
5050
collectorStatus "github.com/open-telemetry/opentelemetry-operator/internal/status/collector"
51+
"github.com/open-telemetry/opentelemetry-operator/pkg/constants"
5152
"github.com/open-telemetry/opentelemetry-operator/pkg/featuregate"
5253
)
5354

54-
const collectorTargetAllocatorLabelName = "opentelemetry.io/target-allocator"
55-
5655
var (
5756
ownedClusterObjectTypes = []client.Object{
5857
&rbacv1.ClusterRole{},
@@ -191,10 +190,9 @@ func (r *OpenTelemetryCollectorReconciler) GetParams(ctx context.Context, instan
191190
}
192191

193192
func (r *OpenTelemetryCollectorReconciler) getTargetAllocator(ctx context.Context, params manifests.Params) (*v1alpha1.TargetAllocator, error) {
194-
otelcol := params.OtelCol
195-
if taName, ok := otelcol.GetLabels()[collectorTargetAllocatorLabelName]; ok {
193+
if taName, ok := params.OtelCol.GetLabels()[constants.LabelTargetAllocator]; ok {
196194
targetAllocator := &v1alpha1.TargetAllocator{}
197-
taKey := client.ObjectKey{Name: taName, Namespace: otelcol.GetNamespace()}
195+
taKey := client.ObjectKey{Name: taName, Namespace: params.OtelCol.GetNamespace()}
198196
err := r.Client.Get(ctx, taKey, targetAllocator)
199197
if err != nil {
200198
return nil, err

controllers/suite_test.go

-1
Original file line numberDiff line numberDiff line change
@@ -64,7 +64,6 @@ import (
6464
"github.com/open-telemetry/opentelemetry-operator/internal/manifests/collector/testdata"
6565
"github.com/open-telemetry/opentelemetry-operator/internal/manifests/manifestutils"
6666
"github.com/open-telemetry/opentelemetry-operator/internal/rbac"
67-
// +kubebuilder:scaffold:imports
6867
)
6968

7069
var (

controllers/targetallocator_controller.go

+4-3
Original file line numberDiff line numberDiff line change
@@ -42,6 +42,7 @@ import (
4242
"github.com/open-telemetry/opentelemetry-operator/internal/config"
4343
"github.com/open-telemetry/opentelemetry-operator/internal/manifests/targetallocator"
4444
taStatus "github.com/open-telemetry/opentelemetry-operator/internal/status/targetallocator"
45+
"github.com/open-telemetry/opentelemetry-operator/pkg/constants"
4546
"github.com/open-telemetry/opentelemetry-operator/pkg/featuregate"
4647
)
4748

@@ -102,7 +103,7 @@ func (r *TargetAllocatorReconciler) getCollector(ctx context.Context, instance v
102103
listOpts := []client.ListOption{
103104
client.InNamespace(instance.GetNamespace()),
104105
client.MatchingLabels{
105-
collectorTargetAllocatorLabelName: instance.GetName(),
106+
constants.LabelTargetAllocator: instance.GetName(),
106107
},
107108
}
108109
err := r.List(ctx, &collectors, listOpts...)
@@ -216,7 +217,7 @@ func (r *TargetAllocatorReconciler) SetupWithManager(mgr ctrl.Manager) error {
216217
collectorSelector := metav1.LabelSelector{
217218
MatchExpressions: []metav1.LabelSelectorRequirement{
218219
{
219-
Key: collectorTargetAllocatorLabelName,
220+
Key: constants.LabelTargetAllocator,
220221
Operator: metav1.LabelSelectorOpExists,
221222
},
222223
},
@@ -246,7 +247,7 @@ func getTargetAllocatorForCollector(_ context.Context, collector client.Object)
246247
}
247248

248249
func getTargetAllocatorRequestsFromLabel(_ context.Context, collector client.Object) []reconcile.Request {
249-
if taName, ok := collector.GetLabels()[collectorTargetAllocatorLabelName]; ok {
250+
if taName, ok := collector.GetLabels()[constants.LabelTargetAllocator]; ok {
250251
return []reconcile.Request{
251252
{
252253
NamespacedName: types.NamespacedName{

controllers/targetallocator_reconciler_test.go

+3-2
Original file line numberDiff line numberDiff line change
@@ -36,6 +36,7 @@ import (
3636
"github.com/open-telemetry/opentelemetry-operator/apis/v1alpha1"
3737
"github.com/open-telemetry/opentelemetry-operator/apis/v1beta1"
3838
"github.com/open-telemetry/opentelemetry-operator/internal/config"
39+
"github.com/open-telemetry/opentelemetry-operator/pkg/constants"
3940
)
4041

4142
var testLogger = logf.Log.WithName("opamp-bridge-controller-unit-tests")
@@ -57,7 +58,7 @@ func TestTargetAllocatorReconciler_GetCollector(t *testing.T) {
5758
ObjectMeta: metav1.ObjectMeta{
5859
Name: "test",
5960
Labels: map[string]string{
60-
collectorTargetAllocatorLabelName: "label-ta",
61+
constants.LabelTargetAllocator: "label-ta",
6162
},
6263
},
6364
}
@@ -163,7 +164,7 @@ func TestGetTargetAllocatorRequestsFromLabel(t *testing.T) {
163164
Name: "test",
164165
Namespace: "default",
165166
Labels: map[string]string{
166-
collectorTargetAllocatorLabelName: "label-ta",
167+
constants.LabelTargetAllocator: "label-ta",
167168
},
168169
},
169170
}

pkg/constants/env.go

+1
Original file line numberDiff line numberDiff line change
@@ -40,6 +40,7 @@ const (
4040
LabelAppVersion = "app.kubernetes.io/version"
4141
LabelAppPartOf = "app.kubernetes.io/part-of"
4242

43+
LabelTargetAllocator = "opentelemetry.io/target-allocator"
4344
ResourceAttributeAnnotationPrefix = "resource.opentelemetry.io/"
4445

4546
EnvPodName = "OTEL_RESOURCE_ATTRIBUTES_POD_NAME"

0 commit comments

Comments
 (0)