improved waiting logic

Snowfork · Dec 5, 2024 · ea64829 · ea64829
1 parent 252e6b2
commit ea64829
Showing 1 changed file with 20 additions and 3 deletions.
diff --git a/relayer/relays/beefy/on-demand-sync.go b/relayer/relays/beefy/on-demand-sync.go
@@ -88,6 +88,7 @@ func (relay *OnDemandRelay) Start(ctx context.Context) error {
 	relay.tokenBucket.Start(ctx)
 
 	for {
+		sleep(ctx, time.Minute*1)
 		log.Info("Starting check")
 
 		paraNonce, ethNonce, err := relay.queryNonces(ctx)
@@ -105,7 +106,7 @@ func (relay *OnDemandRelay) Start(ctx context.Context) error {
 
 			// Check if we are rate-limited
 			if !relay.tokenBucket.TryConsume(1) {
-				sleep(ctx, time.Minute*1)
+				log.Info("Rate-limit exceeded")
 				continue
 			}
 
@@ -131,12 +132,28 @@ func (relay *OnDemandRelay) Start(ctx context.Context) error {
 
 			log.Info("Sync completed")
 
-			// Sleep for 10 minute to allow message relayer to sync nonces
+			// Sleep for 10 minutes to allow message relayer to sync nonces
 			sleep(ctx, time.Minute*10)
-		} else {
+
+			relay.waitUntilMessagesSynced(ctx, paraNonce)
+		}
+	}
+}
+
+func (relay *OnDemandRelay) waitUntilMessagesSynced(ctx context.Context, paraNonce uint64) {
+	for {
+		ethNonce, err := relay.fetchEthereumNonce(ctx)
+		if err != nil {
+			log.WithError(err).Error("fetch latest ethereum nonce")
 			sleep(ctx, time.Minute*1)
+			continue
+		}
+
+		if ethNonce >= paraNonce {
+			return
 		}
 	}
+
 }
 
 func sleep(ctx context.Context, d time.Duration) {