From 9410aa78a0941764074b82ec20d511742d4780fd Mon Sep 17 00:00:00 2001
From: Mark Holt <mark.holt@erigon.tech>
Date: Thu, 17 Oct 2024 10:56:31 +0100
Subject: [PATCH] post merge fixes

---
 polygon/heimdall/service.go | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/polygon/heimdall/service.go b/polygon/heimdall/service.go
index 5ff6aa70136..f1229b7ef7f 100644
--- a/polygon/heimdall/service.go
+++ b/polygon/heimdall/service.go
@@ -60,6 +60,11 @@ type service struct {
 	spanBlockProducersTracker *spanBlockProducersTracker
 }
 
+func AssembleService(config ServiceConfig) Service {
+	client := NewHeimdallClient(config.HeimdallURL, config.Logger)
+	return NewService(config.BorConfig, client, config.Store, config.Logger)
+}
+
 func NewService(borConfig *borcfg.BorConfig, client HeimdallClient, store Store, logger log.Logger) Service {
 	return newService(borConfig, client, store, logger)
 }
@@ -107,7 +112,7 @@ func newService(borConfig *borcfg.BorConfig, client HeimdallClient, store Store,
 	return &service{
 		logger:                    logger,
 		store:                     store,
-		reader:                    NewReader(calculateSprintNumberFn, store, logger),
+		reader:                    NewReader(borConfig, store, logger),
 		checkpointScraper:         checkpointScraper,
 		milestoneScraper:          milestoneScraper,
 		spanScraper:               spanScraper,