wait until next boundary

Author: zedimytch

leader/leader.go

@@ -71,18 +71,23 @@ func (e *elector) Run(ctx context.Context, lockKey uint32, period time.Duration,
 		tx, acquired := e.tryAdvisoryLock(workCtx, lockKey)
 		if !acquired {
 			workCancel()
+			waitUntilNextBoundary(ctx, period)
 			continue
 		}
 
 		onWork(workCtx)
 		tx.Rollback()
 		workCancel()
 
-		nextBoundary := time.Now().Truncate(period).Add(period)
-		select {
-		case <-ctx.Done():
-			return
-		case <-time.After(time.Until(nextBoundary)):
-		}
+		waitUntilNextBoundary(ctx, period)
+	}
+}
+
+func waitUntilNextBoundary(ctx context.Context, period time.Duration) {
+	nextBoundary := time.Now().Truncate(period).Add(period)
+	select {
+	case <-ctx.Done():
+		return
+	case <-time.After(time.Until(nextBoundary)):
 	}
 }