cancel contexts

ava-labs · Oct 15, 2024 · 67c8dda · 67c8dda
1 parent f1853d5
commit 67c8dda
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/relayer/main/main.go b/relayer/main/main.go
@@ -231,7 +231,10 @@ func main() {
 		logger.Fatal("Failed to create proposer height cache", zap.Error(err))
 		panic(err)
 	}
-	proposerHeightCache.Start(context.Background())
+
+	proposerHeightCacheCtx, cancelFunc := context.WithCancel(context.Background())
+	defer cancelFunc()
+	proposerHeightCache.Start(proposerHeightCacheCtx)
 
 	signatureAggregator, err := aggregator.NewSignatureAggregator(
 		network,

diff --git a/signature-aggregator/main/main.go b/signature-aggregator/main/main.go
@@ -124,7 +124,9 @@ func main() {
 		logger.Fatal("Failed to create proposer height cache", zap.Error(err))
 		panic(err)
 	}
-	proposerHeightCache.Start(context.Background())
+	ctx, cancelFunc := context.WithCancel(context.Background())
+	defer cancelFunc()
+	proposerHeightCache.Start(ctx)
 
 	signatureAggregator, err := aggregator.NewSignatureAggregator(
 		network,