89 lines
2.0 KiB
Go
89 lines
2.0 KiB
Go
// +build go1.3
|
||
package prometheus
|
||
|
||
import (
|
||
"strings"
|
||
"sync"
|
||
"time"
|
||
|
||
"github.com/prometheus/client_golang/prometheus"
|
||
)
|
||
|
||
type PrometheusSink struct {
|
||
mu sync.Mutex
|
||
gauges map[string]prometheus.Gauge
|
||
summaries map[string]prometheus.Summary
|
||
counters map[string]prometheus.Counter
|
||
}
|
||
|
||
func NewPrometheusSink() (*PrometheusSink, error) {
|
||
return &PrometheusSink{
|
||
gauges: make(map[string]prometheus.Gauge),
|
||
summaries: make(map[string]prometheus.Summary),
|
||
counters: make(map[string]prometheus.Counter),
|
||
}, nil
|
||
}
|
||
|
||
func (p *PrometheusSink) flattenKey(parts []string) string {
|
||
joined := strings.Join(parts, "_")
|
||
joined = strings.Replace(joined, " ", "_", -1)
|
||
joined = strings.Replace(joined, ".", "_", -1)
|
||
joined = strings.Replace(joined, "-", "_", -1)
|
||
return joined
|
||
}
|
||
|
||
func (p *PrometheusSink) SetGauge(parts []string, val float32) {
|
||
p.mu.Lock()
|
||
defer p.mu.Unlock()
|
||
key := p.flattenKey(parts)
|
||
g, ok := p.gauges[key]
|
||
if !ok {
|
||
g = prometheus.NewGauge(prometheus.GaugeOpts{
|
||
Name: key,
|
||
Help: key,
|
||
})
|
||
prometheus.MustRegister(g)
|
||
p.gauges[key] = g
|
||
}
|
||
g.Set(float64(val))
|
||
}
|
||
|
||
func (p *PrometheusSink) AddSample(parts []string, val float32) {
|
||
p.mu.Lock()
|
||
defer p.mu.Unlock()
|
||
key := p.flattenKey(parts)
|
||
g, ok := p.summaries[key]
|
||
if !ok {
|
||
g = prometheus.NewSummary(prometheus.SummaryOpts{
|
||
Name: key,
|
||
Help: key,
|
||
MaxAge: 10 * time.Second,
|
||
})
|
||
prometheus.MustRegister(g)
|
||
p.summaries[key] = g
|
||
}
|
||
g.Observe(float64(val))
|
||
}
|
||
|
||
// EmitKey is not implemented. Prometheus doesn’t offer a type for which an
|
||
// arbitrary number of values is retained, as Prometheus works with a pull
|
||
// model, rather than a push model.
|
||
func (p *PrometheusSink) EmitKey(key []string, val float32) {
|
||
}
|
||
|
||
func (p *PrometheusSink) IncrCounter(parts []string, val float32) {
|
||
p.mu.Lock()
|
||
defer p.mu.Unlock()
|
||
key := p.flattenKey(parts)
|
||
g, ok := p.counters[key]
|
||
if !ok {
|
||
g = prometheus.NewCounter(prometheus.CounterOpts{
|
||
Name: key,
|
||
Help: key,
|
||
})
|
||
prometheus.MustRegister(g)
|
||
p.counters[key] = g
|
||
}
|
||
g.Add(float64(val))
|
||
}
|