65 lines
1.1 KiB
Go
65 lines
1.1 KiB
Go
package influxunifi
|
|
|
|
import (
|
|
"sync"
|
|
"time"
|
|
|
|
"github.com/davidnewhall/unifi-poller/pkg/poller"
|
|
influx "github.com/influxdata/influxdb1-client/v2"
|
|
)
|
|
|
|
// Report is returned to the calling procedure after everything is processed.
|
|
type Report struct {
|
|
Metrics *poller.Metrics
|
|
Errors []error
|
|
Total int
|
|
Fields int
|
|
Start time.Time
|
|
Elapsed time.Duration
|
|
ch chan *metric
|
|
wg sync.WaitGroup
|
|
bp influx.BatchPoints
|
|
}
|
|
|
|
// report is an internal interface that can be mocked and overrridden for tests.
|
|
type report interface {
|
|
add()
|
|
done()
|
|
send(m *metric)
|
|
error(err error)
|
|
batch(m *metric, pt *influx.Point)
|
|
metrics() *poller.Metrics
|
|
}
|
|
|
|
func (r *Report) metrics() *poller.Metrics {
|
|
return r.Metrics
|
|
}
|
|
|
|
// satisfy gomnd
|
|
const one = 1
|
|
|
|
func (r *Report) add() {
|
|
r.wg.Add(one)
|
|
}
|
|
|
|
func (r *Report) done() {
|
|
r.wg.Add(-one)
|
|
}
|
|
|
|
func (r *Report) send(m *metric) {
|
|
r.wg.Add(one)
|
|
r.ch <- m
|
|
}
|
|
|
|
/* The following methods are not thread safe. */
|
|
|
|
func (r *Report) error(err error) {
|
|
r.Errors = append(r.Errors, err)
|
|
}
|
|
|
|
func (r *Report) batch(m *metric, p *influx.Point) {
|
|
r.Total++
|
|
r.Fields += len(m.Fields)
|
|
r.bp.AddPoint(p)
|
|
}
|