Move metrics update to a function

stellar · Oct 22, 2020 · d6f557b · d6f557b
1 parent cf68d25
commit d6f557b
Showing 1 changed file with 10 additions and 10 deletions.
diff --git a/services/horizon/internal/ingest/fsm.go b/services/horizon/internal/ingest/fsm.go
@@ -476,18 +476,10 @@ func (r resumeState) run(s *system) (transition, error) {
 
 	// Update stats metrics
 	changeStatsMap := changeStats.Map()
-	for stat, value := range changeStatsMap {
-		stat = strings.Replace(stat, "stats_", "change_", 1)
-		s.Metrics().LedgerStatsCounter.
-			With(prometheus.Labels{"type": stat}).Add(float64(value.(int64)))
-	}
+	r.addLedgerStatsMetricFromMap(s, "change", changeStatsMap)
 
 	ledgerTransactionStatsMap := ledgerTransactionStats.Map()
-	for stat, value := range ledgerTransactionStatsMap {
-		stat = strings.Replace(stat, "stats_", "ledger_", 1)
-		s.Metrics().LedgerStatsCounter.
-			With(prometheus.Labels{"type": stat}).Add(float64(value.(int64)))
-	}
+	r.addLedgerStatsMetricFromMap(s, "ledger", ledgerTransactionStatsMap)
 
 	log.
 		WithFields(changeStatsMap).
@@ -506,6 +498,14 @@ func (r resumeState) run(s *system) (transition, error) {
 	return resumeImmediately(ingestLedger), nil
 }
 
+func (r resumeState) addLedgerStatsMetricFromMap(s *system, prefix string, m map[string]interface{}) {
+	for stat, value := range m {
+		stat = strings.Replace(stat, "stats_", prefix+"_", 1)
+		s.Metrics().LedgerStatsCounter.
+			With(prometheus.Labels{"type": stat}).Add(float64(value.(int64)))
+	}
+}
+
 type historyRangeState struct {
 	fromLedger uint32
 	toLedger   uint32