*: Register prometheus metric for each type of tracker

Signed-off-by: Evgenii Baidakov <[email protected]>
nspcc-dev · Dec 22, 2023 · 4e8602f · 4e8602f
1 parent 7848479
commit 4e8602f
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 8 deletions.
diff --git a/cmd/neofs-net-monitor/main.go b/cmd/neofs-net-monitor/main.go
@@ -10,7 +10,6 @@ import (
 	"strings"
 	"syscall"
 
-	"github.com/nspcc-dev/neofs-net-monitor/pkg/monitor"
 	"github.com/spf13/viper"
 	"go.uber.org/zap"
 )
@@ -45,8 +44,6 @@ func main() {
 		os.Exit(1)
 	}
 
-	monitor.RegisterMetrics()
-
 	neofsMonitor.Start(ctx)
 
 	neofsMonitor.Logger().Info("application started", zap.String("version", Version))

diff --git a/cmd/neofs-net-monitor/monitor.go b/cmd/neofs-net-monitor/monitor.go
@@ -38,8 +38,10 @@ func New(ctx context.Context, cfg *viper.Viper) (*monitor.Monitor, error) {
 
 	var job monitor.Job
 	if cfg.GetBool(cfgChainFSChain) {
+		monitor.RegisterSideChainMetrics()
 		job, err = sideChainJob(ctx, cfg, sideNeogoClient, logger, sideChainEndpoints)
 	} else {
+		monitor.RegisterMainChainMetrics()
 		job, err = mainChainJob(cfg, sideNeogoClient, logger)
 	}
 

diff --git a/pkg/monitor/metrics.go b/pkg/monitor/metrics.go
@@ -169,22 +169,27 @@ var (
 	)
 )
 
-// RegisterMetrics inits prometheus metrics. Panics if can't do it.
-func RegisterMetrics() {
+// RegisterSideChainMetrics inits prometheus metrics for side chain. Panics if can't do it.
+func RegisterSideChainMetrics() {
 	prometheus.MustRegister(locationPresent)
 	prometheus.MustRegister(droppedNodesCount)
 	prometheus.MustRegister(newNodesCount)
 	prometheus.MustRegister(epochNumber)
 	prometheus.MustRegister(storageNodeGASBalances)
 	prometheus.MustRegister(storageNodeNotaryBalances)
 	prometheus.MustRegister(innerRingBalances)
-	prometheus.MustRegister(alphabetGASBalances)
 	prometheus.MustRegister(alphabetNotaryBalances)
 	prometheus.MustRegister(proxyBalance)
-	prometheus.MustRegister(mainChainSupply)
 	prometheus.MustRegister(sideChainSupply)
-	prometheus.MustRegister(alphabetPubKeys)
+	prometheus.MustRegister(alphabetPubKeys) // used for both monitors
 	prometheus.MustRegister(containersNumber)
 	prometheus.MustRegister(chainHeight)
 	prometheus.MustRegister(chainState)
 }
+
+// RegisterMainChainMetrics inits prometheus metrics for main chain. Panics if can't do it.
+func RegisterMainChainMetrics() {
+	prometheus.MustRegister(alphabetGASBalances)
+	prometheus.MustRegister(mainChainSupply)
+	prometheus.MustRegister(alphabetPubKeys) // used for both monitors
+}