fix: close piper if watcher down

quickube · Jan 11, 2025 · 399a7e6 · 399a7e6
1 parent d869b7c
commit 399a7e6
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/cmd/piper/piper.go b/cmd/piper/piper.go
@@ -51,6 +51,6 @@ func main() {
 	// Create context that listens for the interrupt signal from the OS.
 	ctx, stop := signal.NotifyContext(context.Background(), syscall.SIGINT, syscall.SIGTERM)
 	defer stop()
-	event_handler.Start(ctx, cfg, globalClients)
+	event_handler.Start(ctx, stop, cfg, globalClients)
 	server.Start(ctx, stop, cfg, globalClients)
 }
diff --git a/pkg/event_handler/main.go b/pkg/event_handler/main.go
@@ -8,7 +8,7 @@ import (
 	"log"
 )
 
-func Start(ctx context.Context, cfg *conf.GlobalConfig, clients *clients.Clients) {
+func Start(ctx context.Context, stop context.CancelFunc, cfg *conf.GlobalConfig, clients *clients.Clients) {
 	labelSelector := &metav1.LabelSelector{
 		MatchExpressions: []metav1.LabelSelectorRequirement{
 			{Key: "piper.quickube.com/notified",
@@ -37,6 +37,7 @@ func Start(ctx context.Context, cfg *conf.GlobalConfig, clients *clients.Clients
 				if !ok {
 					log.Print("[event handler] result channel closed")
 					watcher.Stop()
+					stop()
 					return
 				}
 				if err2 := handler.Handle(ctx, &event); err2 != nil {