86 lines
1.7 KiB
Go
86 lines
1.7 KiB
Go
|
package hodu
|
||
|
|
|
||
|
|
import "runtime"
|
||
|
|
import "github.com/prometheus/client_golang/prometheus"
|
||
|
|
|
||
|
|
type ClientCollector struct {
|
||
|
|
client *Client
|
||
|
|
BuildInfo *prometheus.Desc
|
||
|
|
ClientConns *prometheus.Desc
|
||
|
|
ClientRoutes *prometheus.Desc
|
||
|
|
ClientPeers *prometheus.Desc
|
||
|
|
}
|
||
|
|
|
||
|
|
// NewClientCollector returns a new ClientCollector with all prometheus.Desc initialized
|
||
|
|
func NewClientCollector(client *Client) ClientCollector {
|
||
|
|
var prefix string
|
||
|
|
|
||
|
|
prefix = client.Name() + "_"
|
||
|
|
return ClientCollector{
|
||
|
|
client: client,
|
||
|
|
|
||
|
|
BuildInfo: prometheus.NewDesc(
|
||
|
|
prefix + "build_info",
|
||
|
|
"Build information",
|
||
|
|
[]string{
|
||
|
|
"goarch",
|
||
|
|
"goos",
|
||
|
|
"goversion",
|
||
|
|
}, nil,
|
||
|
|
),
|
||
|
|
|
||
|
|
ClientConns: prometheus.NewDesc(
|
||
|
|
prefix + "client_conns",
|
||
|
|
"Number of client connections from clients",
|
||
|
|
nil, nil,
|
||
|
|
),
|
||
|
|
ClientRoutes: prometheus.NewDesc(
|
||
|
|
prefix + "client_routes",
|
||
|
|
"Number of client-side routes",
|
||
|
|
nil, nil,
|
||
|
|
),
|
||
|
|
ClientPeers: prometheus.NewDesc(
|
||
|
|
prefix + "client_peers",
|
||
|
|
"Number of client-side peers",
|
||
|
|
nil, nil,
|
||
|
|
),
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c ClientCollector) Describe(ch chan<- *prometheus.Desc) {
|
||
|
|
ch <- c.BuildInfo
|
||
|
|
ch <- c.ClientConns
|
||
|
|
ch <- c.ClientRoutes
|
||
|
|
ch <- c.ClientPeers
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c ClientCollector) Collect(ch chan<- prometheus.Metric) {
|
||
|
|
|
||
|
|
ch <- prometheus.MustNewConstMetric(
|
||
|
|
c.BuildInfo,
|
||
|
|
prometheus.GaugeValue,
|
||
|
|
1,
|
||
|
|
runtime.GOARCH,
|
||
|
|
runtime.GOOS,
|
||
|
|
runtime.Version(),
|
||
|
|
)
|
||
|
|
|
||
|
|
ch <- prometheus.MustNewConstMetric(
|
||
|
|
c.ClientConns,
|
||
|
|
prometheus.GaugeValue,
|
||
|
|
float64(c.client.stats.conns.Load()),
|
||
|
|
)
|
||
|
|
|
||
|
|
ch <- prometheus.MustNewConstMetric(
|
||
|
|
c.ClientRoutes,
|
||
|
|
prometheus.GaugeValue,
|
||
|
|
float64(c.client.stats.routes.Load()),
|
||
|
|
)
|
||
|
|
|
||
|
|
ch <- prometheus.MustNewConstMetric(
|
||
|
|
c.ClientPeers,
|
||
|
|
prometheus.GaugeValue,
|
||
|
|
float64(c.client.stats.peers.Load()),
|
||
|
|
)
|
||
|
|
}
|