From f0b2a686c81d0c6591cef1ba8c268256c9a518cf Mon Sep 17 00:00:00 2001
From: Rowan Seymour <rowanseymour@gmail.com>
Date: Tue, 29 Mar 2022 14:32:00 -0500
Subject: [PATCH] Poll interval is configurable

---
 cmd/rp-indexer/main.go |  3 ++-
 daemon.go              | 10 ++++++----
 2 files changed, 8 insertions(+), 5 deletions(-)

diff --git a/cmd/rp-indexer/main.go b/cmd/rp-indexer/main.go
index b959c05..69ff29a 100644
--- a/cmd/rp-indexer/main.go
+++ b/cmd/rp-indexer/main.go
@@ -5,6 +5,7 @@ import (
 	"os"
 	"os/signal"
 	"syscall"
+	"time"
 
 	"github.com/evalphobia/logrus_sentry"
 	_ "github.com/lib/pq"
@@ -59,7 +60,7 @@ func main() {
 			log.WithField("indexer", idxr.Name()).WithError(err).Fatal("error during rebuilding")
 		}
 	} else {
-		d := indexer.NewDaemon(cfg, db, idxrs)
+		d := indexer.NewDaemon(cfg, db, idxrs, time.Duration(cfg.Poll)*time.Second)
 		d.Start()
 
 		handleSignals(d)
diff --git a/daemon.go b/daemon.go
index 19b78d5..b57b7d1 100644
--- a/daemon.go
+++ b/daemon.go
@@ -16,18 +16,20 @@ type Daemon struct {
 	wg       *sync.WaitGroup
 	quit     chan bool
 	indexers []indexers.Indexer
+	poll     time.Duration
 
 	prevStats map[indexers.Indexer]indexers.Stats
 }
 
 // NewDaemon creates a new daemon to run the given indexers
-func NewDaemon(cfg *Config, db *sql.DB, ixs []indexers.Indexer) *Daemon {
+func NewDaemon(cfg *Config, db *sql.DB, ixs []indexers.Indexer, poll time.Duration) *Daemon {
 	return &Daemon{
 		cfg:       cfg,
 		db:        db,
 		wg:        &sync.WaitGroup{},
 		quit:      make(chan bool),
 		indexers:  ixs,
+		poll:      poll,
 		prevStats: make(map[indexers.Indexer]indexers.Stats, len(ixs)),
 	}
 }
@@ -41,13 +43,13 @@ func (d *Daemon) Start() {
 	}
 
 	for _, i := range d.indexers {
-		d.startIndexer(i, time.Second*5)
+		d.startIndexer(i)
 	}
 
 	d.startStatsReporter(time.Minute)
 }
 
-func (d *Daemon) startIndexer(indexer indexers.Indexer, interval time.Duration) {
+func (d *Daemon) startIndexer(indexer indexers.Indexer) {
 	d.wg.Add(1) // add ourselves to the wait group
 
 	log := logrus.WithField("indexer", indexer.Name())
@@ -62,7 +64,7 @@ func (d *Daemon) startIndexer(indexer indexers.Indexer, interval time.Duration)
 			select {
 			case <-d.quit:
 				return
-			case <-time.After(interval):
+			case <-time.After(d.poll):
 				_, err := indexer.Index(d.db, d.cfg.Rebuild, d.cfg.Cleanup)
 				if err != nil {
 					log.WithError(err).Error("error during indexing")