frostfs-node/pkg/services/metrics/prometeus.go
Stanislav Bogatyrev b7b5079934 Add Inner Ring code
2020-07-24 17:07:37 +03:00

83 lines
1.6 KiB
Go

package metrics
import (
"strings"
"github.com/prometheus/client_golang/prometheus"
"go.uber.org/atomic"
)
const (
locationLabel = "location"
countryLabel = "country"
cityLabel = "city"
containerLabel = "cid"
)
var (
objectsCount = prometheus.NewGaugeVec(prometheus.GaugeOpts{
Namespace: "neofs",
Name: "count_objects_on_node",
Help: "Number of objects stored on this node",
}, []string{locationLabel, countryLabel, cityLabel})
spaceCounter = prometheus.NewGaugeVec(prometheus.GaugeOpts{
Namespace: "neofs",
Name: "container_space_sizes",
Help: "Space allocated by ContainerID",
}, []string{containerLabel})
counter = atomic.NewFloat64(0)
)
func init() {
prometheus.MustRegister(
objectsCount,
spaceCounter,
)
}
func spaceUpdater(m *syncStore) func() {
return func() {
m.mutex.RLock()
for cid := range m.items {
spaceCounter.
With(prometheus.Labels{
containerLabel: cid.String(),
}).
Set(float64(m.items[cid]))
}
m.mutex.RUnlock()
}
}
func metricsUpdater(opts []string) func() {
var (
locationCode string
countryCode string
cityCode string
)
for i := range opts {
ss := strings.Split(opts[i], "/")
for j := range ss {
switch s := strings.SplitN(ss[j], ":", 2); strings.ToLower(s[0]) {
case locationLabel:
locationCode = s[1]
case countryLabel:
countryCode = s[1]
case cityLabel:
cityCode = s[1]
}
}
}
return func() {
objectsCount.With(prometheus.Labels{
locationLabel: locationCode,
countryLabel: countryCode,
cityLabel: cityCode,
}).Set(counter.Load())
}
}