diff --git a/integrations/inputunifi/influx/README.md b/integrations/inputunifi/influxunifi/README.md similarity index 100% rename from integrations/inputunifi/influx/README.md rename to integrations/inputunifi/influxunifi/README.md diff --git a/integrations/inputunifi/influx/clients.go b/integrations/inputunifi/influxunifi/clients.go similarity index 99% rename from integrations/inputunifi/influx/clients.go rename to integrations/inputunifi/influxunifi/clients.go index 114a9398..5a57331b 100644 --- a/integrations/inputunifi/influx/clients.go +++ b/integrations/inputunifi/influxunifi/clients.go @@ -1,4 +1,4 @@ -package influx +package influxunifi import ( "time" diff --git a/integrations/inputunifi/influx/ids.go b/integrations/inputunifi/influxunifi/ids.go similarity index 98% rename from integrations/inputunifi/influx/ids.go rename to integrations/inputunifi/influxunifi/ids.go index 33f743ba..bf444f5f 100644 --- a/integrations/inputunifi/influx/ids.go +++ b/integrations/inputunifi/influxunifi/ids.go @@ -1,4 +1,4 @@ -package influx +package influxunifi import ( influx "github.com/influxdata/influxdb1-client/v2" diff --git a/integrations/inputunifi/influx/metrics.go b/integrations/inputunifi/influxunifi/metrics.go similarity index 99% rename from integrations/inputunifi/influx/metrics.go rename to integrations/inputunifi/influxunifi/metrics.go index d49cfa8b..3919c8c7 100644 --- a/integrations/inputunifi/influx/metrics.go +++ b/integrations/inputunifi/influxunifi/metrics.go @@ -1,6 +1,6 @@ // Package influx provides the methods to turn UniFi measurements into influx // data-points with appropriate tags and fields. -package influx +package influxunifi import ( "github.com/davidnewhall/unifi-poller/metrics" diff --git a/integrations/inputunifi/influx/site.go b/integrations/inputunifi/influxunifi/site.go similarity index 99% rename from integrations/inputunifi/influx/site.go rename to integrations/inputunifi/influxunifi/site.go index 78619024..0bd647ac 100644 --- a/integrations/inputunifi/influx/site.go +++ b/integrations/inputunifi/influxunifi/site.go @@ -1,4 +1,4 @@ -package influx +package influxunifi import ( "strings" diff --git a/integrations/inputunifi/influx/uap.go b/integrations/inputunifi/influxunifi/uap.go similarity index 99% rename from integrations/inputunifi/influx/uap.go rename to integrations/inputunifi/influxunifi/uap.go index 02394e7c..1744b0c0 100644 --- a/integrations/inputunifi/influx/uap.go +++ b/integrations/inputunifi/influxunifi/uap.go @@ -1,4 +1,4 @@ -package influx +package influxunifi import ( "time" diff --git a/integrations/inputunifi/influx/udm.go b/integrations/inputunifi/influxunifi/udm.go similarity index 99% rename from integrations/inputunifi/influx/udm.go rename to integrations/inputunifi/influxunifi/udm.go index 2bbff525..fa968115 100644 --- a/integrations/inputunifi/influx/udm.go +++ b/integrations/inputunifi/influxunifi/udm.go @@ -1,4 +1,4 @@ -package influx +package influxunifi import ( "time" diff --git a/integrations/inputunifi/influx/usg.go b/integrations/inputunifi/influxunifi/usg.go similarity index 99% rename from integrations/inputunifi/influx/usg.go rename to integrations/inputunifi/influxunifi/usg.go index 8b358d0b..606011c0 100644 --- a/integrations/inputunifi/influx/usg.go +++ b/integrations/inputunifi/influxunifi/usg.go @@ -1,4 +1,4 @@ -package influx +package influxunifi import ( "strings" diff --git a/integrations/inputunifi/influx/usw.go b/integrations/inputunifi/influxunifi/usw.go similarity index 99% rename from integrations/inputunifi/influx/usw.go rename to integrations/inputunifi/influxunifi/usw.go index f2d1f9ea..225a4340 100644 --- a/integrations/inputunifi/influx/usw.go +++ b/integrations/inputunifi/influxunifi/usw.go @@ -1,4 +1,4 @@ -package influx +package influxunifi import ( "time" diff --git a/integrations/inputunifi/poller/unifi.go b/integrations/inputunifi/poller/unifi.go index e103c87a..b9d34011 100644 --- a/integrations/inputunifi/poller/unifi.go +++ b/integrations/inputunifi/poller/unifi.go @@ -6,7 +6,7 @@ import ( "strings" "time" - "github.com/davidnewhall/unifi-poller/influx" + "github.com/davidnewhall/unifi-poller/influxunifi" "github.com/davidnewhall/unifi-poller/metrics" client "github.com/influxdata/influxdb1-client/v2" "golift.io/unifi" @@ -90,7 +90,7 @@ func (u *UnifiPoller) CollectAndProcess(process func(*metrics.Metrics) error) er } // ExportMetrics updates the internal metrics provided via -// HTTP at /metrics for prometheus collection. +// HTTP at /metrics for prometheus collection. This is run by Prometheus. func (u *UnifiPoller) ExportMetrics() *metrics.Metrics { if u.Config.ReAuth { u.LogDebugf("Re-authenticating to UniFi Controller") @@ -100,7 +100,7 @@ func (u *UnifiPoller) ExportMetrics() *metrics.Metrics { return nil } } - + u.LastCheck = time.Now() m, err := u.CollectMetrics() if err != nil { u.LogErrorf("collecting metrics: %v", err) @@ -177,7 +177,7 @@ func (u *UnifiPoller) AugmentMetrics(metrics *metrics.Metrics) { // This creates an InfluxDB writer, and returns an error if the write fails. func (u *UnifiPoller) ReportMetrics(metrics *metrics.Metrics) error { // Batch (and send) all the points. - m := &influx.Metrics{Metrics: metrics} + m := &influxunifi.Metrics{Metrics: metrics} // Make a new Influx Points Batcher. var err error m.BatchPoints, err = client.NewBatchPoints(client.BatchPointsConfig{Database: u.Config.InfluxDB}) @@ -195,7 +195,7 @@ func (u *UnifiPoller) ReportMetrics(metrics *metrics.Metrics) error { } // LogInfluxReport writes a log message after exporting to influxdb. -func (u *UnifiPoller) LogInfluxReport(m *influx.Metrics) { +func (u *UnifiPoller) LogInfluxReport(m *influxunifi.Metrics) { var fields, points int for _, p := range m.Points() { points++ diff --git a/integrations/inputunifi/promunifi/collector.go b/integrations/inputunifi/promunifi/collector.go index b72dc569..1af860d4 100644 --- a/integrations/inputunifi/promunifi/collector.go +++ b/integrations/inputunifi/promunifi/collector.go @@ -50,6 +50,7 @@ func NewUnifiCollector(opts UnifiCollectorOpts) prometheus.Collector { if opts.CollectFn == nil { panic("nil collector function") } + return &unifiCollector{ opts: opts, Client: descClient(opts.Namespace), @@ -76,6 +77,7 @@ func (u *unifiCollector) Describe(ch chan<- *prometheus.Desc) { } } } + describe(u.Client) describe(u.UAP) describe(u.USG)