fix: exit leader-election when command fails

main.go

@@ -250,7 +250,6 @@ func (el *AwaitElection) createSessionAndElection(ctx context.Context) error {
 func (el *AwaitElection) Run() error {
 	for {
 		ctx, cancel := context.WithCancel(context.Background())
-		defer cancel()
 
 		// Initialize session and election
 		err := el.createSessionAndElection(ctx)
@@ -274,14 +273,15 @@ func (el *AwaitElection) Run() error {
 				err = el.LeaderExec(ctx)
 				if err != nil {
 					log.Printf("Leader execution error: %v", err)
+					cancel()
+					return err
 				}
 				return err
 			} else {
 				log.Printf("Current leader is %s, not %s, will continue trying.", currentLeader, el.LeaderIdentity)
 			}
 		} else {
 			log.Printf("Failed to get current leader: %v, will campaign.", err)
-			// Continue with campaigning
 		}
 
 		err = el.Election.Campaign(ctx, el.LeaderIdentity)
@@ -297,6 +297,8 @@ func (el *AwaitElection) Run() error {
 		err = el.LeaderExec(ctx)
 		if err != nil {
 			log.Printf("Leader execution error: %v", err)
+			cancel()
+			return err
 		}
 
 		select {
@@ -306,7 +308,6 @@ func (el *AwaitElection) Run() error {
 			log.Printf("Session expired or lost, losing leadership")
 			cancel()
 			time.Sleep(el.RetryPeriod)
-			// Starting new election cycle
 		}
 	}
 }