Merge pull request #6029 from mysteriumnetwork/add-more-context-timeouts

Add more timeouts to endless contexts
mysteriumnetwork · Apr 4, 2024 · 41bffb1 · 41bffb1
2 parents 6720107 + 1882dab
commit 41bffb1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/session/pingpong/consumer_balance_tracker.go b/session/pingpong/consumer_balance_tracker.go
@@ -348,7 +348,7 @@ func (cbt *ConsumerBalanceTracker) alignWithHermes(chainID int64, id identity.Id
 	eback.InitialInterval = time.Second * 1
 
 	boff = backoff.WithMaxRetries(eback, 5)
-	ctx, cancel := context.WithCancel(context.Background())
+	ctx, cancel := context.WithTimeout(context.Background(), 60*time.Second)
 	defer cancel()
 
 	go func() {
@@ -634,7 +634,7 @@ func (cbt *ConsumerBalanceTracker) recoverGrandTotalPromised(chainID int64, iden
 	eback.InitialInterval = time.Second * 2
 
 	boff = backoff.WithMaxRetries(eback, 10)
-	ctx, cancel := context.WithCancel(context.Background())
+	ctx, cancel := context.WithTimeout(context.Background(), 60*time.Second)
 	defer cancel()
 
 	go func() {