45 lines
879 B
Go
45 lines
879 B
Go
package peers
|
|
|
|
import (
|
|
"github.com/prometheus/client_golang/prometheus"
|
|
"google.golang.org/grpc/connectivity"
|
|
)
|
|
|
|
const stateLabel = "state"
|
|
|
|
var grpcConnections = prometheus.NewGaugeVec(
|
|
prometheus.GaugeOpts{
|
|
Help: "gRPC connections",
|
|
Name: "grpc_connections",
|
|
Namespace: "neofs",
|
|
},
|
|
[]string{stateLabel},
|
|
)
|
|
|
|
var conStates = []connectivity.State{
|
|
connectivity.Idle,
|
|
connectivity.Connecting,
|
|
connectivity.Ready,
|
|
connectivity.TransientFailure,
|
|
connectivity.Shutdown,
|
|
}
|
|
|
|
func updateMetrics(items map[connectivity.State]float64) {
|
|
for _, state := range conStates {
|
|
grpcConnections.With(prometheus.Labels{
|
|
stateLabel: state.String(),
|
|
}).Set(items[state])
|
|
}
|
|
}
|
|
|
|
func init() {
|
|
prometheus.MustRegister(
|
|
grpcConnections,
|
|
)
|
|
|
|
for _, state := range conStates {
|
|
grpcConnections.With(prometheus.Labels{
|
|
stateLabel: state.String(),
|
|
}).Set(0)
|
|
}
|
|
}
|