fix: add ctx to reorg detector start function

0xPolygon · Aug 28, 2024 · 755252c · 755252c
1 parent 97addb4
commit 755252c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/cmd/main.go b/cmd/main.go
@@ -127,7 +127,7 @@ func start(cliCtx *cli.Context) error {
 		log.Fatal(err)
 	}
 
-	if err = detector.Start(); err != nil {
+	if err = detector.Start(cliCtx.Context); err != nil {
 		log.Fatal(err)
 	}
 

diff --git a/synchronizer/reorg.go b/synchronizer/reorg.go
@@ -41,10 +41,10 @@ func (rd *ReorgDetector) Subscribe() <-chan BlockReorg {
 }
 
 // Start starts the ReorgDetector tracking for reorg events
-func (rd *ReorgDetector) Start() error {
+func (rd *ReorgDetector) Start(ctx context.Context) error {
 	log.Info("starting block reorganization detector")
 
-	ctx, cancel := context.WithCancel(context.Background())
+	ctx, cancel := context.WithCancel(ctx)
 	rd.cancel = cancel
 
 	blocks := make(chan *ethgo.Block)