Check for cancellation error (#1346)

Snowfork · Dec 5, 2024 · f9b3fc2 · f9b3fc2
1 parent 11eb194
commit f9b3fc2
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/relayer/relays/beefy/on-demand-sync.go b/relayer/relays/beefy/on-demand-sync.go
@@ -2,6 +2,7 @@ package beefy
 
 import (
 	"context"
+	"errors"
 	"fmt"
 	"time"
 
@@ -94,6 +95,9 @@ func (relay *OnDemandRelay) Start(ctx context.Context) error {
 
 		paraNonce, ethNonce, err := relay.queryNonces(ctx)
 		if err != nil {
+			if errors.Is(err, context.Canceled) {
+				return nil
+			}
 			log.WithError(err).Error("Query nonces")
 			continue
 		}
@@ -115,18 +119,27 @@ func (relay *OnDemandRelay) Start(ctx context.Context) error {
 
 			beefyBlockHash, err := relay.relaychainConn.API().RPC.Beefy.GetFinalizedHead()
 			if err != nil {
+				if errors.Is(err, context.Canceled) {
+					return nil
+				}
 				log.WithError(err).Error("Fetch latest beefy block hash")
 				continue
 			}
 
 			header, err := relay.relaychainConn.API().RPC.Chain.GetHeader(beefyBlockHash)
 			if err != nil {
+				if errors.Is(err, context.Canceled) {
+					return nil
+				}
 				log.WithError(err).Error("Fetch latest beefy block header")
 				continue
 			}
 
 			err = relay.sync(ctx, uint64(header.Number))
 			if err != nil {
+				if errors.Is(err, context.Canceled) {
+					return nil
+				}
 				log.WithError(err).Error("Sync failed")
 				continue
 			}
@@ -143,6 +156,9 @@ func (relay *OnDemandRelay) waitUntilMessagesSynced(ctx context.Context, paraNon
 	for {
 		ethNonce, err := relay.fetchEthereumNonce(ctx)
 		if err != nil {
+			if errors.Is(err, context.Canceled) {
+				return
+			}
 			log.WithError(err).Error("fetch latest ethereum nonce")
 			sleep(ctx, time.Minute*1)
 			continue