Migrate odigos-instrumentation label to Source CRD

instrumentor/controllers/startlangdetection/manager.go

@@ -14,7 +14,9 @@ func SetupWithManager(mgr ctrl.Manager) error {
 		ControllerManagedBy(mgr).
 		Named("startlangdetection-deployment").
 		For(&appsv1.Deployment{}).
-		WithEventFilter(&WorkloadAvailablePredicate{}).
+		WithEventFilter(&WorkloadAvailablePredicate{
+			Client: mgr.GetClient(),
+		}).
 		Complete(&DeploymentReconciler{
 			Client: mgr.GetClient(),
 			Scheme: mgr.GetScheme(),
@@ -27,7 +29,9 @@ func SetupWithManager(mgr ctrl.Manager) error {
 		ControllerManagedBy(mgr).
 		Named("startlangdetection-daemonset").
 		For(&appsv1.DaemonSet{}).
-		WithEventFilter(&WorkloadAvailablePredicate{}).
+		WithEventFilter(&WorkloadAvailablePredicate{
+			Client: mgr.GetClient(),
+		}).
 		Complete(&DaemonSetReconciler{
 			Client: mgr.GetClient(),
 			Scheme: mgr.GetScheme(),
@@ -40,7 +44,9 @@ func SetupWithManager(mgr ctrl.Manager) error {
 		ControllerManagedBy(mgr).
 		Named("startlangdetection-statefulset").
 		For(&appsv1.StatefulSet{}).
-		WithEventFilter(&WorkloadAvailablePredicate{}).
+		WithEventFilter(&WorkloadAvailablePredicate{
+			Client: mgr.GetClient(),
+		}).
 		Complete(&StatefulSetReconciler{
 			Client: mgr.GetClient(),
 			Scheme: mgr.GetScheme(),

instrumentor/controllers/startlangdetection/workload_predicate.go

@@ -1,23 +1,35 @@
 package startlangdetection
 
 import (
+	"context"
+
 	"github.com/odigos-io/odigos/k8sutils/pkg/env"
 	"github.com/odigos-io/odigos/k8sutils/pkg/workload"
+	"sigs.k8s.io/controller-runtime/pkg/client"
 	"sigs.k8s.io/controller-runtime/pkg/event"
 	"sigs.k8s.io/controller-runtime/pkg/predicate"
+
+	sourceutils "github.com/odigos-io/odigos/k8sutils/pkg/source"
 )
 
 // this predicate is used for workload reconciler to only pass events
 // where the workload has replicas available to instrument.
 type WorkloadAvailablePredicate struct {
 	predicate.Funcs
+	Client client.Client
 }
 
 func (i *WorkloadAvailablePredicate) Create(e event.CreateEvent) bool {
 	w, err := workload.ObjectToWorkload(e.Object)
 	if err != nil {
 		return false
 	}
+
+	instrumented, _ := sourceutils.IsObjectInstrumentedBySource(context.Background(), i.Client, e.Object)
+	if !instrumented || !workload.IsObjectLabeledForInstrumentation(e.Object) {
+		return false
+	}
+
 	return w.AvailableReplicas() > 0
 }
 
@@ -43,6 +55,11 @@ func (i *WorkloadAvailablePredicate) Update(e event.UpdateEvent) bool {
 		return false
 	}
 
+	instrumented, _ := sourceutils.IsObjectInstrumentedBySource(context.Background(), i.Client, e.ObjectNew)
+	if !instrumented || !workload.IsObjectLabeledForInstrumentation(e.ObjectNew) {
+		return false
+	}
+
 	newReplicas := wNew.AvailableReplicas()
 	oldReplicas := wOld.AvailableReplicas()