scriptoriumadmin/featurebase
1
0
Fork 0
mirror of https://github.com/featurebasedb/featurebase.git synced 2026-09-10 23:11:01 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add options to prometheus client to support setting namespace

Browse source
This commit is contained in:
Alan Bernstein 2020-04-02 19:05:55 -05:00
parent 7432546af8
commit f15d031c55
1 changed files with 25 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

33
prometheus/prometheus.go
View file

@ -26,8 +26,7 @@ import (
)
const (
// namespace is prepended to each metric event name with "_"
namespace = "pilosa"
defaultNamespace = "pilosa"
)
// Ensure client implements interface.
@ -46,11 +45,22 @@ type prometheusClient struct {
gaugeVecs map[string]*prometheus.GaugeVec
observers map[string]prometheus.Observer
summaryVecs map[string]*prometheus.SummaryVec
namespace string
}
// ClientOption is a functional option type for prometheusClient
type ClientOption func(c *prometheusClient)
// OptClientPrefix is a functional option on prometheusClient used to set the namespace
func OptClientNamespace(namespace string) ClientOption {
return func(c *prometheusClient) {
c.namespace = namespace
}
}
// NewPrometheusClient returns a new instance of StatsClient.
func NewPrometheusClient() (*prometheusClient, error) {
return &prometheusClient{
func NewPrometheusClient(opts ...ClientOption) (*prometheusClient, error) {
client := &prometheusClient{
logger: logger.NopLogger,
counters: make(map[string]prometheus.Counter),
counterVecs: make(map[string]*prometheus.CounterVec),
@ -58,7 +68,14 @@ func NewPrometheusClient() (*prometheusClient, error) {
gaugeVecs: make(map[string]*prometheus.GaugeVec),
observers: make(map[string]prometheus.Observer),
summaryVecs: make(map[string]*prometheus.SummaryVec),
}, nil
namespace: defaultNamespace,
}
for _, opt := range opts {
opt(client)
}
return client, nil
}
// Open no-op to satisfy interface
@ -103,7 +120,7 @@ func (c *prometheusClient) Count(name string, value int64, rate float64) {
name = strings.Replace(name, ".", "_", -1)
labels := c.labels()
opts := prometheus.CounterOpts{
Namespace: namespace,
Namespace: c.namespace,
Name: name,
}
if len(labels) == 0 {
@ -152,7 +169,7 @@ func (c *prometheusClient) Gauge(name string, value float64, rate float64) {
name = strings.Replace(name, ".", "_", -1)
labels := c.labels()
opts := prometheus.GaugeOpts{
Namespace: namespace,
Namespace: c.namespace,
Name: name,
}
if len(labels) == 0 {
@ -193,7 +210,7 @@ func (c *prometheusClient) Histogram(name string, value float64, rate float64) {
name = strings.Replace(name, ".", "_", -1)
labels := c.labels()
opts := prometheus.SummaryOpts{
Namespace: namespace,
Namespace: c.namespace,
Name: name,
Objectives: map[float64]float64{0.5: 0.05, 0.9: 0.01, 0.99: 0.001},
}