conn_pool_stats_replicasets.go

package collector

import (
	"sync"
	"time"

	"github.com/prometheus/client_golang/prometheus"
)

var (
	pingTime = prometheus.NewGaugeVec(prometheus.GaugeOpts{
		Namespace: Namespace,
		Subsystem: "connpoolstats",
		Name:      "ping_time_seconds",
		Help:      "Corresponds to the ping time from this mongos to the corresponding host in seconds",
	}, []string{"host", "rs"})

	// Lock for using these metrics
	connPoolReplicaSetStatsLock = sync.Mutex{}
)

type ReplicaSetHostStats struct {
	Host     string  `bson:"addr"`
	PingTime float64 `bson:"pingTimeMillis"`
}

type ReplicaSetStats struct {
	Hosts []*ReplicaSetHostStats `bson:"hosts"`
}

// Export exports the server status to be consumed by prometheus.
func (stats *ReplicaSetStats) Export(replicaSet string, ch chan<- prometheus.Metric) {
	connPoolReplicaSetStatsLock.Lock()
	defer connPoolReplicaSetStatsLock.Unlock()

	for _, rsHostStat := range stats.Hosts {
		pingTime.WithLabelValues(rsHostStat.Host, replicaSet).Set(rsHostStat.PingTime / float64(time.Second/time.Millisecond))
		pingTime.Collect(ch)
		pingTime.Reset()
	}
}

// Describe describes the server status for prometheus.
func (stats *ReplicaSetStats) Describe(ch chan<- *prometheus.Desc) {
	pingTime.Describe(ch)
}