fix poller not restarting after indexer mode changes

Author: iuwqyir

internal/orchestrator/poller.go

@@ -29,6 +29,8 @@ type Poller struct {
 	pollUntilBlock    *big.Int
 	parallelPollers   int
 	workModeChan      chan WorkMode
+	currentWorkMode   WorkMode
+	workModeMutex     sync.RWMutex
 }
 
 type BlockNumberWithError struct {
@@ -117,6 +119,15 @@ func (p *Poller) Start(ctx context.Context) {
 					if !ok {
 						return
 					}
+
+					// Do not poll if not in backfill mode
+					p.workModeMutex.RLock()
+					if p.currentWorkMode != WorkModeBackfill {
+						p.workModeMutex.RUnlock()
+						continue
+					}
+					p.workModeMutex.RUnlock()
+
 					blockRangeMutex.Lock()
 					blockNumbers, err := p.getNextBlockRange(pollCtx)
 					blockRangeMutex.Unlock()
@@ -149,10 +160,24 @@ func (p *Poller) Start(ctx context.Context) {
 			p.shutdown(cancel, tasks, &wg)
 			return
 		case workMode := <-p.workModeChan:
-			if workMode == WorkModeLive {
-				log.Info().Msg("Switching to live mode, stopping poller")
-				p.shutdown(cancel, tasks, &wg)
-				return
+			p.workModeMutex.RLock()
+			currentWorkMode := p.currentWorkMode
+			p.workModeMutex.RUnlock()
+			if workMode != currentWorkMode && workMode != "" {
+				log.Info().Msgf("Poller work mode changing from %s to %s", currentWorkMode, workMode)
+				p.workModeMutex.Lock()
+				changedToBackfillFromLive := currentWorkMode == WorkModeLive && workMode == WorkModeBackfill
+				p.currentWorkMode = workMode
+				p.workModeMutex.Unlock()
+				if changedToBackfillFromLive {
+					lastBlockInMainStorage, err := p.storage.MainStorage.GetMaxBlockNumber(p.rpc.GetChainID())
+					if err != nil {
+						log.Error().Err(err).Msg("Error getting last block in main storage")
+					} else {
+						p.lastPolledBlock = lastBlockInMainStorage
+						log.Debug().Msgf("Switching to backfill mode, updating last polled block to %s", p.lastPolledBlock.String())
+					}
+				}
 			}
 		case <-ticker.C:
 			select {