stellar#4433: fixes to incorporate after latest merge of metrics

sreuland · Jul 12, 2022 · ad2b90a · ad2b90a
1 parent 427611d
commit ad2b90a
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 2 deletions.
diff --git a/exp/lighthorizon/actions/main.go b/exp/lighthorizon/actions/main.go
@@ -16,7 +16,7 @@ import (
 
 var (
 	//go:embed static
-	staticFiles embed.FS
+	staticFiles  embed.FS
 	requestCount = promauto.NewCounter(prometheus.CounterOpts{
 		Name: "horizon_lite_request_count",
 		Help: "How many requests have occurred?",

diff --git a/exp/lighthorizon/index/mock_store.go b/exp/lighthorizon/index/mock_store.go
@@ -1,6 +1,7 @@
 package index
 
 import (
+	"github.com/prometheus/client_golang/prometheus"
 	types "github.com/stellar/go/exp/lighthorizon/index/types"
 	"github.com/stretchr/testify/mock"
 )
@@ -71,3 +72,7 @@ func (m *MockStore) MergeTransactions(prefix string, other *types.TrieIndex) err
 	args := m.Called(prefix, other)
 	return args.Error(0)
 }
+
+func (m *MockStore) RegisterMetrics(registry *prometheus.Registry) {
+	m.Called(registry)
+}
diff --git a/exp/lighthorizon/main.go b/exp/lighthorizon/main.go
@@ -65,7 +65,7 @@ func main() {
 	router.MethodFunc(http.MethodGet, "/operations", actions.Operations(lightHorizon))
 	router.MethodFunc(http.MethodGet, "/transactions", actions.Transactions(lightHorizon))
 	router.MethodFunc(http.MethodGet, "/", actions.ApiDocs())
-	router.MethodFunc(http.MethodGet, "/metrics", promhttp.HandlerFor(registry, promhttp.HandlerOpts{}))
+	router.Method(http.MethodGet, "/metrics", promhttp.HandlerFor(registry, promhttp.HandlerOpts{}))
 
 	log.Fatal(http.ListenAndServe(":8080", router))
 }