diff --git a/cmd/main.go b/cmd/main.go index b18bf6c..409c2fd 100644 --- 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 index 71725b1..af2d79d 100644 --- 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)