node: add metrics for shard's capacity

CHANGELOG.md

@@ -23,6 +23,7 @@ attribute, which is used for container domain name in NNS contracts (#2954)
 - Reloading node attributes with SIGHUP (#3005)
 - Reloading pool sizes (#3018)
 - Reloading pprof/metrics services with SIGHUP (#3016)
+- Add metrics for shard's capacity (#3021)
 
 ### Fixed
 - Do not search for tombstones when handling their expiration, use local indexes instead (#2929)

cmd/neofs-node/config.go

@@ -948,17 +948,9 @@
 	var paths []string
 	for _, sh := range c.applicationConfiguration.engine.shards {
 		for _, subStorage := range sh.SubStorages {
-			path := subStorage.Path
-
-			for len(path) > 1 { // Dir returns / or . if nothing else left.
-				fi, err := os.Stat(path)
-				if err == nil && fi.IsDir() {
-					break
-				}
-				if err != nil && !errors.Is(err, fs.ErrNotExist) {
-					return fmt.Errorf("accessing %q: %w", path, err)
-				}
-				path = filepath.Dir(path)
+			path, err := getInitPath(subStorage.Path)
+			if err != nil {
+				return err
 			}
 
 			paths = append(paths, path)
@@ -975,6 +967,21 @@
 	return nil
 }
 
+func getInitPath(path string) (string, error) {
+	for len(path) > 1 { // Dir returns / or . if nothing else left.
+		fi, err := os.Stat(path)
+		if err == nil && fi.IsDir() {
+			break
+		}
+		if err != nil && !errors.Is(err, fs.ErrNotExist) {
+			return "", fmt.Errorf("accessing %q: %w", path, err)
+		}
+		path = filepath.Dir(path)
+	}
+
+	return path, nil
+}
+
 func (c *cfg) reloadMetricsAndPprof(oldMetrics metricConfig, oldProfiler profilerConfig) {
 	c.veryLastClosersLock.Lock()
 	defer c.veryLastClosersLock.Unlock()

cmd/neofs-node/main.go

@@ -61,6 +61,9 @@
 
 	initApp(c)
 
+	err = c.setShardsCapacity()
+	fatalOnErr(err)
+
 	c.setHealthStatus(control.HealthStatus_STARTING)
 
 	bootUp(c)

cmd/neofs-node/metrics.go

@@ -1,6 +1,7 @@
 package main
 
 import (
+	"fmt"
 	"time"
 
 	"github.com/nspcc-dev/neofs-node/cmd/neofs-node/config"
@@ -48,3 +49,26 @@
 		m1.shutdownTimeout != metricsconfig.ShutdownTimeout(c) ||
 		m1.address != metricsconfig.Address(c)
 }
+
+func (c *cfg) setShardsCapacity() error {
+	for _, sh := range c.cfgObject.cfgLocalStorage.localStorage.DumpInfo().Shards {
+		paths := make([]string, 0, len(sh.BlobStorInfo.SubStorages))
+		for _, subStorage := range sh.BlobStorInfo.SubStorages {
+			path, err := getInitPath(subStorage.Path)
+			if err != nil {
+				return err
+			}
+
+			paths = append(paths, path)
+		}
+
+		capacity, err := totalBytes(paths)
+		if err != nil {
+			return fmt.Errorf("calculating capacity on shard %s: %w", sh.ID.String(), err)
+		}
+
+		c.metricsCollector.SetCapacitySize(sh.ID.String(), capacity)
+	}
+
+	return nil
+}

pkg/metrics/engine.go

@@ -22,6 +22,7 @@
 
 		containerSize prometheus.GaugeVec
 		payloadSize   prometheus.GaugeVec
+		capacitySize  prometheus.GaugeVec
 	}
 )
 
@@ -121,6 +122,13 @@
 			Name:      "payload_size",
 			Help:      "Accumulated size of all objects in a shard",
 		}, []string{shardIDLabelKey})
+
+		capacitySize = prometheus.NewGaugeVec(prometheus.GaugeOpts{
+			Namespace: storageNodeNameSpace,
+			Subsystem: engineSubsystem,
+			Name:      "capacity",
+			Help:      "Contains the shard's capacity",
+		}, []string{shardIDLabelKey})
 	)
 
 	return engineMetrics{
@@ -137,6 +145,7 @@
 		listObjectsDuration:           listObjectsDuration,
 		containerSize:                 *containerSize,
 		payloadSize:                   *payloadSize,
+		capacitySize:                  *capacitySize,
 	}
 }
 
@@ -154,6 +163,7 @@
 	prometheus.MustRegister(m.listObjectsDuration)
 	prometheus.MustRegister(m.containerSize)
 	prometheus.MustRegister(m.payloadSize)
+	prometheus.MustRegister(m.capacitySize)
 }
 
 func (m engineMetrics) AddListContainersDuration(d time.Duration) {
@@ -211,3 +221,7 @@
 func (m engineMetrics) AddToPayloadCounter(shardID string, size int64) {
 	m.payloadSize.With(prometheus.Labels{shardIDLabelKey: shardID}).Add(float64(size))
 }
+
+func (m engineMetrics) SetCapacitySize(shardID string, capacity uint64) {
+	m.capacitySize.With(prometheus.Labels{shardIDLabelKey: shardID}).Set(float64(capacity))
+}