refactor: update GetEventChannel to accept context and stop channel f…

…or better event handling

Signed-off-by: slashexx <[email protected]>

KubeArmor/core/dockerHandler.go

@@ -215,17 +215,22 @@ func (dh *DockerHandler) GetContainerInfo(containerID string, OwnerInfo map[stri
 // ========================== //
 
 // GetEventChannel Function
-func (dh *DockerHandler) GetEventChannel() <-chan events.Message {
+func (dh *DockerHandler) GetEventChannel(ctx context.Context, StopChan <- chan struct{}) <-chan events.Message {
 	if dh.DockerClient != nil {
 		eventBuffer := make(chan events.Message, 256)
 
 		go func() {
-			eventStream, _ := dh.DockerClient.Events(context.Background(), events.ListOptions{})
+
+			eventStream, _ := dh.DockerClient.Events(ctx, events.ListOptions{})
 			defer close(eventBuffer)
 
 			for event := range eventStream {
 				select {
 				case eventBuffer <- event:
+				case <-ctx.Done():
+					return
+				case <-StopChan:
+					return
 				default:
 					kg.Warnf("Docker channel full.")
 				}
@@ -765,7 +770,11 @@ func (dm *KubeArmorDaemon) MonitorDockerEvents() {
 
 	dm.Logger.Print("Started to monitor Docker events")
 
-	EventChan := Docker.GetEventChannel()
+	ctx, cancel := context.WithCancel(context.Background())
+	defer cancel()
+
+
+	EventChan := Docker.GetEventChannel(ctx, StopChan)
 
 	for {
 		select {