added container receivers

Signed-off-by: Afek Berger <[email protected]>

Author: afek854

pkg/containerwatcher/container_watcher_interface.go

@@ -23,6 +23,8 @@ type ContainerWatcher interface {
 	GetContainerCollection() *containercollection.ContainerCollection
 	GetSocketEnricher() *socketenricher.SocketEnricher
 	GetContainerSelector() *containercollection.ContainerSelector
+	RegisterContainerReceiver(receiver ContainerReceiver)
+	UnregisterContainerReceiver(receiver ContainerReceiver)
 }
 
 type CustomTracerInitializer interface {

pkg/containerwatcher/v2/container_watcher.go

@@ -76,6 +76,7 @@ type ContainerWatcher struct {
 	// Third party components
 	thirdPartyTracersInitializers mapset.Set[containerwatcher.CustomTracerInitializer]
 	thirdPartyEnricher            containerwatcher.TaskBasedEnricher
+	thirdPartyContainerReceivers mapset.Set[containerwatcher.ContainerReceiver]
 
 	// Cache and state
 	objectCache          objectcache.ObjectCache
@@ -380,6 +381,15 @@ func (cw *ContainerWatcher) GetContainerSelector() *containercollection.Containe
 	return &cw.containerSelector
 }
 
+func (cw *ContainerWatcher) RegisterContainerReceiver(receiver containerwatcher.ContainerReceiver) {
+	cw.thirdPartyContainerReceivers.Add(receiver)
+}
+
+// UnregisterContainerReceiver unregisters a container receiver
+func (cw *ContainerWatcher) UnregisterContainerReceiver(receiver containerwatcher.ContainerReceiver) {
+	cw.thirdPartyContainerReceivers.Remove(receiver)
+}
+
 func (cw *ContainerWatcher) eventProcessingLoop() {
 	ticker := time.NewTicker(50 * time.Millisecond)
 	defer ticker.Stop()

pkg/containerwatcher/v2/container_watcher_collection.go

@@ -57,6 +57,10 @@ func (cw *ContainerWatcher) StartContainerCollection(ctx context.Context) error
 		cw.networkStreamClient.ContainerCallback,
 	}
 
+	for receiver := range cw.thirdPartyContainerReceivers.Iter() {
+		cw.callbacks = append(cw.callbacks, receiver.ContainerCallback)
+	}
+
 	// Define the different options for the container collection instance
 	opts := []containercollection.ContainerCollectionOption{
 		// Get Notifications from the container collection