fix: quit when loosing leadership (#6718)

Signed-off-by: Charles-Edouard Brétéché <charles.edouard@nirmata.com>
Co-authored-by: shuting <shuting@nirmata.com>

cmd/background-controller/main.go

@@ -309,13 +309,6 @@ func main() {
 		controller.Run(signalCtx, logger.WithName("controllers"), &wg)
 	}
 	// start leader election
-	for {
+	le.Run(signalCtx)
-		select {
+	wg.Wait()
-		case <-signalCtx.Done():
-			wg.Wait()
-			return
-		default:
-			le.Run(signalCtx)
-		}
-	}
 }

cmd/cleanup-controller/main.go

@@ -229,13 +229,6 @@ func main() {
 	)
 	// start server
 	server.Run(ctx.Done())
-	// wait for termination signal and run leader election loop
+	// start leader election
-	for {
+	le.Run(ctx)
-		select {
-		case <-ctx.Done():
-			return
-		default:
-			le.Run(ctx)
-		}
-	}
 }

cmd/reports-controller/main.go

@@ -415,13 +415,6 @@ func main() {
 	for _, controller := range nonLeaderControllers {
 		controller.Run(ctx, logger.WithName("controllers"), &wg)
 	}
-	for {
+	le.Run(ctx)
-		select {
+	wg.Wait()
-		case <-ctx.Done():
-			wg.Wait()
-			return
-		default:
-			le.Run(ctx)
-		}
-	}
 }