fix: update metrics on startup

Author: blockchainluffy

rolling-shutter/keyper/epochkghandler/key.go

@@ -3,7 +3,6 @@ package epochkghandler
 import (
 	"bytes"
 	"context"
-	"strconv"
 
 	"github.com/jackc/pgx/v4"
 	"github.com/jackc/pgx/v4/pgxpool"
@@ -15,7 +14,6 @@ import (
 	"github.com/shutter-network/shutter/shlib/shcrypto"
 
 	"github.com/shutter-network/rolling-shutter/rolling-shutter/keyper/database"
-	"github.com/shutter-network/rolling-shutter/rolling-shutter/keyper/keypermetrics"
 	"github.com/shutter-network/rolling-shutter/rolling-shutter/medley"
 	"github.com/shutter-network/rolling-shutter/rolling-shutter/p2p"
 	"github.com/shutter-network/rolling-shutter/rolling-shutter/p2pmsg"
@@ -54,31 +52,25 @@ func (handler *DecryptionKeyHandler) ValidateMessage(ctx context.Context, msg p2
 		return pubsub.ValidationReject, err
 	}
 	if !isKeyper {
-		keypermetrics.MetricsKeyperIsKeyper.WithLabelValues(strconv.FormatInt(eon, 10)).Set(0)
 		log.Debug().Int64("eon", eon).Msg("Ignoring decryptionKey for eon; we're not a Keyper")
 		return pubsub.ValidationReject, nil
 	}
-	keypermetrics.MetricsKeyperIsKeyper.WithLabelValues(strconv.FormatInt(eon, 10)).Set(1)
 
 	dkgResultDB, err := queries.GetDKGResultForKeyperConfigIndex(ctx, eon)
 	if errors.Is(err, pgx.ErrNoRows) {
-		keypermetrics.MetricsKeyperSuccessfulDKG.WithLabelValues(strconv.FormatInt(eon, 10)).Set(0)
 		return pubsub.ValidationReject, errors.Errorf("no DKG result found for eon %d", eon)
 	}
 	if err != nil {
-		keypermetrics.MetricsKeyperSuccessfulDKG.WithLabelValues(strconv.FormatInt(eon, 10)).Set(0)
 		return pubsub.ValidationReject, errors.Wrapf(err, "failed to get dkg result for eon %d from db", eon)
 	}
 	if !dkgResultDB.Success {
-		keypermetrics.MetricsKeyperSuccessfulDKG.WithLabelValues(strconv.FormatInt(eon, 10)).Set(0)
 		return pubsub.ValidationReject, errors.Errorf("no successful DKG result found for eon %d", eon)
 	}
 	pureDKGResult, err := shdb.DecodePureDKGResult(dkgResultDB.PureResult)
 	if err != nil {
 		return pubsub.ValidationReject, errors.Wrapf(err, "error while decoding pure DKG result for eon %d", eon)
 	}
 
-	keypermetrics.MetricsKeyperSuccessfulDKG.WithLabelValues(strconv.FormatInt(eon, 10)).Set(1)
 	if len(decryptionKeys.Keys) == 0 {
 		return pubsub.ValidationReject, errors.New("no keys in message")
 	}

rolling-shutter/keyper/keyper.go

@@ -147,7 +147,7 @@ func (kpr *KeyperCore) Start(ctx context.Context, runner service.Runner) error {
 	messageSender := fx.NewRPCMessageSender(shuttermintClient, config.Ethereum.PrivateKey.Key)
 
 	if kpr.config.Metrics.Enabled {
-		keypermetrics.InitMetrics()
+		keypermetrics.InitMetrics(kpr.dbpool, kpr.config)
 		epochkghandler.InitMetrics()
 		deployment.InitMetrics()
 		kpr.metricsServer = metricsserver.New(kpr.config.Metrics)

rolling-shutter/keyper/keypermetrics/metrics.go

@@ -1,7 +1,15 @@
 package keypermetrics
 
 import (
+	"context"
+	"strconv"
+
+	"github.com/jackc/pgx/v4/pgxpool"
 	"github.com/prometheus/client_golang/prometheus"
+	"github.com/rs/zerolog/log"
+
+	"github.com/shutter-network/rolling-shutter/rolling-shutter/keyper/database"
+	"github.com/shutter-network/rolling-shutter/rolling-shutter/keyper/kprconfig"
 )
 
 var MetricsKeyperCurrentBlockL1 = prometheus.NewGauge(
@@ -89,7 +97,7 @@ var MetricsKeyperSuccessfulDKG = prometheus.NewGaugeVec(
 	[]string{"eon"},
 )
 
-func InitMetrics() {
+func InitMetrics(dbpool *pgxpool.Pool, config *kprconfig.Config) {
 	prometheus.MustRegister(MetricsKeyperCurrentBlockL1)
 	prometheus.MustRegister(MetricsKeyperCurrentBlockShuttermint)
 	prometheus.MustRegister(MetricsKeyperCurrentEon)
@@ -99,4 +107,33 @@ func InitMetrics() {
 	prometheus.MustRegister(MetricsKeyperCurrentBatchConfigIndex)
 	prometheus.MustRegister(MetricsKeyperBatchConfigInfo)
 	prometheus.MustRegister(MetricsKeyperSuccessfulDKG)
+
+	queries := database.New(dbpool)
+	eons, err := queries.GetAllEons(context.Background())
+	if err != nil {
+		log.Error().Err(err).Msg("Failed to get all eons")
+		return
+	}
+	keyperIndex, isKeyper, err := queries.GetKeyperIndex(context.Background(), eons[len(eons)-1].KeyperConfigIndex, config.GetAddress())
+	if err != nil {
+		log.Error().Err(err).Msg("Failed to get keyper index")
+		return
+	}
+	if isKeyper {
+		MetricsKeyperIsKeyper.WithLabelValues(strconv.FormatInt(keyperIndex, 10)).Set(1)
+	} else {
+		MetricsKeyperIsKeyper.WithLabelValues(strconv.FormatInt(keyperIndex, 10)).Set(0)
+	}
+
+	dkgResult, err := queries.GetDKGResultForKeyperConfigIndex(context.Background(), eons[len(eons)-1].KeyperConfigIndex)
+	if err != nil {
+		MetricsKeyperSuccessfulDKG.WithLabelValues(strconv.FormatInt(eons[len(eons)-1].Eon, 10)).Set(0)
+		log.Error().Err(err).Msg("Failed to get dkg result")
+		return
+	}
+	if dkgResult.Success {
+		MetricsKeyperSuccessfulDKG.WithLabelValues(strconv.FormatInt(eons[len(eons)-1].Eon, 10)).Set(1)
+	} else {
+		MetricsKeyperSuccessfulDKG.WithLabelValues(strconv.FormatInt(eons[len(eons)-1].Eon, 10)).Set(0)
+	}
 }