report.go

package influxunifi

import (
	"sync"
	"time"

	"github.com/davidnewhall/unifi-poller/pkg/metrics"
	influx "github.com/influxdata/influxdb1-client/v2"
)

// Report is returned to the calling procedure after everything is processed.
type Report struct {
	Metrics *metrics.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() *metrics.Metrics
}

func (r *Report) metrics() *metrics.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)
}