72 lines
1.5 KiB
Go
72 lines
1.5 KiB
Go
package promunifi
|
|
|
|
import (
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/davidnewhall/unifi-poller/metrics"
|
|
"github.com/prometheus/client_golang/prometheus"
|
|
)
|
|
|
|
// This file contains the report interface.
|
|
// This interface can be mocked and overrridden for tests.
|
|
|
|
// report is an internal interface used to "process metrics"
|
|
type report interface {
|
|
send([]*metricExports)
|
|
add()
|
|
done()
|
|
metrics() *metrics.Metrics
|
|
report(descs map[*prometheus.Desc]bool)
|
|
export(m *metricExports, v float64) prometheus.Metric
|
|
channel() chan []*metricExports
|
|
error(ch chan<- prometheus.Metric, d *prometheus.Desc, v interface{})
|
|
}
|
|
|
|
// satisfy gomnd
|
|
const one = 1
|
|
|
|
func (r *Report) report(descs map[*prometheus.Desc]bool) {
|
|
if r.cf.LoggingFn == nil {
|
|
return
|
|
}
|
|
r.Descs, r.Elapsed = len(descs), time.Since(r.Start)
|
|
r.cf.LoggingFn(r)
|
|
}
|
|
|
|
func (r *Report) export(m *metricExports, v float64) prometheus.Metric {
|
|
r.Total++
|
|
if v == 0 {
|
|
r.Zeros++
|
|
}
|
|
return prometheus.MustNewConstMetric(m.Desc, m.ValueType, v, m.Labels...)
|
|
}
|
|
|
|
func (r *Report) metrics() *metrics.Metrics {
|
|
return r.Metrics
|
|
}
|
|
|
|
func (r *Report) channel() chan []*metricExports {
|
|
return r.ch
|
|
}
|
|
|
|
func (r *Report) error(ch chan<- prometheus.Metric, d *prometheus.Desc, v interface{}) {
|
|
r.Errors++
|
|
if r.cf.ReportErrors {
|
|
ch <- prometheus.NewInvalidMetric(d, fmt.Errorf("not a number: %v", v))
|
|
}
|
|
}
|
|
|
|
func (r *Report) add() {
|
|
r.wg.Add(one)
|
|
}
|
|
|
|
func (r *Report) done() {
|
|
r.wg.Add(-one)
|
|
}
|
|
|
|
func (r *Report) send(m []*metricExports) {
|
|
r.wg.Add(one)
|
|
r.ch <- m
|
|
}
|