From 49d60ff1b4e898fa7c99372b523c519ecadde0cb Mon Sep 17 00:00:00 2001
From: Alexseij <79832610507@yandex.ru>
Date: Sun, 7 Jul 2024 02:04:11 +0700
Subject: [PATCH] add function initLeaderElector

---
 pkg/manager/internal.go | 67 +++++++++++++++++++++++------------------
 1 file changed, 38 insertions(+), 29 deletions(-)

diff --git a/pkg/manager/internal.go b/pkg/manager/internal.go
index 8182969e6d..d2f8c4bac3 100644
--- a/pkg/manager/internal.go
+++ b/pkg/manager/internal.go
@@ -353,35 +353,7 @@ func (cm *controllerManager) Start(ctx context.Context) (err error) {
 
 	var leaderElector *leaderelection.LeaderElector
 	if cm.resourceLock != nil {
-		leaderElector, err = leaderelection.NewLeaderElector(leaderelection.LeaderElectionConfig{
-			Lock:          cm.resourceLock,
-			LeaseDuration: cm.leaseDuration,
-			RenewDeadline: cm.renewDeadline,
-			RetryPeriod:   cm.retryPeriod,
-			Callbacks: leaderelection.LeaderCallbacks{
-				OnStartedLeading: func(_ context.Context) {
-					if err := cm.startLeaderElectionRunnables(); err != nil {
-						cm.errChan <- err
-						return
-					}
-					close(cm.elected)
-				},
-				OnStoppedLeading: func() {
-					if cm.onStoppedLeading != nil {
-						cm.onStoppedLeading()
-					}
-					// Make sure graceful shutdown is skipped if we lost the leader lock without
-					// intending to.
-					cm.gracefulShutdownTimeout = time.Duration(0)
-					// Most implementations of leader election log.Fatal() here.
-					// Since Start is wrapped in log.Fatal when called, we can just return
-					// an error here which will cause the program to exit.
-					cm.errChan <- errors.New("leader election lost")
-				},
-			},
-			ReleaseOnCancel: cm.leaderElectionReleaseOnCancel,
-			Name:            cm.leaderElectionID,
-		})
+		leaderElector, err = cm.initLeaderElector()
 		if err != nil {
 			return fmt.Errorf("failed during initialization leader election process: %w", err)
 		}
@@ -603,6 +575,43 @@ func (cm *controllerManager) engageStopProcedure(stopComplete <-chan struct{}) e
 	return nil
 }
 
+func (cm *controllerManager) initLeaderElector() (*leaderelection.LeaderElector, error) {
+	leaderElector, err := leaderelection.NewLeaderElector(leaderelection.LeaderElectionConfig{
+		Lock:          cm.resourceLock,
+		LeaseDuration: cm.leaseDuration,
+		RenewDeadline: cm.renewDeadline,
+		RetryPeriod:   cm.retryPeriod,
+		Callbacks: leaderelection.LeaderCallbacks{
+			OnStartedLeading: func(_ context.Context) {
+				if err := cm.startLeaderElectionRunnables(); err != nil {
+					cm.errChan <- err
+					return
+				}
+				close(cm.elected)
+			},
+			OnStoppedLeading: func() {
+				if cm.onStoppedLeading != nil {
+					cm.onStoppedLeading()
+				}
+				// Make sure graceful shutdown is skipped if we lost the leader lock without
+				// intending to.
+				cm.gracefulShutdownTimeout = time.Duration(0)
+				// Most implementations of leader election log.Fatal() here.
+				// Since Start is wrapped in log.Fatal when called, we can just return
+				// an error here which will cause the program to exit.
+				cm.errChan <- errors.New("leader election lost")
+			},
+		},
+		ReleaseOnCancel: cm.leaderElectionReleaseOnCancel,
+		Name:            cm.leaderElectionID,
+	})
+	if err != nil {
+		return nil, err
+	}
+
+	return leaderElector, nil
+}
+
 func (cm *controllerManager) startLeaderElectionRunnables() error {
 	return cm.runnables.LeaderElection.Start(cm.internalCtx)
 }